Feat : MCP agent

sources/agents/__init__.py

@@ -5,5 +5,6 @@ from .casual_agent import CasualAgent
 from .file_agent import FileAgent
 from .planner_agent import PlannerAgent
 from .browser_agent import BrowserAgent
+from .mcp_agent import McpAgent
 
-__all__ = ["Agent", "CoderAgent", "CasualAgent", "FileAgent", "PlannerAgent", "BrowserAgent"]
+__all__ = ["Agent", "CoderAgent", "CasualAgent", "FileAgent", "PlannerAgent", "BrowserAgent", "McpAgent"]

sources/agents/agent.py

@@ -90,6 +90,12 @@ class Agent():
             raise TypeError("Tool must be a callable object (a method)")
         self.tools[name] = tool
     
+    def get_tools_name(self) -> list:
+        """
+        Get the list of tools names.
+        """
+        return list(self.tools.keys())
+    
     def load_prompt(self, file_path: str) -> str:
         try:
             with open(file_path, 'r', encoding="utf-8") as f:
@@ -235,11 +241,13 @@ class Agent():
             answer = "I will execute:\n" + answer # there should always be a text before blocks for the function that display answer
 
         self.success = True
+        self.blocks_result = []
         for name, tool in self.tools.items():
             feedback = ""
             blocks, save_path = tool.load_exec_block(answer)
 
             if blocks != None:
+                pretty_print(f"Executing {len(blocks)} {name} blocks...", color="status")
                 for block in blocks:
                     self.show_block(block)
                     output = tool.execute([block])

sources/agents/code_agent.py

@@ -62,14 +62,14 @@ class CoderAgent(Agent):
             animate_thinking("Executing code...", color="status")
             self.status_message = "Executing code..."
             self.logger.info(f"Attempt {attempt + 1}:\n{answer}")
-            exec_success, _ = self.execute_modules(answer)
+            exec_success, feedback = self.execute_modules(answer)
             self.logger.info(f"Execution result: {exec_success}")
             answer = self.remove_blocks(answer)
             self.last_answer = answer
             await asyncio.sleep(0)
             if exec_success and self.get_last_tool_type() != "bash":
                 break
-            pretty_print("Execution failure", color="failure")
+            pretty_print(f"Execution failure:\n{feedback}", color="failure")
             pretty_print("Correcting code...", color="status")
             self.status_message = "Correcting code..."
             attempt += 1

sources/agents/mcp_agent.py

@@ -0,0 +1,70 @@
+import os
+import asyncio
+
+from sources.utility import pretty_print, animate_thinking
+from sources.agents.agent import Agent
+from sources.tools.mcpFinder import MCP_finder
+
+# NOTE MCP agent is an active work in progress, not functional yet.
+
+class McpAgent(Agent):
+
+    def __init__(self, name, prompt_path, provider, verbose=False):
+        """
+        The mcp agent is a special agent for using MCPs.
+        MCP agent will be disabled if the user does not explicitly set the MCP_FINDER_API_KEY in environment variable.
+        """
+        super().__init__(name, prompt_path, provider, verbose, None)
+        keys = self.get_api_keys()
+        self.tools = {
+            "mcp_finder": MCP_finder(keys["mcp_finder"]),
+            # add mcp tools here
+        }
+        self.role = "mcp"
+        self.type = "mcp_agent"
+        self.enabled = True
+    
+    def get_api_keys(self) -> dict:
+        """
+        Returns the API keys for the tools.
+        """
+        api_key_mcp_finder = os.getenv("MCP_FINDER_API_KEY")
+        if not api_key_mcp_finder or api_key_mcp_finder == "":
+            pretty_print("MCP Finder API key not found. Please set the MCP_FINDER_API_KEY environment variable.", color="failure")
+            pretty_print("MCP Finder disabled.", color="failure")
+            self.enabled = False
+        return {
+            "mcp_finder": api_key_mcp_finder
+        }
+    
+    def expand_prompt(self, prompt):
+        """
+        Expands the prompt with the tools available.
+        """
+        tools_name = self.get_tools_name()
+        tools_str = ", ".join(tools_name)
+        prompt += f"""
+        You can use the following tools and MCPs:
+        {tools_str}
+        """
+        return prompt
+    
+    async def process(self, prompt, speech_module) -> str:
+        if self.enabled == False:
+            return "MCP Agent is disabled."
+        prompt = self.expand_prompt(prompt)
+        self.memory.push('user', prompt)
+        working = True
+        while working == True:
+            animate_thinking("Thinking...", color="status")
+            answer, reasoning = await self.llm_request()
+            exec_success, _ = self.execute_modules(answer)
+            answer = self.remove_blocks(answer)
+            self.last_answer = answer
+            self.status_message = "Ready"
+            if len(self.blocks_result) == 0:
+                working = False
+        return answer, reasoning
+
+if __name__ == "__main__":
+    pass