fix issues for gemini working with mcps

Author: daltonnyx

AgentCrew/__init__.py

@@ -1 +1 @@
-__version__ = "0.6.1-1"
+__version__ = "0.6.1-2"

AgentCrew/main.py

@@ -216,12 +216,12 @@ def run_update_command():
 
         if system == "linux" or system == "darwin":  # Darwin is macOS
             # Linux/macOS update command
-            command = "uv tool upgrade agentcrew-ai"
+            command = "uv tool install --reinstall --python=3.12 agentcrew-ai@latest"
             click.echo("🐧 Running Linux/macOS update command...")
 
         elif system == "windows":
             # Windows update command
-            command = "uv tool upgrade agentcrew-ai"
+            command = "uv tool install --reinstall --python=3.12 agentcrew-ai@latest"
             click.echo("🪟 Running Windows update command...")
 
         else:

AgentCrew/modules/agents/local_agent.py

@@ -1,4 +1,5 @@
 from datetime import datetime
+import os
 from typing import Dict, Any, List, Optional
 from AgentCrew.modules.llm.base import BaseLLMService
 from AgentCrew.modules.llm.message import MessageTransformer
@@ -176,7 +177,7 @@ def set_system_prompt(self, prompt: str):
         """
         self.system_prompt = prompt.replace(
             "{current_date}", datetime.today().strftime("%A, %d/%m/%Y")
-        )
+        ).replace("{cwd}", os.getcwd())
 
     def set_custom_system_prompt(self, prompt: str):
         """

AgentCrew/modules/google/native_service.py

@@ -255,11 +255,21 @@ def _build_schema(self, param_def):
             type=types.Type(param_type),
             description=param_def.get("description", None),
         )
+        if "const" in param_def.keys():
+            schema.default = param_def.get("const", None)
+
+        if "enum" in param_def.keys():
+            schema.enum = param_def.get("enum", [])
+
+        if "anyOf" in param_def.keys():
+            schema.any_of = [
+                self._build_schema(item) for item in param_def.get("anyOf", [])
+            ]
         if param_type == "OBJECT":
             schema.properties = {}
-            if "properties" in param_def:
+            if "properties" in param_def.keys():
                 for key in param_def.get("properties", {}):
-                    prop = param_def.get("properties").get("key", {})
+                    prop = param_def.get("properties").get(key, {})
                     schema.properties[key] = self._build_schema(prop)
         elif param_type == "ARRAY":
             itemsSchema = self._build_schema(param_def.get("items"))
@@ -293,10 +303,12 @@ def register_tool(self, tool_definition, handler_function):
                 tool_definition["function"].get("parameters", {}).get("required", [])
             )
             description = tool_definition["function"].get("description", "")
+            defs = tool_definition["function"].get("parameters", {}).get("$defs", {})
         else:
             parameters = tool_definition.get("parameters", {}).get("properties", {})
             required = tool_definition.get("parameters", {}).get("required", [])
             description = tool_definition.get("description", "")
+            defs = tool_definition.get("parameters", {}).get("$defs", {})
 
         # Create a function declaration for Google GenAI
         function_declaration = types.FunctionDeclaration(
@@ -315,6 +327,14 @@ def register_tool(self, tool_definition, handler_function):
                 function_declaration.parameters is not None
                 and function_declaration.parameters.properties is not None
             ):
+                if param_def.get("$ref", ""):
+                    ref_key = param_def["$ref"].lstrip("#/$defs/")
+                    if ref_key in defs.keys():
+                        function_declaration.parameters.properties[param_name] = (
+                            self._build_schema(defs[ref_key])
+                        )
+                        continue
+
                 function_declaration.parameters.properties[param_name] = (
                     self._build_schema(param_def)
                 )

AgentCrew/modules/gui/components/chat_components.py

@@ -143,7 +143,7 @@ def append_message(self, text, is_user=True, message_index=None, agent_name=None
             )
         if is_user:
             container_layout.addStretch(1)  # Push to left
-            container_layout.addWidget(message_bubble, 1)
+            container_layout.addWidget(message_bubble, 2)
         else:
             # container_layout.addStretch(1)  # Push to right
             container_layout.addWidget(message_bubble)

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agentcrew-ai"
-version = "0.6.1-1"
+version = "0.6.1-2"
 requires-python = ">=3.12"
 classifiers = [
     "Programming Language :: Python :: 3",