fix: bug in traces, missing input values

ax-llm · Nov 20, 2024 · d8a8ee5 · d8a8ee5
1 parent 2c96fc1
commit d8a8ee5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/ax/dsp/generate.ts b/src/ax/dsp/generate.ts
@@ -422,7 +422,7 @@ export class AxGen<
           }
 
           assertRequiredFields(sig, output);
-          this.trace = { ...output };
+          this.trace = { ...values, ...output };
           return output;
         } catch (e) {
           let extraFields;

diff --git a/src/examples/summarize.ts b/src/examples/summarize.ts
@@ -5,7 +5,6 @@ const noteText = `The technological singularity—or simply the singularity[1]
 // const ai = new AxAI({ name: 'ollama', model: 'nous-hermes2' });
 
 const gen = new AxChainOfThought(
-  ai,
   `noteText -> shortSummary "summarize in 5 to 10 words"`
 );
 gen.setExamples([
@@ -44,4 +43,6 @@ const res = await gen.forward(
   { modelConfig: { stream: true } }
 );
 
-console.log('>', res);
+console.log('Traces:\n', gen.getTraces());
+
+console.log('Result:\n', res);