alita-sdk 0.3.123__py3-none-any.whl → 0.3.125__py3-none-any.whl

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/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,50 @@ 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):
+    try:
+        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
+    except Exception:
+        logger.error("Error while fetching MCP tools", exc_info=True)
+        return []
+
+
+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/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.dist-info → alita_sdk-0.3.125.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.123
+Version: 0.3.125
 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.dist-info → alita_sdk-0.3.125.dist-info}/RECORD

@@ -4,7 +4,7 @@ alita_sdk/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,
 alita_sdk/agents/llamaAgentParser.py,sha256=N_Nw6WJ8xrNX3fr_JFwjUIg_Urai9lLU3poMgZz0Cyk,1701
 alita_sdk/clients/__init__.py,sha256=5O_7WsZmvg5z5uZf_Jkx-f8j5C6yKIuSwQR6AQ-TM84,31
 alita_sdk/clients/artifact.py,sha256=W6oLlthtsUHfUWjhihaDdhLZdRVqj1D2j7oHHELxJfs,2639
-alita_sdk/clients/client.py,sha256=j97yBXpmy5nCIfkZ6hbzCHnVDbgqrEiTOTQdBDpKQiQ,17912
+alita_sdk/clients/client.py,sha256=aBVoT9rX5D9vF2iGPywmjAi2CPM0m43NnHiKmKlTmPA,18475
 alita_sdk/clients/datasource.py,sha256=HAZovoQN9jBg0_-lIlGBQzb4FJdczPhkHehAiVG3Wx0,1020
 alita_sdk/clients/prompt.py,sha256=li1RG9eBwgNK_Qf0qUaZ8QNTmsncFrAL2pv3kbxZRZg,1447
 alita_sdk/community/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -68,7 +68,7 @@ alita_sdk/toolkits/application.py,sha256=LrxbBV05lkRP3_WtKGBKtMdoQHXVY-_AtFr1cUu
 alita_sdk/toolkits/artifact.py,sha256=7zb17vhJ3CigeTqvzQ4VNBsU5UOCJqAwz7fOJGMYqXw,2348
 alita_sdk/toolkits/datasource.py,sha256=v3FQu8Gmvq7gAGAnFEbA8qofyUhh98rxgIjY6GHBfyI,2494
 alita_sdk/toolkits/prompt.py,sha256=WIpTkkVYWqIqOWR_LlSWz3ug8uO9tm5jJ7aZYdiGRn0,1192
-alita_sdk/toolkits/tools.py,sha256=7qECx6O6IHIE5GjpJet4XhF_9in2-EYYy5BEKIGxr0o,3285
+alita_sdk/toolkits/tools.py,sha256=RFMJ2bYNotPNmz6-clUdAakuIyrunSA16nFZ1gDFSF0,5273
 alita_sdk/toolkits/vectorstore.py,sha256=di08-CRl0KJ9xSZ8_24VVnPZy58iLqHtXW8vuF29P64,2893
 alita_sdk/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/tools/application.py,sha256=UJlYd3Sub10LpAoKkKEpvd4miWyrS-yYE5NKyqx-H4Q,2194
@@ -80,6 +80,7 @@ alita_sdk/tools/indexer_tool.py,sha256=P9S_omk5TZkizb6UXyxMO87Pzj4UCaye0CuXBgCnh
 alita_sdk/tools/llm.py,sha256=JA0OnSU13CLdkS5NFv6iRk8P7k-B47L-oPjk8xrzk48,3223
 alita_sdk/tools/loop.py,sha256=uds0WhZvwMxDVFI6MZHrcmMle637cQfBNg682iLxoJA,8335
 alita_sdk/tools/loop_output.py,sha256=NoGIGYc42wY3NNcWRijYzRnUVXcCn5cRVd8QmuIpoHU,8068
+alita_sdk/tools/mcp_server_tool.py,sha256=xcH9AiqfR2TYrwJ3Ixw-_A7XDodtJCnwmq1SsikXpYk,1930
 alita_sdk/tools/pgvector_search.py,sha256=NN2BGAnq4SsDHIhUcFZ8d_dbEOM8QwB0UwpsWCYruXU,11692
 alita_sdk/tools/prompt.py,sha256=nJafb_e5aOM1Rr3qGFCR-SKziU9uCsiP2okIMs9PppM,741
 alita_sdk/tools/router.py,sha256=wCvZjVkdXK9dMMeEerrgKf5M790RudH68pDortnHSz0,1517
@@ -91,10 +92,10 @@ alita_sdk/utils/evaluate.py,sha256=iM1P8gzBLHTuSCe85_Ng_h30m52hFuGuhNXJ7kB1tgI,1
 alita_sdk/utils/logging.py,sha256=hBE3qAzmcLMdamMp2YRXwOOK9P4lmNaNhM76kntVljs,3124
 alita_sdk/utils/streamlit.py,sha256=zp8owZwHI3HZplhcExJf6R3-APtWx-z6s5jznT2hY_k,29124
 alita_sdk/utils/utils.py,sha256=dM8whOJAuFJFe19qJ69-FLzrUp6d2G-G6L7d4ss2XqM,346
-alita_sdk-0.3.123.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.125.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
 tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/test_jira_analysis.py,sha256=I0cErH5R_dHVyutpXrM1QEo7jfBuKWTmDQvJBPjx18I,3281
-alita_sdk-0.3.123.dist-info/METADATA,sha256=UBv__Ag0bAjUYgjkYIE3pjx2tzIWlSfHZSc-XxiaFkg,7075
-alita_sdk-0.3.123.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-alita_sdk-0.3.123.dist-info/top_level.txt,sha256=SWRhxB7Et3cOy3RkE5hR7OIRnHoo3K8EXzoiNlkfOmc,25
-alita_sdk-0.3.123.dist-info/RECORD,,
+alita_sdk-0.3.125.dist-info/METADATA,sha256=raIYIJIfySOwIA5jp0DHo4UlVizFOhLpWwLwx9AhQi8,7075
+alita_sdk-0.3.125.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+alita_sdk-0.3.125.dist-info/top_level.txt,sha256=SWRhxB7Et3cOy3RkE5hR7OIRnHoo3K8EXzoiNlkfOmc,25
+alita_sdk-0.3.125.dist-info/RECORD,,