fix intent and decision agent

Author: fishonamos

apps/web/test/integration/health.test.ts

@@ -15,12 +15,9 @@ describe('Health Check Endpoints', () => {
   });
 
   it('GET / should return API info', async () => {
-    const response = await request(testApp)
-      .get('/')
-      .expect('Content-Type', /json/)
-      .expect(200);
+    const response = await request(testApp).get('/').expect('Content-Type', /json/).expect(200);
 
     expect(response.body).toHaveProperty('message', 'Atoma Agents API');
     expect(response.body).toHaveProperty('version', '1.0.0');
   });
-}); 
+});

apps/web/test/setup.ts

@@ -30,4 +30,4 @@ process.on('SIGTERM', () => {
 
 process.on('SIGINT', () => {
   process.exit(0);
-}); 
+});

packages/sui-agent/src/agents/SuiAgent.ts

@@ -20,7 +20,9 @@ class Agents {
 
   constructor(bearerAuth: string) {
     this.tools = new Tools(bearerAuth, intent_agent_prompt);
-    this.utils = new Utils(bearerAuth, final_answer_agent_prompt);
+    this.utils = new Utils(bearerAuth, final_answer_agent_prompt, this.tools);
+    // Register tools when agent is instantiated
+    registerAllTools(this.tools);
   }
 
   /**
@@ -29,8 +31,6 @@ class Agents {
    * @returns IntentAgentResponse containing tool selection and processing details
    */
   async IntentAgent(prompt: string) {
-    // Register all available tools before processing
-    registerAllTools(this.tools);
     const IntentResponse: IntentAgentResponse =
       (await this.tools.selectAppropriateTool(prompt)) as IntentAgentResponse;
     return IntentResponse;
@@ -46,7 +46,12 @@ class Agents {
     intentResponse: IntentAgentResponse,
     query: string,
   ) {
-    return await this.utils.processQuery(query);
+    // Pass both the selected tool name and arguments to processQuery
+    return await this.utils.processQuery(
+      query,
+      intentResponse.selected_tool,
+      intentResponse.tool_arguments,
+    );
   }
 
   /**
@@ -58,11 +63,11 @@ class Agents {
   async SuperVisorAgent(prompt: string) {
     // Process intent
     const res = await this.IntentAgent(prompt);
-    console.log(res);
+    console.log('Intent Response:', res);
 
     // Make decision based on intent
     const finalAnswer = await this.DecisionMakingAgent(res, prompt);
-    console.log(finalAnswer, 'final');
+    console.log('Final Answer:', finalAnswer);
     return finalAnswer;
   }
 }

packages/sui-agent/src/config/atoma.ts

@@ -4,7 +4,8 @@ import * as dotenv from 'dotenv';
 // Load environment variables
 dotenv.config();
 
-const ATOMA_CHAT_COMPLETIONS_MODEL = process.env.ATOMA_CHAT_MODEL || 'meta-llama/Llama-3.3-70B-Instruct';
+const ATOMA_CHAT_COMPLETIONS_MODEL =
+  process.env.ATOMA_CHAT_MODEL || 'meta-llama/Llama-3.3-70B-Instruct';
 const sdkInstance: AtomaSDK | null = null;
 
 /**

packages/sui-agent/src/utils/index.ts

@@ -9,32 +9,43 @@ import { ToolArgument } from '../@types/interface';
  * Handles the execution of tools and formatting of final responses
  */
 class Utils {
-  private tools: Tools;
   private sdk: AtomaSDK;
   private prompt: string;
+  private tools: Tools;
 
-  constructor(bearerAuth: string, prompt: string) {
-    this.tools = new Tools(bearerAuth, prompt);
+  constructor(bearerAuth: string, prompt: string, tools?: Tools) {
     this.sdk = new AtomaSDK({ bearerAuth });
     this.prompt = prompt;
+    // Use provided tools instance or create new one
+    this.tools = tools || new Tools(bearerAuth, prompt);
+  }
+
+  /**
+   * Set tools instance
+   * @param tools - Tools instance to use
+   */
+  setTools(tools: Tools) {
+    this.tools = tools;
   }
 
   /**
    * Process user query and execute appropriate tool
    * @param query - User query
+   * @param selectedTool - Name of the tool to execute
+   * @param toolArguments - Arguments to pass to the tool
    * @returns Processed response
    */
-  async processQuery(query: string) {
+  async processQuery(
+    query: string,
+    selectedTool: string | null,
+    toolArguments: ToolArgument[] = [],
+  ) {
     try {
-      const selectedTool = await this.tools.selectAppropriateTool(query);
       if (!selectedTool) {
-        return this.finalAnswer('No tool found for the query', query);
+        return this.finalAnswer('No tool selected for the query', query);
       }
 
-      return this.executeTools(
-        selectedTool.selected_tool || '',
-        selectedTool.tool_arguments || [],
-      );
+      return this.executeTools(selectedTool, toolArguments);
     } catch (error: unknown) {
       console.error('Error processing query:', error);
       return handleError(error, {