alita-sdk 0.3.445__tar.gz → 0.3.446__tar.gz

{alita_sdk-0.3.445 → alita_sdk-0.3.446}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.445
+Version: 0.3.446
 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.445 → alita_sdk-0.3.446}/alita_sdk/runtime/tools/mcp_remote_tool.py

@@ -12,7 +12,7 @@ from concurrent.futures import ThreadPoolExecutor
 from typing import Any, Dict, Optional
 
 from .mcp_server_tool import McpServerTool
-from pydantic import Field, computed_field
+from pydantic import Field
 from ..utils.mcp_oauth import (
     McpAuthorizationRequired,
     canonical_resource,
@@ -38,11 +38,20 @@ class McpRemoteTool(McpServerTool):
     prompt_name: Optional[str] = None  # Original prompt name if this is a prompt
     session_id: Optional[str] = Field(default=None, description="MCP session ID for stateful SSE servers")
     
-    @computed_field
-    @property
-    def metadata(self) -> dict:
-        """Return tool metadata including session information for UI."""
-        return self.get_session_metadata()
+    def model_post_init(self, __context: Any) -> None:
+        """Update metadata with session info after model initialization."""
+        super().model_post_init(__context)
+        self._update_metadata_with_session()
+    
+    def _update_metadata_with_session(self):
+        """Update the metadata dict with current session information."""
+        if self.session_id:
+            if self.metadata is None:
+                self.metadata = {}
+            self.metadata.update({
+                'mcp_session_id': self.session_id,
+                'mcp_server_url': canonical_resource(self.server_url)
+            })
     
     def __getstate__(self):
         """Custom serialization for pickle compatibility."""
@@ -201,6 +210,8 @@ class McpRemoteTool(McpServerTool):
                             if session_id:
                                 # Retry the request with session
                                 self.session_id = session_id
+                                # Update metadata so callbacks can access session info
+                                self._update_metadata_with_session()
                                 separator = '&' if '?' in self.server_url else '?'
                                 retry_url = f"{self.server_url}{separator}sessionId={session_id}"
                                 logger.info(f"[MCP Session] Retrying tool call with new session: {session_id}")

{alita_sdk-0.3.445 → alita_sdk-0.3.446}/alita_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.445
+Version: 0.3.446
 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.445 → alita_sdk-0.3.446}/pyproject.toml

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