Merge pull request #26 from RSS3-Network/feat/executor_optimization

chore: experts rename

Author: naaive

src/.env.example

@@ -39,7 +39,7 @@ LLM_API_BASE=http://ollama:11434
 
 ## A database is required for vector and user management.
 ## Database connection string
-DB_CONNECTION=postgresql+psycopg://postgres:password@vec_db:5434/vec
+DB_CONNECTION=postgresql+psycopg://postgres:password@vector_db:5434/vec
 
 # End of basic configuration
 ############################

src/docker-compose.yml

@@ -12,7 +12,7 @@ services:
     networks:
       - openagent-network
 
-  vec_db:
+  vector_db:
     image: pgvector/pgvector:pg16
     container_name: vec_db
     restart: unless-stopped

src/openagent/experts/article_expert.py

@@ -18,7 +18,7 @@ class ARGS(BaseModel):
 
 
 class ArticleExpert(BaseTool):
-    name = "article"
+    name = "ArticleExecutor"
     description = (
         "A tool for searching web3-related articles. If you lack knowledge about web3, "
         "you can use this tool to find relevant articles that can help answer "

src/openagent/experts/feed_expert.py

@@ -21,12 +21,12 @@ class ParamSchema(BaseModel):
 
     type: str = Field(
         description="""Retrieve activities for the specified type,
-for example: all, post, comment, share."""
+eg. : all, post, comment, share."""
     )
 
 
 class FeedExpert(BaseTool):
-    name = "feed"
+    name = "FeedExecutor"
     description = """Use this tool to get the activities of a wallet address or \
 blockchain domain name and know what this address has done or doing recently."""
     args_schema: Type[ParamSchema] = ParamSchema
@@ -50,7 +50,7 @@ async def _arun(
 
 async def fetch_feeds(address: str, type: str):
     url = f"{settings.RSS3_DATA_API}/decentralized/{address}?limit=5&action_limit=10&tag=social"
-    if type != "all":
+    if type in ["post", "comment", "share"]:
         url += f"&type={type}"
     headers = {"Accept": "application/json"}
     async with aiohttp.ClientSession() as session:

src/openagent/experts/nft_expert.py

@@ -32,7 +32,7 @@ class ARGS(BaseModel):
 
 
 class NFTExpert(BaseTool):
-    name = "NFT"
+    name = "NFTExecutor"
     description = "A tool for searching NFT collections or getting collection rankings."
     args_schema: Type[ARGS] = ARGS
 

src/openagent/experts/price_expert.py

@@ -15,7 +15,7 @@ class ARGS(BaseModel):
 
 
 class PriceExpert(BaseTool):
-    name = "price"
+    name = "PriceExecutor"
     description = "use this tool to get the price of a token."
     args_schema: Type[ARGS] = ARGS
 

src/openagent/experts/search_expert.py

@@ -37,7 +37,7 @@ async def google_search(query: str) -> str:
 
 
 class SearchExpert(BaseTool):
-    name = "search"
+    name = "SearchExecutor"
     description = """
     A versatile search tool that can perform various types of searches based on the query type:
     - For queries related to charts, data visualization, or dashboards, use Dune search.

src/openagent/experts/swap_expert.py

@@ -31,7 +31,7 @@ class SwapExpert(BaseTool):
     Tool for generating a swap widget for cryptocurrency swaps.
     """
 
-    name = "swap"
+    name = "SwapExecutor"
     description = "Use this tool to generate a swap widget for the user to swap cryptocurrencies."
     args_schema: Type[ParamSchema] = ParamSchema
     return_direct = False

src/openagent/experts/transfer_expert.py

@@ -34,7 +34,7 @@ class ParamSchema(BaseModel):
 
 
 class TransferExpert(BaseTool):
-    name = "transfer"
+    name = "TransferExecutor"
     description = """Use this tool to send cryptocurrencies to another address."""
     args_schema: Type[ParamSchema] = ParamSchema
     return_direct = False

src/openagent/ui/app.py

@@ -78,7 +78,7 @@ def build_token(token_symbol: str, token_address: str):
 
 async def handle_function_message(message: FunctionMessage, msg: cl.Message):
     """Handle FunctionMessage type of messages."""
-    if message.name == "swap":
+    if message.name == "SwapExecutor":
         swap_dict = json.loads(message.content)
         await do_stream_swap_widget(msg, swap_dict)
 
@@ -130,7 +130,7 @@ async def on_message(message: cl.Message):
 async def react_tool_call_handle(message, msg):
     try:
         swap_dict = parse_json_markdown(message.content)
-        if "type" in swap_dict and swap_dict["type"] == "swap":
+        if "type" in swap_dict and swap_dict["type"] == "SwapExecutor":
             await do_stream_swap_widget(msg, swap_dict)
     except Exception as e:
         logger.warning("Failed to handle react tool call message", e)