@ai.ntellect/core 0.1.85 → 0.1.89

package/agent/index.ts

@@ -186,7 +186,6 @@ export class Agent {
     const sanitizedResults = ResultSanitizer.sanitize(this.accumulatedResults);
     const summaryData = JSON.stringify({
       result: sanitizedResults,
-      initialPrompt: actionsResult.initialPrompt,
     });
 
     this.accumulatedResults = [];
@@ -200,8 +199,13 @@ export class Agent {
     });
 
     return this.stream
-      ? (await synthesizer.streamProcess(summaryData)).toDataStreamResponse()
-      : await synthesizer.process(summaryData);
+      ? (
+          await synthesizer.streamProcess(
+            actionsResult.initialPrompt,
+            summaryData
+          )
+        ).toDataStreamResponse()
+      : await synthesizer.process(actionsResult.initialPrompt, summaryData);
   }
 
   private transformActions(actions: ActionSchema[]) {

package/llm/synthesizer/context.ts

@@ -25,10 +25,12 @@ export const synthesizerContext = {
       "NEVER explain technical errors or issues. Just say retry later.",
     ],
   },
-  compose: (results: string) => {
+  compose: (initialPrompt: string, summaryData?: string) => {
     return `
       ${JSON.stringify(synthesizerContext.guidelines)}
-      Results: ${results}
+
+      Initial prompt: ${initialPrompt} (Speak in the same language as the initial prompt)
+      Results: ${summaryData}
 
       1. FOR ALL ANALYSIS OF SPECIFIC TOKEN, RESPECT THE FOLLOWING FORMAT:
       --------------------------------
@@ -44,8 +46,21 @@ export const synthesizerContext = {
 
       STOP AFTER TECHNICAL ANALYSIS SECTION WITHOUT ANY CONCLUDING STATEMENT OR DISCLAIMER OR ADDITIONAL COMMENTS
       --------------------------------
+
+      2. FOR SECURITY CHECKS, USE THE FOLLOWING FORMAT:
+      --------------------------------
+      ## Security check of x/y:
+
+      ### Good:
+      Speak about the good points of the security check. If there is no good point, say "No good point found"
+
+      ### Bad:
+      Speak about the bad points of the security check. If there is no bad point, say "No bad point found"
+
+      STOP AFTER SECURITY CHECK SECTION WITHOUT ANY CONCLUDING STATEMENT OR DISCLAIMER OR ADDITIONAL COMMENTS
+      --------------------------------
       
-      2. OTHERWISE FOR OTHER REQUESTS, USE THE FORMAT YOU WANT.
+      3. OTHERWISE FOR OTHER REQUESTS, USE THE FORMAT YOU WANT.
     `;
   },
 };

package/llm/synthesizer/index.ts

@@ -9,6 +9,7 @@ export class Synthesizer implements BaseLLM {
 
   async process(
     prompt: string,
+    summaryData?: string,
     onFinish?: (event: any) => void
   ): Promise<
     | {
@@ -32,7 +33,7 @@ export class Synthesizer implements BaseLLM {
         ),
         response: z.string(),
       }),
-      prompt: synthesizerContext.compose(prompt),
+      prompt: synthesizerContext.compose(prompt, summaryData || ""),
       system: synthesizerContext.role,
     });
     console.log("Synthesizer");
@@ -43,11 +44,12 @@ export class Synthesizer implements BaseLLM {
 
   async streamProcess(
     prompt: string,
+    summaryData?: string,
     onFinish?: (event: any) => void
   ): Promise<StreamTextResult<Record<string, any>>> {
     const result = await streamText({
       model: this.model,
-      prompt: synthesizerContext.compose(prompt),
+      prompt: synthesizerContext.compose(prompt, summaryData || ""),
       onFinish: onFinish,
       system: synthesizerContext.role,
     });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai.ntellect/core",
-  "version": "0.1.85",
+  "version": "0.1.89",
   "description": "",
   "main": "dist/index.js",
   "scripts": {