fix bug related to copilot cannot see image

Author: daltonnyx

AgentCrew/__init__.py

@@ -1 +1 @@
-__version__ = "0.6.2"
+__version__ = "0.6.3"

AgentCrew/main.py

@@ -36,11 +36,17 @@ def cli():
 
 
 def cli_prod():
+    from AgentCrew.modules import FileLogIO
+
+    sys.stderr = FileLogIO()
+
+    os.environ["AGENTCREW_LOG_PATH"] = os.path.expanduser("~/.AgentCrew/logs")
     os.environ["MEMORYDB_PATH"] = os.path.expanduser("~/.AgentCrew/memorydb")
     os.environ["MCP_CONFIG_PATH"] = os.path.expanduser("~/.AgentCrew/mcp_servers.json")
     os.environ["SW_AGENTS_CONFIG"] = os.path.expanduser("~/.AgentCrew/agents.toml")
     os.environ["PERSISTENCE_DIR"] = os.path.expanduser("~/.AgentCrew/persistents")
     os.environ["AGENTCREW_CONFIG_PATH"] = os.path.expanduser("~/.AgentCrew/config.json")
+
     check_and_update()
     cli()  # Delegate to main CLI function
 

AgentCrew/modules/__init__.py

@@ -1,5 +1,10 @@
 import logging
 import sys
+import tempfile
+from datetime import datetime
+import os
+
+from typing import TextIO, AnyStr
 
 LOG_LEVEL = logging.ERROR
 # Create a logger
@@ -22,3 +27,38 @@
 
 # Optional: Prevent duplicate logging if this module is imported multiple times
 logger.propagate = False
+
+
+class FileLogIO(TextIO):
+    """File-like object compatible with sys.stderr for MCP logging."""
+
+    def __init__(self, file_format: str = "agentcrew"):
+        log_dir_path = os.getenv("AGENTCREW_LOG_PATH", tempfile.gettempdir())
+        os.makedirs(log_dir_path, exist_ok=True)
+        self.log_path = (
+            log_dir_path + f"/{file_format}_{datetime.now().timestamp()}.log"
+        )
+        self.file = open(self.log_path, "w+")
+
+    def write(self, data: AnyStr) -> int:
+        """Write data to the log file."""
+        if isinstance(data, bytes):
+            # Convert bytes to string for writing
+            str_data = data.decode("utf-8", errors="replace")
+        else:
+            str_data = str(data)
+        self.file.write(str_data)
+        self.file.flush()  # Ensure data is written immediately
+        return 0
+
+    def flush(self):
+        """Flush the file buffer."""
+        self.file.flush()
+
+    def close(self):
+        """Close the file."""
+        self.file.close()
+
+    def fileno(self):
+        """Return the file descriptor."""
+        return self.file.fileno()

AgentCrew/modules/chat/console_ui.py

@@ -1103,7 +1103,16 @@ def _(event):
             else:
                 self._last_ctrl_c_time = current_time
                 if self.live:
-                    self.message_handler.stop_streaming = True
+                    if self.message_handler.stream_generator:
+                        try:
+                            asyncio.run(self.message_handler.stream_generator.aclose())
+                        except RuntimeError as e:
+                            logger.warning(f"Error closing stream generator: {e}")
+                        except Exception as e:
+                            logger.warning(f"Exception closing stream generator: {e}")
+                        finally:
+                            self.message_handler.stop_streaming = True
+                            self.message_handler.stream_generator = None
 
                 self.console.print(
                     Text(

AgentCrew/modules/chat/message/command_processor.py

@@ -1,5 +1,5 @@
 from dataclasses import dataclass
-from typing import Optional, Tuple, List, Dict, Any, Union
+from typing import Optional, Tuple, List, Dict
 import os
 
 from AgentCrew.modules.agents.local_agent import LocalAgent

AgentCrew/modules/custom_llm/github_copilot_service.py

@@ -97,7 +97,9 @@ def format_tool_result(
                 parsed_tool_result = []
                 for res in tool_result:
                     if res.get("type", "text") == "image_url":
-                        if "vision" in ModelRegistry.get_model_capabilities(self.model):
+                        if "vision" in ModelRegistry.get_model_capabilities(
+                            f"{self._provider_name}/{self.model}"
+                        ):
                             parsed_tool_result.append(res)
                     else:
                         parsed_tool_result.append(res)
@@ -170,7 +172,9 @@ async def stream_assistant_response(self, messages):
                     )
                     > 0
                 ):
-                    if "vision" in ModelRegistry.get_model_capabilities(self.model):
+                    if "vision" in ModelRegistry.get_model_capabilities(
+                        f"{self._provider_name}/{self.model}"
+                    ):
                         self.extra_headers["Copilot-Vision-Request"] = "true"
 
                 # if self._interaction_id:

AgentCrew/modules/llm/service_manager.py

@@ -138,9 +138,11 @@ def get_service(self, provider: str) -> BaseLLMService:
                     .endswith(".githubcopilot.com")
                 ):
                     # Special case for OpenAI compatible custom providers
-                    return GithubCopilotService(api_key=api_key, provider_name=provider)
+                    service_instance = GithubCopilotService(
+                        api_key=api_key, provider_name=provider
+                    )
                 else:
-                    return CustomLLMService(
+                    service_instance = CustomLLMService(
                         base_url=details["api_base_url"],
                         api_key=api_key,
                         provider_name=provider,

AgentCrew/modules/mcpclient/service.py

@@ -1,5 +1,5 @@
 from AgentCrew.modules import logger
-from typing import Dict, Any, List, Optional, Callable, TextIO, AnyStr
+from typing import Dict, Any, List, Optional, Callable
 from mcp import ClientSession, StdioServerParameters
 from mcp.types import Prompt, ContentBlock, TextContent, ImageContent
 from mcp.client.stdio import stdio_client
@@ -8,46 +8,10 @@
 from .config import MCPServerConfig
 import asyncio
 import threading
-import tempfile
-from datetime import datetime
-
-
-class MCPLogIO(TextIO):
-    """File-like object compatible with sys.stderr for MCP logging."""
-
-    def __init__(self):
-        self.log_path = (
-            tempfile.gettempdir() + f"/mcp_agentcrew_{datetime.now().timestamp()}.log"
-        )
-        print(f"Routing MCP logs to {self.log_path}")
-        self.file = open(self.log_path, "w+")
-
-    def write(self, data: AnyStr) -> int:
-        """Write data to the log file."""
-        if isinstance(data, bytes):
-            # Convert bytes to string for writing
-            str_data = data.decode("utf-8", errors="replace")
-        else:
-            str_data = str(data)
-        self.file.write(str_data)
-        self.file.flush()  # Ensure data is written immediately
-        return 0
-
-    def flush(self):
-        """Flush the file buffer."""
-        self.file.flush()
-
-    def close(self):
-        """Close the file."""
-        self.file.close()
-
-    def fileno(self):
-        """Return the file descriptor."""
-        return self.file.fileno()
-
+from AgentCrew.modules import FileLogIO
 
 # Initialize the logger
-mcp_log_io = MCPLogIO()
+mcp_log_io = FileLogIO("mcpclient_agentcrew")
 
 
 class MCPService:

pyproject.toml

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