Merge pull request #33 from RSS3-Network/fix/debug-transfer-widget

Fix: modify transfer widget and write automated test script

openagent/ui/app.py

@@ -149,10 +149,11 @@ async def handle_tool_end(event, msg):
         output = event["data"]["output"]
         transfer_dict = json.loads(output)
         token = transfer_dict["token"]
+        token_address = transfer_dict["token_address"]
         to_address = transfer_dict["to_address"]
         amount = transfer_dict["amount"]
 
-        url = f"/widget/transfer?token={token}&amount={amount}&toAddress={to_address}"
+        url = f"/widget/transfer?token={token}&tokenAddress={token_address}&amount={amount}&toAddress={to_address}"
 
         iframe_html = f"""
                 <iframe src="{url}" width="100%" height="600px" style="border:none;">
@@ -163,5 +164,5 @@ async def handle_tool_end(event, msg):
     if event["name"] == "PriceExecutor":
         output = event["data"]["output"]
         price_dict = json.loads(output)
-        widget = f"""<iframe src="/widget/price-chart?token={list(price_dict.keys())[0]}" height="400"></iframe>"""  # noqa
+        widget = f"""<iframe src="/widget/price-chart?token={list(price_dict.keys())[0]}" height="600px"></iframe>"""  # noqa
         await msg.stream_token(widget)

public/index.css

@@ -13,4 +13,4 @@ div.MuiStack-root.message-content .markdown-body {
 
 div.MuiBox-root div.MuiStack-root.watermark{
     display: none;
-}
+}

tests/agent_trajectory/asset_management.py

@@ -1,17 +1,30 @@
 import asyncio
+import os
+import sys
 import unittest
 
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "../..")))
+
 from langchain_core.messages import HumanMessage
 
 from openagent.agents.asset_management import asset_management_agent
 from openagent.conf.llm_provider import set_current_llm
-
-
-class TestAssetManagementAgent(unittest.TestCase):
-    def setUp(self):
-        set_current_llm("gemini-1.5-pro")
-        # set_current_llm("gpt-3.5-turbo")
-        # set_current_llm("llama3.1:latest")
+from tests.base_test import BaseAgentTest
+
+
+class TestAssetManagementAgent(BaseAgentTest):
+    # def setUp(self):
+    #     set_current_llm("gemini-1.5-pro")
+    #     # set_current_llm("gpt-3.5-turbo")
+    #     # set_current_llm("llama3.1:latest")
+    @classmethod
+    def setUpClass(cls):
+        super().setUpClass()
+        if len(sys.argv) > 2 and not sys.argv[2].startswith("-"):
+            set_current_llm(sys.argv[2])
+            del sys.argv[2]
+        else:
+            set_current_llm("default_model")
 
     def test_swap_eth_to_usdt(self):
         async def async_test():

tests/agent_trajectory/block_explore.py

@@ -1,17 +1,30 @@
 import asyncio
+import os
+import sys
 import unittest
 
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "../..")))
+
 from langchain_core.messages import HumanMessage
 
 from openagent.agents.block_explore import block_explorer_agent
 from openagent.conf.llm_provider import set_current_llm
-
-
-class TestBlockExploreAgent(unittest.TestCase):
-    def setUp(self):
-        # set_current_llm("gemini-1.5-pro")
-        set_current_llm("gpt-3.5-turbo")
-        # set_current_llm("llama3.1:latest")
+from tests.base_test import BaseAgentTest
+
+
+class TestBlockExploreAgent(BaseAgentTest):
+    # def setUp(self):
+    #     # set_current_llm("gemini-1.5-pro")
+    #     set_current_llm("gpt-3.5-turbo")
+    #     # set_current_llm("llama3.1:latest")
+    @classmethod
+    def setUpClass(cls):
+        super().setUpClass()
+        if len(sys.argv) > 2 and not sys.argv[2].startswith("-"):
+            set_current_llm(sys.argv[2])
+            del sys.argv[2]
+        else:
+            set_current_llm("default_model")
 
     def test_query_block_height(self):
         async def async_test():

tests/agent_trajectory/feed_explore.py

@@ -1,15 +1,28 @@
 import asyncio
+import os
+import sys
 import unittest
 
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "../..")))
+
 from langchain_core.messages import HumanMessage
 
 from openagent.agents.feed_explore import feed_explorer_agent
 from openagent.conf.llm_provider import set_current_llm
-
-
-class TestFeedExploreAgent(unittest.TestCase):
-    def setUp(self):
-        set_current_llm("gpt-3.5-turbo")
+from tests.base_test import BaseAgentTest
+
+
+class TestFeedExploreAgent(BaseAgentTest):
+    # def setUp(self):
+    #     set_current_llm("gpt-3.5-turbo")
+    @classmethod
+    def setUpClass(cls):
+        super().setUpClass()
+        if len(sys.argv) > 2 and not sys.argv[2].startswith("-"):
+            set_current_llm(sys.argv[2])
+            del sys.argv[2]
+        else:
+            set_current_llm("default_model")
 
     def test_query_defi_activities(self):
         """

tests/agent_trajectory/market_analysis.py

@@ -1,17 +1,30 @@
 import asyncio
+import os
+import sys
 import unittest
 
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "../..")))
+
 from langchain_core.messages import HumanMessage
 
 from openagent.agents.market_analysis import market_analysis_agent
 from openagent.conf.llm_provider import set_current_llm
-
-
-class TestMarketAnalysisAgent(unittest.TestCase):
-    def setUp(self):
-        # set_current_llm("gemini-1.5-pro")
-        set_current_llm("gpt-3.5-turbo")
-        # set_current_llm("llama3.1:latest")
+from tests.base_test import BaseAgentTest
+
+
+class TestMarketAnalysisAgent(BaseAgentTest):
+    # def setUp(self):
+    #     # set_current_llm("gemini-1.5-pro")
+    #     set_current_llm("gpt-3.5-turbo")
+    #     # set_current_llm("llama3.1:latest")
+    @classmethod
+    def setUpClass(cls):
+        super().setUpClass()
+        if len(sys.argv) > 2 and not sys.argv[2].startswith("-"):
+            set_current_llm(sys.argv[2])
+            del sys.argv[2]
+        else:
+            set_current_llm("default_model")
 
     def test_query_btc_price(self):
         async def async_test():

tests/agent_trajectory/research_analyst.py

@@ -1,17 +1,30 @@
 import asyncio
+import os
+import sys
 import unittest
 
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "../..")))
+
 from langchain_core.messages import HumanMessage
 
 from openagent.agents.project_management import research_analyst_agent
 from openagent.conf.llm_provider import set_current_llm
-
-
-class TestResearchAnalystAgent(unittest.TestCase):
-    def setUp(self):
-        # set_current_llm("gemini-1.5-pro")
-        set_current_llm("gpt-3.5-turbo")
-        # set_current_llm("llama3.1:latest")
+from tests.base_test import BaseAgentTest
+
+
+class TestResearchAnalystAgent(BaseAgentTest):
+    # def setUp(self):
+    #     # set_current_llm("gemini-1.5-pro")
+    #     set_current_llm("gpt-3.5-turbo")
+    #     # set_current_llm("llama3.1:latest")
+    @classmethod
+    def setUpClass(cls):
+        super().setUpClass()
+        if len(sys.argv) > 2 and not sys.argv[2].startswith("-"):
+            set_current_llm(sys.argv[2])
+            del sys.argv[2]
+        else:
+            set_current_llm("default_model")
 
     def test_query_project(self):
         async def async_test():

tests/base_test.py

@@ -0,0 +1,10 @@
+import unittest
+
+
+class BaseAgentTest(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        pass
+
+    def setUp(self):
+        pass

tests/run_tests.py

@@ -0,0 +1,39 @@
+import sys
+import unittest
+
+from openagent.conf.llm_provider import set_current_llm
+
+
+def run_tests(model_name, specific_test=None):
+    # Set the model name globally
+    set_current_llm(model_name)
+
+    # Discover and run all tests
+    test_loader = unittest.TestLoader()
+
+    if specific_test:
+        specific_test = specific_test.replace("tests.", "", 1)
+        test_suite = test_loader.loadTestsFromName(specific_test)
+    else:
+        test_suite = test_loader.discover("agent_trajectory", pattern="*.py")
+
+    runner = unittest.TextTestRunner(verbosity=2)
+    result = runner.run(test_suite)
+
+    return result
+
+
+if __name__ == "__main__":
+    if len(sys.argv) < 2:
+        print("Usage: python run_tests.py <model_name> OR python run_tests.py <model_name> [specific_test] ")
+        sys.exit(1)
+
+    model_name = sys.argv[1]
+    specific_test = sys.argv[2] if len(sys.argv) > 2 else None
+
+    result = run_tests(model_name, specific_test)
+
+    if result.wasSuccessful():
+        sys.exit(0)
+    else:
+        sys.exit(1)

widget/src/App.tsx

@@ -1,26 +1,41 @@
 import React from 'react';
 import '@rainbow-me/rainbowkit/styles.css';
-import { RainbowKitProvider, getDefaultConfig } from '@rainbow-me/rainbowkit';
-import { WagmiProvider } from 'wagmi';
+import {RainbowKitProvider, getDefaultConfig, getDefaultWallets} from '@rainbow-me/rainbowkit';
+import {createConfig, http, WagmiProvider} from 'wagmi';
 import { QueryClient, QueryClientProvider } from '@tanstack/react-query';
-import { mainnet, polygon, optimism, arbitrum } from 'wagmi/chains';
+import {mainnet, sepolia} from 'wagmi/chains';
 
-/* eslint-disable no-console */
+let MAINNET_RPC_URL = process.env.REACT_APP_MAINNET_RPC_URL || "";
+let SEPOLIA_RPC_URL = process.env.REACT_APP_SEPOLIA_RPC_URL || "";
 
+/* eslint-disable no-console */
 export function App() {
-
   return <>
-  <h1>hello world</h1>
   </>
 }
 
-// Configuration for RainbowKit
-const config = getDefaultConfig({
-  appName: 'My RainbowKit App',
-  projectId: 'No_ID',
-    chains: [mainnet, polygon, optimism, arbitrum],
+const { connectors } = getDefaultWallets({
+  appName: "OpenAgent Widget",
+  appDescription: "Widget",
+  projectId: "project id",
+})
+
+export const wagmiConfig = createConfig({
+  chains:[mainnet,sepolia],
+  connectors,
   ssr: true,
-});
+  transports: {
+    [mainnet.id]: http(MAINNET_RPC_URL, { batch: true }),
+    [sepolia.id]: http(SEPOLIA_RPC_URL, { batch: true }),
+  },
+  syncConnectedChain: true,
+})
+
+declare module "wagmi" {
+  interface Register {
+    config: typeof wagmiConfig
+  }
+}
 
 // Create a new QueryClient instance for React Query
 const queryClient = new QueryClient();
@@ -34,7 +49,7 @@ const queryClient = new QueryClient();
  */
 export function TransferWidgetApp({ children }: { children: React.ReactNode }) {
   return (
-      <WagmiProvider config={config}>
+      <WagmiProvider config={wagmiConfig}>
       <QueryClientProvider client={queryClient}>
         <RainbowKitProvider>
           {children}