fix:Don't set default imageId

Author: AgbAccount

agb/agb.py

@@ -12,7 +12,6 @@
 from agb.api.client import Client as mcp_client
 from agb.api.models import (
     CreateSessionRequest,
-    GetMcpResourceRequest,
     ReleaseSessionRequest,
     CreateSessionResponse,
 )
@@ -50,8 +49,8 @@ def __init__(self, api_key: str = "", cfg: Optional[Config] = None):
         self.config = load_config(cfg)
 
         self.api_key = api_key
-        self.endpoint = self.config .endpoint
-        self.timeout_ms = self.config .timeout_ms
+        self.endpoint = self.config.endpoint
+        self.timeout_ms = self.config.timeout_ms
 
         # Initialize the HTTP API client with the complete config
         self.client = mcp_client(self.config)
@@ -76,14 +75,19 @@ def create(self, params: Optional[CreateSessionParams] = None) -> SessionResult:
 
             request = CreateSessionRequest(authorization=f"Bearer {self.api_key}")
 
-            # Use default image if no image_id specified
-            if not params.image_id:
-                params.image_id = "code_latest"
-            else:
+            if params.image_id:
                 request.image_id = params.image_id
 
             response : CreateSessionResponse = self.client.create_mcp_session(request)
 
+            # Check if response is empty
+            if response is None:
+                return SessionResult(
+                    request_id="",
+                    success=False,
+                    error_message="OpenAPI client returned None response",
+                )
+
             try:
                 print("Response body:")
                 print(response.to_dict())

agb/api/base_service.py

@@ -83,8 +83,7 @@ def _call_mcp_tool(self, name: str, args: Dict[str, Any], read_timeout: int = No
 
                 if response.is_tool_successful():
                     # Tool execution successful
-                    print("response.json_data =", response.json_data)
-                    result = self._parse_response_body(response.json_data or {})
+                    result = response.get_tool_result()
                     return OperationResult(request_id=request_id, success=True, data=result)
                 else:
                     # Tool execution failed
@@ -120,105 +119,4 @@ def _call_mcp_tool(self, name: str, args: Dict[str, Any], read_timeout: int = No
                 request_id=request_id,
                 success=False,
                 error_message=f"Failed to call MCP tool {name}: {handled_error}",
-            )
-
-    def _sanitize_error(self, error_str: str) -> str:
-        """
-        Sanitizes error messages to remove sensitive information like API keys.
-
-        Args:
-            error_str (str): The error string to sanitize.
-
-        Returns:
-            str: The sanitized error string.
-        """
-        import re
-
-        if not error_str:
-            return error_str
-
-        # Remove API key from URLs
-        # Pattern: apiKey=akm-xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx
-        api_key_pattern = re.compile(r'apiKey=akm-[a-f0-9-]+')
-        error_str = api_key_pattern.sub('apiKey=***REDACTED***', error_str)
-
-        # Remove API key from Bearer tokens
-        # Pattern: Bearer akm-xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx
-        bearer_pattern = re.compile(r'Bearer akm-[a-f0-9-]+')
-        error_str = bearer_pattern.sub('Bearer ***REDACTED***', error_str)
-
-        # Remove API key from query parameters
-        # Pattern: &apiKey=akm-xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx
-        query_pattern = re.compile(r'&apiKey=akm-[a-f0-9-]+')
-        error_str = query_pattern.sub('&apiKey=***REDACTED***', error_str)
-
-        # Remove API key from URL paths
-        # Pattern: /callTool?apiKey=akm-xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx
-        url_pattern = re.compile(r'/callTool\?apiKey=akm-[a-f0-9-]+')
-        error_str = url_pattern.sub('/callTool?apiKey=***REDACTED***', error_str)
-
-        return error_str
-
-    def _parse_response_body(
-        self, body: Dict[str, Any], parse_json: bool = False
-    ) -> Any:
-        """
-        Parses the response body from the MCP tool.
-
-        Args:
-            body (Dict[str, Any]): The response body.
-            parse_json (bool, optional): Whether to parse the text as JSON.
-            Defaults to False.
-
-        Returns:
-            Any: The parsed content. If parse_json is True, returns the parsed
-                 JSON object; otherwise returns the raw text.
-
-
-        Raises:
-            AGBError: If the response contains errors or is invalid.
-        """
-        try:
-            response_data = body.get("data", {})
-            if response_data.get("isError", False):
-                error_content = response_data.get("content", [])
-                try:
-                    print("error_content =")
-                    print(json.dumps(error_content, ensure_ascii=False, indent=2))
-                except Exception:
-                    print(f"error_content: {error_content}")
-                error_message = "; ".join(
-                    item.get("text", "Unknown error")
-                    for item in error_content
-                    if isinstance(item, dict)
-                )
-                raise AGBError(f"Error in response: {error_message}")
-
-
-            if not response_data:
-                raise AGBError("No data field in response")
-
-            # Handle 'content' field for other methods
-            content = response_data.get("content", [])
-            if not content or not isinstance(content, list):
-                raise AGBError("No content found in response")
-
-            content_item = content[0]
-            text_string = content_item.get("text")
-
-            # Allow text field to be empty string
-            if text_string is None:
-                raise AGBError("Text field not found in response")
-
-            return text_string
-
-        except AGBError as e:
-            # Transform AGBError to the expected type
-            handled_error = self._handle_error(e)
-            raise handled_error
-        except Exception as e:
-            # Transform AGBError to the expected type
-            handled_error = self._handle_error(
-                AGBError(f"Error parsing response body: {e}")
-            )
-            raise handled_error
+            )

agb/api/models/create_session_response.py

@@ -67,6 +67,7 @@ def __init__(
         success: bool,
         json_data: Optional[Dict[str, Any]] = None,
         text: Optional[str] = None,
+        error: Optional[str] = None,
         request_id: Optional[str] = None
     ):
         self.status_code = status_code
@@ -75,6 +76,7 @@ def __init__(
         self.success = success
         self.json_data = json_data
         self.text = text
+        self.error = error
         self.request_id = request_id
 
         # Parse fields from JSON data
@@ -124,6 +126,8 @@ def get_resource_url(self) -> Optional[str]:
 
     def get_error_message(self) -> str:
         """Get error message"""
+        if self.error:
+            return self.error
         if self.data and self.data.err_msg:
             return self.data.err_msg
         if self.message:
@@ -144,6 +148,8 @@ def to_dict(self) -> Dict[str, Any]:
             result['json'] = self.json_data
         if self.text:
             result['text'] = self.text
+        if self.error:
+            result['error'] = self.error
 
         return result
 

agb/modules/browser/browser_agent.py

@@ -115,16 +115,7 @@ def act(self, page, action_input: Union[ObserveResult, ActOptions]) -> 'ActResul
                 # Map snake_case keys to ActResult members
                 message = data.get("message", "")
                 action = data.get("action", "")
-
-                # Parse the message to extract the actual success status
-                # Format: "is_done=False success=True extracted_content='...'"
-                if message:
-                    # Check if message contains success=True
-                    if "success=True" in message:
-                        success = True
-                    elif "success=False" in message:
-                        success = False
-                    # If success=None, keep the original value
+                success = data.get("success", False)
 
                 return ActResult(success=success, message=message, action=action)
             else:
@@ -173,15 +164,7 @@ async def act_async(self, page, action_input: Union[ObserveResult, ActOptions])
                 # Map snake_case keys to ActResult members
                 message = data.get("message", "")
                 action = data.get("action", "")
-
-                # Parse the message to extract the actual success status
-                # Format: "is_done=True success=True extracted_content='...'"
-                if message:
-                    # Extract success value from message
-                    if "success=True" in message:
-                        success = True
-                    elif "is_done=False" in message or "error=" in message:
-                        success = False
+                success = data.get("success", False)
 
                 return ActResult(success=success, message=message, action=action)
             else:

pyproject.toml

@@ -28,7 +28,7 @@ dependencies = [
     "pydantic>=2.0",
     "aiohttp>=3.12.5",
 ]
-version = "0.1.0"
+version = "0.1.1"
 
 [project.optional-dependencies]
 test = [