alita-sdk 0.3.440__tar.gz → 0.3.442__tar.gz

{alita_sdk-0.3.440 → alita_sdk-0.3.442}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.440
+Version: 0.3.442
 Summary: SDK for building langchain agents using resources from Alita
 Author-email: Artem Rozumenko <artyom.rozumenko@gmail.com>, Mikalai Biazruchka <mikalai_biazruchka@epam.com>, Roman Mitusov <roman_mitusov@epam.com>, Ivan Krakhmaliuk <lifedj27@gmail.com>, Artem Dubrovskiy <ad13box@gmail.com>
 License-Expression: Apache-2.0

{alita_sdk-0.3.440 → alita_sdk-0.3.442}/alita_sdk/runtime/clients/client.py

@@ -342,6 +342,9 @@ class AlitaClient:
             app_type = "react"
         elif app_type == 'autogen':
             app_type = "react"
+        
+        # LangChainAssistant constructor calls get_tools() which may raise McpAuthorizationRequired
+        # The exception will propagate naturally to the indexer worker's outer handler
         if runtime == 'nonrunnable':
             return LangChainAssistant(self, data, llm, chat_history, app_type,
                                       tools=tools, memory=memory, store=store, mcp_tokens=mcp_tokens)
@@ -605,6 +608,9 @@ class AlitaClient:
             'tools': tools,  # Tool configs that will be processed by get_tools()
             'variables': variables
         }
+        
+        # LangChainAssistant constructor calls get_tools() which may raise McpAuthorizationRequired
+        # The exception will propagate naturally to the indexer worker's outer handler
         return LangChainAssistant(
             self,
             agent_data,
@@ -756,7 +762,23 @@ class AlitaClient:
                 }
 
             # Instantiate the toolkit with client and LLM support
-            tools = instantiate_toolkit_with_client(toolkit_config, llm, self)
+            try:
+                tools = instantiate_toolkit_with_client(toolkit_config, llm, self)
+            except Exception as toolkit_error:
+                # Re-raise McpAuthorizationRequired to allow proper handling upstream
+                from ..utils.mcp_oauth import McpAuthorizationRequired
+                if isinstance(toolkit_error, McpAuthorizationRequired):
+                    raise
+                # For other errors, return error response
+                return {
+                    "success": False,
+                    "error": f"Failed to instantiate toolkit '{toolkit_config.get('toolkit_name')}': {str(toolkit_error)}",
+                    "tool_name": tool_name,
+                    "toolkit_config": toolkit_config_parsed_json,
+                    "llm_model": llm_model,
+                    "events_dispatched": events_dispatched,
+                    "execution_time_seconds": 0.0
+                }
 
             if not tools:
                 return {

{alita_sdk-0.3.440 → alita_sdk-0.3.442}/alita_sdk/runtime/toolkits/mcp.py

@@ -545,7 +545,7 @@ class McpToolkit(BaseToolkit):
 
             auth_header = response.headers.get('WWW-Authenticate', '')
             if response.status_code == 401:
-                resource_metadata_url = extract_resource_metadata_url(auth_header)
+                resource_metadata_url = extract_resource_metadata_url(auth_header, connection_config.url)
                 metadata = fetch_resource_metadata(resource_metadata_url, timeout=timeout) if resource_metadata_url else None
                 raise McpAuthorizationRequired(
                     message=f"MCP server {connection_config.url} requires OAuth authorization",
@@ -554,6 +554,7 @@ class McpToolkit(BaseToolkit):
                     www_authenticate=auth_header,
                     resource_metadata=metadata,
                     status=response.status_code,
+                    tool_name=toolkit_name,
                 )
 
             if response.status_code != 200:

{alita_sdk-0.3.440 → alita_sdk-0.3.442}/alita_sdk/runtime/tools/mcp_remote_tool.py

@@ -121,7 +121,7 @@ class McpRemoteTool(McpServerTool):
                 async with session.post(self.server_url, json=mcp_request, headers=headers) as response:
                     auth_header = response.headers.get('WWW-Authenticate') or response.headers.get('Www-Authenticate')
                     if response.status == 401:
-                        resource_metadata_url = extract_resource_metadata_url(auth_header)
+                        resource_metadata_url = extract_resource_metadata_url(auth_header, self.server_url)
                         metadata = None
                         if resource_metadata_url:
                             metadata = await fetch_resource_metadata_async(
@@ -136,6 +136,7 @@ class McpRemoteTool(McpServerTool):
                             www_authenticate=auth_header,
                             resource_metadata=metadata,
                             status=response.status,
+                            tool_name=self.name,
                         )
 
                     if response.status != 200:

{alita_sdk-0.3.440 → alita_sdk-0.3.442}/alita_sdk/runtime/utils/mcp_oauth.py

@@ -21,6 +21,7 @@ class McpAuthorizationRequired(ToolException):
         www_authenticate: Optional[str] = None,
         resource_metadata: Optional[Dict[str, Any]] = None,
         status: Optional[int] = None,
+        tool_name: Optional[str] = None,
     ):
         super().__init__(message)
         self.server_url = server_url
@@ -28,6 +29,7 @@ class McpAuthorizationRequired(ToolException):
         self.www_authenticate = www_authenticate
         self.resource_metadata = resource_metadata
         self.status = status
+        self.tool_name = tool_name
 
     def to_dict(self) -> Dict[str, Any]:
         return {
@@ -37,18 +39,31 @@ class McpAuthorizationRequired(ToolException):
             "www_authenticate": self.www_authenticate,
             "resource_metadata": self.resource_metadata,
             "status": self.status,
+            "tool_name": self.tool_name,
         }
 
 
-def extract_resource_metadata_url(www_authenticate: Optional[str]) -> Optional[str]:
-    """Pull the resource_metadata URL from a WWW-Authenticate header if present."""
-    if not www_authenticate:
+def extract_resource_metadata_url(www_authenticate: Optional[str], server_url: Optional[str] = None) -> Optional[str]:
+    """
+    Pull the resource_metadata URL from a WWW-Authenticate header if present.
+    If not found and server_url is provided, construct the standard .well-known URL.
+    """
+    if not www_authenticate and not server_url:
         return None
 
     # RFC9728 returns `resource_metadata="<url>"` inside the header value
-    match = re.search(r'resource_metadata\s*=\s*\"?([^\", ]+)\"?', www_authenticate)
-    if match:
-        return match.group(1)
+    if www_authenticate:
+        match = re.search(r'resource_metadata\s*=\s*\"?([^\", ]+)\"?', www_authenticate)
+        if match:
+            return match.group(1)
+    
+    # Fallback: construct standard .well-known URL if server_url provided
+    if server_url:
+        parsed = urlparse(server_url)
+        # Build the standard resource metadata URL
+        base_url = f"{parsed.scheme}://{parsed.netloc}"
+        return f"{base_url}/.well-known/resource-metadata"
+    
     return None
 
 

{alita_sdk-0.3.440 → alita_sdk-0.3.442}/alita_sdk/runtime/utils/toolkit_utils.py

@@ -29,13 +29,14 @@ def instantiate_toolkit_with_client(toolkit_config: Dict[str, Any],
         
     Raises:
         ValueError: If required configuration or client is missing
+        McpAuthorizationRequired: If MCP server requires OAuth authorization
         Exception: If toolkit instantiation fails
     """
+    toolkit_name = toolkit_config.get('toolkit_name', 'unknown')
     try:
         from ..toolkits.tools import get_tools
         
-        toolkit_name = toolkit_config.get('toolkit_name')
-        if not toolkit_name:
+        if not toolkit_name or toolkit_name == 'unknown':
             raise ValueError("toolkit_name is required in configuration")
         
         if not llm_client:
@@ -70,6 +71,12 @@ def instantiate_toolkit_with_client(toolkit_config: Dict[str, Any],
         return tools
             
     except Exception as e:
+        # Re-raise McpAuthorizationRequired without logging as error
+        from ..utils.mcp_oauth import McpAuthorizationRequired
+        if isinstance(e, McpAuthorizationRequired):
+            logger.info(f"Toolkit {toolkit_name} requires MCP OAuth authorization")
+            raise
+        # Log and re-raise other errors
         logger.error(f"Error instantiating toolkit {toolkit_name} with client: {str(e)}")
         raise
 

{alita_sdk-0.3.440 → alita_sdk-0.3.442}/alita_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.440
+Version: 0.3.442
 Summary: SDK for building langchain agents using resources from Alita
 Author-email: Artem Rozumenko <artyom.rozumenko@gmail.com>, Mikalai Biazruchka <mikalai_biazruchka@epam.com>, Roman Mitusov <roman_mitusov@epam.com>, Ivan Krakhmaliuk <lifedj27@gmail.com>, Artem Dubrovskiy <ad13box@gmail.com>
 License-Expression: Apache-2.0

{alita_sdk-0.3.440 → alita_sdk-0.3.442}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "alita_sdk"
-version = "0.3.440"
+version = "0.3.442"
 description = "SDK for building langchain agents using resources from Alita"
 readme = "README.md"
 requires-python = ">=3.10"