update function names and more fixes

fishonamos · fishonamos · commit b4a8d794c3f9 · 2025-02-18T10:19:54.000+01:00
diff --git a/README.md b/README.md
@@ -53,7 +53,7 @@ const suiAgent = new Agent('your_atoma_sdk_bearer_token');
 // Use the agent to process queries
 async function processQuery(query: string) {
   try {
-    const result = await suiAgent.SuperVisorAgent(query);
+    const result = await suiAgent.processUserQueryPipeline(query);
     return result;
   } catch (error) {
     console.error('Error processing query:', error);
@@ -72,7 +72,7 @@ const suiAgent = new Agent(config.atomaSdkBearerAuth);
 app.post('/query', async (req, res) => {
   try {
     const { query } = req.body;
-    const result = await suiAgent.SuperVisorAgent(query);
+    const result = await suiAgent.processUserQueryPipeline(query);
     res.json(result);
   } catch (error) {
     res.status(500).json({ error: 'Internal server error' });
diff --git a/apps/web/src/controllers/conversation.controller.ts b/apps/web/src/controllers/conversation.controller.ts
@@ -39,7 +39,7 @@ class ConversationController {
         res.status(400).json({ error: 'Invalid conversation ID' });
         return;
       }
-      const result = await suiAgent.SuperVisorAgent(message, walletAddress);
+      const result = await suiAgent.processUserQueryPipeline(message, walletAddress);
       const newMessage = await this.messageService.createMessage({
         sender: sender ? sender : 'user',
         walletAddress,
diff --git a/apps/web/src/routes/v1/query.ts b/apps/web/src/routes/v1/query.ts
@@ -20,7 +20,7 @@ const handleQuery = async (req: Request, res: Response): Promise<void> => {
       return;
     }
     // Get agent response first
-    const result = await suiAgent.SuperVisorAgent(query);
+    const result = await suiAgent.processUserQueryPipeline(query);
     res.status(200).json(result);
   } catch (error) {
     console.error('Error handling query:', error);
diff --git a/packages/sui-agent/src/agents/SuiAgent.ts b/packages/sui-agent/src/agents/SuiAgent.ts
@@ -2,7 +2,7 @@ import { IntentAgentResponse } from '../@types/interface';
 import Tools from '../utils/tools';
 import { registerAllTools } from './ToolRegistry';
 import Utils from '../utils';
-import intent_agent_prompt from '../prompts/intent_agent_prompt';
+import intent_agent_prompt from '../prompts/tool_selector';
 import final_answer_agent_prompt from '../prompts/final_answer_agent';
 import Atoma from '../config/atoma';
 import decomposerPrompt from '../prompts/decomposer';
@@ -13,7 +13,7 @@ import decomposerPrompt from '../prompts/decomposer';
  *
  * @example
  * const agent = new Agents("your-bearer-auth-token");
- * const response = await agent.SuperVisorAgent("What is the current price of the Sui token?");
+ * const response = await agent.processUserQueryPipeline("What is the current price of the Sui token?");
  * console.log(response);
  */
 class Agents {
@@ -41,7 +41,7 @@ class Agents {
    * @param prompt - User's input query
    * @returns IntentAgentResponse containing tool selection and processing details
    */
-  async IntentAgent(subqueries: string[], address?: string) {
+  async toolsSelector(subqueries: string[], address?: string) {
     const IntentResponse: IntentAgentResponse[] =
       (await this.tools.selectAppropriateTool(
         this.AtomaClass,
@@ -58,10 +58,7 @@ class Agents {
    * @param query - Original user query
    * @returns Processed response after decision making
    */
-  async DecisionMakingAgent(
-    intentResponse: IntentAgentResponse[],
-    query: string,
-  ) {
+  async QueryProcessor(intentResponse: IntentAgentResponse[], query: string) {
     // Pass both the selected tool name and arguments to processQuery
 
     return await this.utils.processQuery(
@@ -77,17 +74,17 @@ class Agents {
    * @param prompt - User's input query
    * @returns Final processed response
    */
-  async SuperVisorAgent(prompt: string, walletAddress?: string) {
+  async processUserQueryPipeline(prompt: string, walletAddress?: string) {
     // Process intent
     const decomposer = await this.QueryDecomposer(prompt);
     const decomposed: string[] = JSON.parse(
       decomposer.choices[0].message.content,
     );
     console.log(decomposed);
-    const res = await this.IntentAgent(decomposed, walletAddress);
+    const res = await this.toolsSelector(decomposed, walletAddress);
     console.log(res, 'this is intent agent response');
     // Make decision based on intent
-    const finalAnswer = await this.DecisionMakingAgent(res, prompt);
+    const finalAnswer = await this.QueryProcessor(res, prompt);
     console.log('Final Answer:', finalAnswer);
     return finalAnswer;
   }
diff --git a/packages/sui-agent/src/prompts/tool_selector.ts b/packages/sui-agent/src/prompts/tool_selector.ts
diff --git a/packages/sui-agent/src/protocols/aftermath/README.md b/packages/sui-agent/src/protocols/aftermath/README.md
@@ -32,20 +32,22 @@ The Sui Agent includes comprehensive integration with the Aftermath Protocol, pr
 const agent = new Agents(YOUR_BEARER_TOKEN);
 
 // Get pool information
-const poolInfo = await agent.SuperVisorAgent(
+const poolInfo = await agent.processUserQueryPipeline(
   'Get information about Aftermath pool 0x123...abc',
 );
 
 // Get APR for a token
-const apr = await agent.SuperVisorAgent('Get APR for token 0x456...def');
+const apr = await agent.processUserQueryPipeline(
+  'Get APR for token 0x456...def',
+);
 
 // Execute a multi-pool deposit
-const deposit = await agent.SuperVisorAgent(
+const deposit = await agent.processUserQueryPipeline(
   'Deposit 100 SUI into top 3 Aftermath pools by APR with 1% slippage',
 );
 
 // Check staking positions
-const stakingInfo = await agent.SuperVisorAgent(
+const stakingInfo = await agent.processUserQueryPipeline(
   'Get my staking positions for address 0x789...ghi',
 );
 ```
diff --git a/packages/sui-agent/src/protocols/cetus/README.md b/packages/sui-agent/src/protocols/cetus/README.md
@@ -31,22 +31,22 @@ The Sui Agent includes comprehensive integration with the Cetus Protocol, provid
 const agent = new Agents(YOUR_BEARER_TOKEN);
 
 // Get pool information
-const poolInfo = await agent.SuperVisorAgent(
+const poolInfo = await agent.processUserQueryPipeline(
   'Get information about Cetus pool 0x123...abc',
 );
 
 // Get positions for an address
-const positions = await agent.SuperVisorAgent(
+const positions = await agent.processUserQueryPipeline(
   'Get all Cetus positions for address 0x456...def',
 );
 
 // Calculate and execute a swap
-const swap = await agent.SuperVisorAgent(
+const swap = await agent.processUserQueryPipeline(
   'Swap 1 SUI for USDC on Cetus pool 0x123...abc with 1% slippage',
 );
 
 // Add liquidity
-const addLiquidity = await agent.SuperVisorAgent(
+const addLiquidity = await agent.processUserQueryPipeline(
   'Add 100 USDC as liquidity to Cetus pool 0x123...abc with 0.5% slippage',
 );
 ```
diff --git a/packages/sui-agent/src/tests/cetus.test.ts b/packages/sui-agent/src/tests/cetus.test.ts
@@ -8,7 +8,7 @@ const agent = new Agents(process.env.BEARER_TOKEN || '');
 describe('Cetus Protocol Integration Tests', () => {
   describe('Pool Operations', () => {
     test('should get pool information', async () => {
-      const response = await agent.SuperVisorAgent(
+      const response = await agent.processUserQueryPipeline(
         `Get information about Cetus pool ${TEST_POOLS.SUI_USDC}`,
       );
       expect(response).toBeDefined();
@@ -19,7 +19,7 @@ describe('Cetus Protocol Integration Tests', () => {
     });
 
     test('should get pool statistics and TVL', async () => {
-      const response = await agent.SuperVisorAgent(
+      const response = await agent.processUserQueryPipeline(
         `Get statistics and TVL for Cetus pool ${TEST_POOLS.USDC_USDT}`,
       );
       expect(response).toBeDefined();
@@ -29,7 +29,7 @@ describe('Cetus Protocol Integration Tests', () => {
     });
 
     test('should list all pools', async () => {
-      const response = await agent.SuperVisorAgent(
+      const response = await agent.processUserQueryPipeline(
         'List all available Cetus pools',
       );
       expect(response).toBeDefined();
@@ -41,7 +41,7 @@ describe('Cetus Protocol Integration Tests', () => {
 
   describe('Position Operations', () => {
     test('should get positions for address', async () => {
-      const response = await agent.SuperVisorAgent(
+      const response = await agent.processUserQueryPipeline(
         `Get all Cetus positions for address ${TEST_ADDRESSES.LP_PROVIDER}`,
       );
       expect(response).toBeDefined();
@@ -50,7 +50,7 @@ describe('Cetus Protocol Integration Tests', () => {
     });
 
     test('should calculate add liquidity parameters', async () => {
-      const response = await agent.SuperVisorAgent(
+      const response = await agent.processUserQueryPipeline(
         `Calculate parameters for adding 1000 USDC to Cetus pool ${TEST_POOLS.USDC_USDT} with ${SLIPPAGE.MEDIUM * 100}% slippage`,
       );
       expect(response).toBeDefined();
@@ -63,12 +63,12 @@ describe('Cetus Protocol Integration Tests', () => {
 
     test('should calculate remove liquidity parameters', async () => {
       // First get a position
-      const positionsResponse = await agent.SuperVisorAgent(
+      const positionsResponse = await agent.processUserQueryPipeline(
         `Get all Cetus positions for address ${TEST_ADDRESSES.LP_PROVIDER}`,
       );
       const position = positionsResponse.response[0];
 
-      const response = await agent.SuperVisorAgent(
+      const response = await agent.processUserQueryPipeline(
         `Calculate parameters for removing liquidity from Cetus position ${position.pos_object_id} with ${SLIPPAGE.LOW * 100}% slippage`,
       );
       expect(response).toBeDefined();
@@ -80,7 +80,7 @@ describe('Cetus Protocol Integration Tests', () => {
 
   describe('Trading Operations', () => {
     test('should calculate swap quote for exact input', async () => {
-      const response = await agent.SuperVisorAgent(
+      const response = await agent.processUserQueryPipeline(
         `Calculate swap quote for trading 1 SUI to USDC on Cetus pool ${TEST_POOLS.SUI_USDC} with ${SLIPPAGE.HIGH * 100}% slippage`,
       );
       expect(response).toBeDefined();
@@ -90,7 +90,7 @@ describe('Cetus Protocol Integration Tests', () => {
     });
 
     test('should calculate swap quote for exact output', async () => {
-      const response = await agent.SuperVisorAgent(
+      const response = await agent.processUserQueryPipeline(
         `Calculate swap quote to receive exactly 100 USDC by trading SUI on Cetus pool ${TEST_POOLS.SUI_USDC} with ${SLIPPAGE.MEDIUM * 100}% slippage`,
       );
       expect(response).toBeDefined();
@@ -100,7 +100,7 @@ describe('Cetus Protocol Integration Tests', () => {
     });
 
     test('should handle price impact calculation', async () => {
-      const response = await agent.SuperVisorAgent(
+      const response = await agent.processUserQueryPipeline(
         `Calculate price impact for swapping 1000 USDC to USDT on Cetus pool ${TEST_POOLS.USDC_USDT}`,
       );
       expect(response).toBeDefined();
@@ -114,7 +114,7 @@ describe('Cetus Protocol Integration Tests', () => {
 
   describe('Pool Creation', () => {
     test('should validate pool creation parameters', async () => {
-      const response = await agent.SuperVisorAgent(
+      const response = await agent.processUserQueryPipeline(
         'Create a new Cetus pool for SUI/USDC with tick spacing 2 and initial price 1.0',
       );
       expect(response).toBeDefined();
@@ -126,7 +126,7 @@ describe('Cetus Protocol Integration Tests', () => {
 
   describe('Error Handling', () => {
     test('should handle invalid pool ID gracefully', async () => {
-      const response = await agent.SuperVisorAgent(
+      const response = await agent.processUserQueryPipeline(
         'Get information about Cetus pool 0xinvalid_pool_id',
       );
       expect(response).toBeDefined();
@@ -135,7 +135,7 @@ describe('Cetus Protocol Integration Tests', () => {
     });
 
     test('should handle invalid address gracefully', async () => {
-      const response = await agent.SuperVisorAgent(
+      const response = await agent.processUserQueryPipeline(
         'Get all Cetus positions for address 0xinvalid_address',
       );
       expect(response).toBeDefined();
@@ -144,7 +144,7 @@ describe('Cetus Protocol Integration Tests', () => {
     });
 
     test('should handle excessive slippage values', async () => {
-      const response = await agent.SuperVisorAgent(
+      const response = await agent.processUserQueryPipeline(
         `Calculate swap quote for trading 1 SUI to USDC on Cetus pool ${TEST_POOLS.SUI_USDC} with 50% slippage`,
       );
       expect(response).toBeDefined();

Original file line number	Diff line number	Diff line change
`@@ -39,7 +39,7 @@ class ConversationController {`
`39`	`39`	`res.status(400).json({ error: 'Invalid conversation ID' });`
`40`	`40`	`return;`
`41`	`41`	`}`
`42`		`- const result = await suiAgent.SuperVisorAgent(message, walletAddress);`
	`42`	`+ const result = await suiAgent.processUserQueryPipeline(message, walletAddress);`
`43`	`43`	`const newMessage = await this.messageService.createMessage({`
`44`	`44`	`sender: sender ? sender : 'user',`
`45`	`45`	`walletAddress,`
Original file line number	Diff line number	Diff line change
`@@ -20,7 +20,7 @@ const handleQuery = async (req: Request, res: Response): Promise<void> => {`
`20`	`20`	`return;`
`21`	`21`	`}`
`22`	`22`	`// Get agent response first`
`23`		`- const result = await suiAgent.SuperVisorAgent(query);`
	`23`	`+ const result = await suiAgent.processUserQueryPipeline(query);`
`24`	`24`	`res.status(200).json(result);`
`25`	`25`	`} catch (error) {`
`26`	`26`	`console.error('Error handling query:', error);`