minor fixes

Author: lorenss-m

.gitignore

@@ -30,4 +30,6 @@ TODO.md
 
 *.log
 
-/ref/
+/ref/
+
+.cursor/

hud/datasets.py

@@ -143,7 +143,7 @@ async def run_dataset(
         ... )
     """
     # Import here to avoid circular imports
-    from mcp_use import MCPClient
+    from hud.mcp.client import MCPClient
 
     import hud
 

hud/mcp/base.py

@@ -448,7 +448,7 @@ async def run_conversation(self, prompt: str, max_steps: int = 10) -> AgentResul
                             logger.error("Tool execution failed: %s", e)
                             # Create error MCPToolResult
                             error_result = MCPToolResult(
-                                content=[types.TextContent(text=str(e))], isError=True
+                                content=[types.TextContent(text=str(e), type="text")], isError=True
                             )
                             tool_results.append(error_result)
 
@@ -535,7 +535,7 @@ async def _run_prompt(self, prompt: str, max_steps: int = 10) -> AgentResult:
                             logger.error("Tool execution failed: %s", e)
                             # Create error MCPToolResult
                             error_result = MCPToolResult(
-                                content=[types.TextContent(text=str(e))], isError=True
+                                content=[types.TextContent(text=str(e), type="text")], isError=True
                             )
                             tool_results.append(error_result)
 

hud/mcp/client.py

@@ -13,7 +13,7 @@
     from typing import Self
 
     from mcp import types
-    from mcp_use.session import MCPUseSession
+    from mcp_use.session import MCPSession as MCPUseSession
 
 logger = logging.getLogger(__name__)
 

hud/mcp/openai.py

@@ -93,7 +93,7 @@ def __init__(
         """  # noqa: E501
 
     async def run(
-        self, prompt_or_task: str | TaskConfig, max_steps: int = 10, conversation_mode: bool = False
+        self, prompt_or_task: str | TaskConfig, max_steps: int = 10
     ) -> dict[str, Any]:
         """
         Run the agent with the given prompt or task.
@@ -106,7 +106,7 @@ async def run(
         self.pending_safety_checks = []
 
         # Use base implementation
-        return await super().run(prompt_or_task, max_steps, conversation_mode)
+        return await super().run(prompt_or_task, max_steps)
 
     async def create_initial_messages(self, prompt: str, screenshot: str | None) -> list[Any]:
         """