Merge pull request #229 from aziontech/fix-ai-agent

fix(AI Agent)

Author: PedroMiolaSilva

templates/typescript/azion-react-agent/src/nodes/callModel.ts

@@ -17,8 +17,6 @@ export async function callModel(
 ): Promise<Partial<typeof MessagesAnnotation.State>> {
 
   const { messages } = state
-
-  messages.unshift(new SystemMessage({content: SYSTEM_PROMPT}))
 
   const model = new ChatOpenAI({
     model: OPENAI_MODEL,
@@ -29,7 +27,7 @@ export async function callModel(
     apiKey: OPENAI_API_KEY,
   }).bindTools(TOOLS);
 
-  const response = new AIMessage(await model.invoke(messages,{recursionLimit:3}))
+  const response = new AIMessage(await model.invoke([new SystemMessage(SYSTEM_PROMPT),...messages],{recursionLimit:3}))
 
   return { messages: [response] };
 }

templates/typescript/azion-react-agent/src/nodes/tools.ts

@@ -22,8 +22,8 @@ const azionRetriever = new AzionRetriever(embeddingModel, entityExtractor,
   {dbName:VECTOR_STORE_DB_NAME,
    vectorTable:VECTOR_STORE_TABLE_NAME,
    ftsTable:VECTOR_STORE_TABLE_NAME+"_fts",
-   ftsK:1,
-   similarityK:1,
+   ftsK:3,
+   similarityK:3,
    searchType:"hybrid",
    promptEntityExtractor:'Extraia as entidades relevantes para a busca. Apenas responda com as entidades, sem explicações e pontuação.'
   });