alita-sdk 0.3.123__tar.gz → 0.3.124__tar.gz

{alita_sdk-0.3.123 → alita_sdk-0.3.124}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.123
+Version: 0.3.124
 Summary: SDK for building langchain agents using resouces from Alita
 Author-email: Artem Rozumenko <artyom.rozumenko@gmail.com>, Mikalai Biazruchka <mikalai_biazruchka@epam.com>, Roman Mitusov <roman_mitusov@epam.com>, Ivan Krakhmaliuk <lifedjik@gmail.com>
 Project-URL: Homepage, https://projectalita.ai

{alita_sdk-0.3.123 → alita_sdk-0.3.124}/pyproject.toml

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

{alita_sdk-0.3.123 → alita_sdk-0.3.124}/src/alita_sdk/clients/client.py

@@ -50,6 +50,8 @@ class AlitaClient:
         self.datasources_predict = f"{self.base_url}{self.api_path}/datasources/predict/prompt_lib/{self.project_id}"
         self.datasources_search = f"{self.base_url}{self.api_path}/datasources/search/prompt_lib/{self.project_id}"
         self.app = f"{self.base_url}{self.api_path}/applications/application/prompt_lib/{self.project_id}"
+        self.mcp_tools_list = f"{self.base_url}{self.api_path}/mcp_sse/tools_list/{self.project_id}"
+        self.mcp_tools_call = f"{self.base_url}{self.api_path}/mcp_sse/tools_call/{self.project_id}"
         self.application_versions = f"{self.base_url}{self.api_path}/applications/version/prompt_lib/{self.project_id}"
         self.list_apps_url = f"{self.base_url}{self.api_path}/applications/applications/prompt_lib/{self.project_id}"
         self.integration_details = f"{self.base_url}{self.api_path}/integrations/integration/{self.project_id}"
@@ -61,6 +63,16 @@ class AlitaClient:
         self.ai_section_url = f'{self.base_url}{self.api_path}/integrations/integrations/default/{self.project_id}?section=ai'
         self.configurations: list = configurations or []
 
+    def get_mcp_toolkits(self):
+        url = self.mcp_tools_list
+        data = requests.get(url, headers=self.headers, verify=False).json()
+        return data
+
+    def mcp_tool_call(self, params: dict[str, Any]):
+        url = self.mcp_tools_call
+        data = requests.post(url, headers=self.headers, json=params, verify=False).json()
+        return data
+
     def prompt(self, prompt_id, prompt_version_id, chat_history=None, return_tool=False):
         url = f"{self.prompt_versions}/{prompt_id}/{prompt_version_id}"
         data = requests.get(url, headers=self.headers, verify=False).json()

{alita_sdk-0.3.123 → alita_sdk-0.3.124}/src/alita_sdk/toolkits/tools.py

@@ -13,6 +13,8 @@ from .vectorstore import VectorStoreToolkit
 from ..community.analysis.jira_analyse import AnalyseJira
 from ..community.browseruse import BrowserUseToolkit
 
+from ..tools.mcp_server_tool import McpServerTool
+
 logger = logging.getLogger(__name__)
 
 
@@ -36,6 +38,7 @@ def get_toolkits():
 def get_tools(tools_list: list, alita: 'AlitaClient', llm: 'LLMLikeObject') -> list:
     prompts = []
     tools = []
+
     for tool in tools_list:
         if tool['type'] == 'prompt':
             prompts.append([
@@ -82,4 +85,46 @@ def get_tools(tools_list: list, alita: 'AlitaClient', llm: 'LLMLikeObject') -> l
     if len(prompts) > 0:
         tools += PromptToolkit.get_toolkit(alita, prompts).get_tools()
     tools += alita_tools(tools_list, alita, llm)
+    tools += _mcp_tools(tools_list, alita)
+    return tools
+
+
+def _mcp_tools(tools_list, alita):
+    all_available_toolkits = alita.get_mcp_toolkits()
+    toolkit_lookup = {tk["name"].lower(): tk for tk in all_available_toolkits}
+    tools = []
+    #
+    for selected_toolkit in tools_list:
+        toolkit_name = selected_toolkit['type'].lower()
+        toolkit_conf = toolkit_lookup.get(toolkit_name)
+        #
+        if not toolkit_conf:
+            logger.warning(f"Toolkit '{toolkit_name}' not found in available toolkits.")
+            continue
+        #
+        available_tools = toolkit_conf.get("tools", [])
+        selected_tools = [name.lower() for name in selected_toolkit['settings'].get('selected_tools', [])]
+        for available_tool in available_tools:
+            tool_name = available_tool.get("name", "").lower()
+            if not selected_tools or tool_name in selected_tools:
+                if server_tool := _init_single_mcp_tool(toolkit_name, available_tool, alita, selected_toolkit['settings']):
+                    tools.append(server_tool)
     return tools
+
+
+def _init_single_mcp_tool(toolkit_name, available_tool, alita, toolkit_settings):
+    try:
+        tool_name = available_tool["name"]
+        return McpServerTool(
+            name=tool_name,
+            description=available_tool.get("description", ""),
+            args_schema=McpServerTool.create_pydantic_model_from_schema(
+                available_tool.get("inputSchema", {})
+            ),
+            client=alita,
+            server=toolkit_name,
+            tool_timeout_sec=toolkit_settings["timeout"]
+        )
+    except Exception as e:
+        logger.error(f"Failed to create McpServerTool for '{toolkit_name}.{tool_name}': {e}")
+        return None

alita_sdk-0.3.124/src/alita_sdk/tools/mcp_server_tool.py

@@ -0,0 +1,58 @@
+import uuid
+from logging import getLogger
+from typing import Any, Type, Literal, Optional
+
+from langchain_core.tools import BaseTool
+from pydantic import BaseModel, Field, create_model
+
+logger = getLogger(__name__)
+
+
+class McpServerTool(BaseTool):
+    name: str
+    description: str
+    args_schema: Optional[Type[BaseModel]] = None
+    return_type: str = "str"
+    client: Any
+    server: str
+    tool_timeout_sec: int = 60
+
+
+    @staticmethod
+    def create_pydantic_model_from_schema(schema: dict):
+        fields = {}
+        for field_name, field_info in schema['properties'].items():
+            field_type = field_info['type']
+            field_description = field_info.get('description', '')
+            if field_type == 'string':
+                if 'enum' in field_info:
+                    field_type = Literal[tuple(field_info['enum'])]
+                else:
+                    field_type = str
+            elif field_type == 'integer':
+                field_type = int
+            elif field_type == 'number':
+                field_type = float
+            elif field_type == 'boolean':
+                field_type = bool
+            else:
+                raise ValueError(f"Unsupported field type: {field_type}")
+
+            if field_name in schema.get('required', []):
+                fields[field_name] = (field_type, Field(..., description=field_description))
+            else:
+                fields[field_name] = (Optional[field_type], Field(None, description=field_description))
+        return create_model('DynamicModel', **fields)
+
+    def _run(self, *args, **kwargs):
+        call_data = {
+            "server": self.server,
+            "tool_timeout_sec": self.tool_timeout_sec,
+            "tool_call_id": str(uuid.uuid4()),
+            "params": {
+                "name": self.name,
+                "arguments": kwargs
+            }
+        }
+        
+        return self.client.mcp_tool_call(call_data)

{alita_sdk-0.3.123 → alita_sdk-0.3.124}/src/alita_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.123
+Version: 0.3.124
 Summary: SDK for building langchain agents using resouces from Alita
 Author-email: Artem Rozumenko <artyom.rozumenko@gmail.com>, Mikalai Biazruchka <mikalai_biazruchka@epam.com>, Roman Mitusov <roman_mitusov@epam.com>, Ivan Krakhmaliuk <lifedjik@gmail.com>
 Project-URL: Homepage, https://projectalita.ai

{alita_sdk-0.3.123 → alita_sdk-0.3.124}/src/alita_sdk.egg-info/SOURCES.txt

@@ -89,6 +89,7 @@ src/alita_sdk/tools/indexer_tool.py
 src/alita_sdk/tools/llm.py
 src/alita_sdk/tools/loop.py
 src/alita_sdk/tools/loop_output.py
+src/alita_sdk/tools/mcp_server_tool.py
 src/alita_sdk/tools/pgvector_search.py
 src/alita_sdk/tools/prompt.py
 src/alita_sdk/tools/router.py