Update client

Author: fishonamos

apps/client/app/page.tsx

@@ -4,7 +4,6 @@ import api from './lib/api';
 import { useWallet } from '@suiet/wallet-kit';
 import JSONFormatter from './utils/JSONFormatter';
 
-import { keywords } from './data';
 
 import Messages from './components/sections/Messages';
 import SampleQuestions from './components/sections/SampleQuestions';
@@ -21,8 +20,8 @@ export default function Home() {
   useEffect(() => {
     if (address && connected) {
       loadChatHistory();
-      // Send initial wallet connection message
-      handleSend(`Connected wallet: ${address}`);
+      // // Send initial wallet connection message
+      // handleSend(`Connected wallet: ${address}`);
     }
   }, [address, connected]);
 

apps/web/src/app.ts

@@ -2,7 +2,7 @@ import express, { Application } from 'express';
 import cors from 'cors';
 import v1routes from './routes/v1';
 import { connectDB } from './utils/db';
-import queryRouter from './routes/v1/query';
+//import queryRouter from './routes/v1/query';
 
 /**
  * Express application instance.

apps/web/src/routes/v1/query.ts

@@ -15,17 +15,19 @@ queryRouter.get('/health', (req: Request, res: Response) => {
 // Query endpoint
 const handleQuery = async (req: Request, res: Response): Promise<void> => {
   try {
+    
     const { query, walletAddress } = req.body;
+   
 
     if (!query) {
       res.status(400).json({
         error: 'Missing query in request body'
       });
+      
       return;
     }
-
     // Get agent response first
-    const result = await suiAgent.SuperVisorAgent(query);
+    const result = await suiAgent.SuperVisorAgent(query,walletAddress);
 
     // Only try to save chat history if walletAddress is provided
     if (walletAddress) {

apps/web/src/types/global.d.ts

@@ -1,20 +1,20 @@
-import mongoose from 'mongoose';
+// import mongoose from 'mongoose';
 
-declare global {
-  var _mongooseCache: {
-    conn: typeof mongoose | null;
-    promise: Promise<typeof mongoose> | null;
-  } | undefined;
-}
+// declare global {
+//   var _mongooseCache: {
+//     conn: typeof mongoose | null;
+//     promise: Promise<typeof mongoose> | null;
+//   } | undefined;
+// }
 
-// Augment the NodeJS namespace to include our custom property
-declare module 'node:process' {
-  global: typeof globalThis & {
-    _mongooseCache?: {
-      conn: typeof mongoose | null;
-      promise: Promise<typeof mongoose> | null;
-    };
-  };
-}
+// // Augment the NodeJS namespace to include our custom property
+// declare module 'node:process' {
+//   global: typeof globalThis & {
+//     _mongooseCache?: {
+//       conn: typeof mongoose | null;
+//       promise: Promise<typeof mongoose> | null;
+//     };
+//   };
+// }
 
-export {}; 
+// export {}; 

apps/web/src/utils/db.ts

@@ -6,56 +6,20 @@ if (!MONGODB_URI) {
   throw new Error('Please define the MONGODB_URI environment variable');
 }
 
-// Type the global object with our custom property
-const globalWithMongoose = global as typeof globalThis & {
-  _mongooseCache?: {
-    conn: typeof mongoose | null;
-    promise: Promise<typeof mongoose> | null;
-  };
-};
-
-// Initialize the cache if it doesn't exist
-if (!globalWithMongoose._mongooseCache) {
-  globalWithMongoose._mongooseCache = { conn: null, promise: null };
-}
-
 export async function connectDB() {
-  // At this point we know _mongooseCache exists because we initialized it above
-  const mongooseCache = globalWithMongoose._mongooseCache!;
-
-  if (mongooseCache.conn) {
-    return mongooseCache.conn;
-  }
-
-  if (!mongooseCache.promise) {
-    const opts = {
-      bufferCommands: false,
-    };
-
-    mongooseCache.promise = mongoose.connect(MONGODB_URI!, opts).then((mongoose) => {
-      return mongoose;
-    });
-  }
-
   try {
-    const mongoose = await mongooseCache.promise;
-    mongooseCache.conn = mongoose;
+    await mongoose.connect(MONGODB_URI|| '');
     console.log('Connected to MongoDB');
     return mongoose;
-  } catch (e) {
-    mongooseCache.promise = null;
-    console.error('MongoDB connection error:', e);
-    throw e;
+  } catch (error) {
+    console.error('MongoDB connection error:', error);
+    throw error;
   }
 }
 
 export async function disconnectDB() {
   try {
     await mongoose.disconnect();
-    // At this point we know _mongooseCache exists because we initialized it above
-    const mongooseCache = globalWithMongoose._mongooseCache!;
-    mongooseCache.conn = null;
-    mongooseCache.promise = null;
     console.log('Disconnected from MongoDB');
   } catch (error) {
     console.error('MongoDB disconnection error:', error);

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

@@ -30,9 +30,9 @@ class Agents {
    * @param prompt - User's input query
    * @returns IntentAgentResponse containing tool selection and processing details
    */
-  async IntentAgent(prompt: string) {
+  async IntentAgent(prompt: string,address?:string) {
     const IntentResponse: IntentAgentResponse =
-      (await this.tools.selectAppropriateTool(prompt)) as IntentAgentResponse;
+      (await this.tools.selectAppropriateTool(prompt,address)) as IntentAgentResponse;
 
     return IntentResponse;
   }
@@ -61,9 +61,9 @@ class Agents {
    * @param prompt - User's input query
    * @returns Final processed response
    */
-  async SuperVisorAgent(prompt: string) {
+  async SuperVisorAgent(prompt: string,walletAddress?:string) {
     // Process intent
-    const res = await this.IntentAgent(prompt);
+    const res = await this.IntentAgent(prompt, walletAddress);
     console.log('Intent Response:', res);
 
     // Make decision based on intent

packages/sui-agent/src/protocols/navi/index.ts

@@ -127,7 +127,7 @@ async function checkUserLiquidations(address: string) {
       },
     });
     const liquidations = response.data;
-    console.log(liquidations.result.rows, 'liquidations');
+
     return {
       asUser: liquidations.result.rows.filter(
         (l: Liquidation) => l.user === address,

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

@@ -44,16 +44,17 @@ class Tools {
    * @param query - User query
    * @returns Selected tool response or null if no tool found
    */
-  async selectAppropriateTool(query: string): Promise<toolResponse | null> {
-    console.log(this.getAllTools());
+  async selectAppropriateTool(query: string,walletAddress?:string): Promise<toolResponse | null> {
+
     const finalPrompt = this.prompt.replace(
       '${toolsList}',
       JSON.stringify(this.getAllTools()),
     );
 
+    
     const response = await atomaChat(this.sdk, [
       {
-        content: finalPrompt,
+        content: walletAddress?`${finalPrompt}.Wallet address is ${walletAddress}.`:finalPrompt,
         role: 'system',
       },
       {