alita-sdk 0.3.122__py3-none-any.whl → 0.3.124__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/langchain/interfaces/llm_processor.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 import importlib
+from json import dumps
 from traceback import format_exc
 from langchain.chains.llm import LLMChain
 
@@ -33,6 +34,7 @@ from ..retrievers.AlitaRetriever import AlitaRetriever
 from ..tools.log import print_log
 
 
+
 def get_model(model_type: str, model_params: dict):
     """ Get LLM or ChatLLM """
     if model_type is None:
@@ -181,6 +183,11 @@ def add_documents(vectorstore, documents):
     metadata = []
     for document in documents:
         texts.append(document.page_content)
+        for key in document.metadata:
+            if isinstance(document.metadata[key], list):
+                document.metadata[key] = "; ".join(document.metadata[key])
+            if isinstance(document.metadata[key], dict):
+                document.metadata[key] = dumps(document.metadata[key])
         metadata.append(document.metadata)
     vectorstore.add_texts(texts, metadatas=metadata)
 

alita_sdk/langchain/langraph_agent.py

@@ -1,5 +1,5 @@
 import logging
-from typing import Union, Any, Optional, Annotated
+from typing import Union, Any, Optional, Annotated, get_type_hints
 from uuid import uuid4
 
 import yaml
@@ -329,11 +329,17 @@ def create_graph(
                                 structured_output=node.get('structured_output', False)))
                         break
             elif node_type == 'llm':
+                output_vars = node.get('output', [])
+                output_vars_dict = {
+                    var: get_type_hints(state_class).get(var, str).__name__
+                    for var in output_vars
+                }
                 lg_builder.add_node(node_id, LLMNode(
                     client=client, prompt=node.get('prompt', {}),
                     name=node['id'], return_type='dict',
                     response_key=node.get('response_key', 'messages'),
-                    output_variables=node.get('output', []),
+                    structured_output_dict=output_vars_dict,
+                    output_variables=output_vars,
                     input_variables=node.get('input', ['messages']),
                     structured_output=node.get('structured_output', False)))
             elif node_type == 'router':

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/tools/function.py

@@ -47,11 +47,10 @@ class FunctionTool(BaseTool):
             )
             logger.info(f"ToolNode response: {tool_result}")
             if not self.output_variables:
-                return {"messages": [{"role": "assistant", "content": tool_result}]}
+                return {"messages": [{"role": "assistant", "content": dumps(tool_result)}]}
             else:
                 return {
-                    self.output_variables[0]: tool_result,
-                    "messages": [{"role": "assistant", "content": tool_result}]
+                    self.output_variables[0]: tool_result
                 }
         except ValidationError:
             return {"messages": [

alita_sdk/tools/llm.py

@@ -25,6 +25,7 @@ class LLMNode(BaseTool):
     client: Any = None
     return_type: str = "str"
     response_key: str = "messages"
+    structured_output_dict: Optional[dict[str, str]] = None
     output_variables: Optional[List[str]] = None
     input_variables: Optional[List[str]] = None
     structured_output: Optional[bool] = False
@@ -35,11 +36,19 @@ class LLMNode(BaseTool):
         llm_input = create_llm_input(self.prompt, params, kwargs)
         try:
             if self.structured_output and len(self.output_variables) > 0:
-                struct_params = {var: {"type": "str", "description": ""} for var in self.output_variables}
+                struct_params = {
+                    key: {
+                        "type": 'list[str]' if 'list' in value else value,
+                        "description": ""
+                    }
+                    for key, value in (self.structured_output_dict or {}).items()
+                }
                 stuct_model = create_pydantic_model(f"LLMOutput", struct_params)
                 llm = self.client.with_structured_output(stuct_model)
                 completion = llm.invoke(llm_input)
                 result = completion.model_dump()
+                if result.get('messages') and isinstance(result['messages'], list):
+                    result['messages'] = [{'role': 'assistant', 'content': '\n'.join(result['messages'])}]
                 return result
             else:
                 completion = self.client.invoke(llm_input)

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.122.dist-info → alita_sdk-0.3.124.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.122
+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.122.dist-info → alita_sdk-0.3.124.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
@@ -19,7 +19,7 @@ alita_sdk/langchain/assistant.py,sha256=V0MpZG9IQRlKMr1SfAabVbUsIp-gNPO9A1SWxuFi
 alita_sdk/langchain/chat_message_template.py,sha256=kPz8W2BG6IMyITFDA5oeb5BxVRkHEVZhuiGl4MBZKdc,2176
 alita_sdk/langchain/constants.py,sha256=eHVJ_beJNTf1WJo4yq7KMK64fxsRvs3lKc34QCXSbpk,3319
 alita_sdk/langchain/indexer.py,sha256=0ENHy5EOhThnAiYFc7QAsaTNp9rr8hDV_hTK8ahbatk,37592
-alita_sdk/langchain/langraph_agent.py,sha256=vxsbjOE0UWZIuyXNGEMR1QJ8t5yHS7Kzjt8a8nLb-jw,20626
+alita_sdk/langchain/langraph_agent.py,sha256=f9rGk6QGQbfbGudwpM5ax9yS-xDlElFqGAOsZbGvrtI,20919
 alita_sdk/langchain/mixedAgentParser.py,sha256=M256lvtsL3YtYflBCEp-rWKrKtcY1dJIyRGVv7KW9ME,2611
 alita_sdk/langchain/mixedAgentRenderes.py,sha256=asBtKqm88QhZRILditjYICwFVKF5KfO38hu2O-WrSWE,5964
 alita_sdk/langchain/utils.py,sha256=Npferkn10dvdksnKzLJLBI5bNGQyVWTBwqp3vQtUqmY,6631
@@ -41,7 +41,7 @@ alita_sdk/langchain/document_loaders/constants.py,sha256=QpgMiKo-riIBuU67DGhXi7Z
 alita_sdk/langchain/document_loaders/utils.py,sha256=ifh0UJiweIb2iEb2_THJy2pbAnTupOffGtTdRA7vjeE,874
 alita_sdk/langchain/interfaces/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/langchain/interfaces/kwextractor.py,sha256=kSJA9L8g8UArmHu7Bd9dIO0Rrq86JPUb8RYNlnN68FQ,3072
-alita_sdk/langchain/interfaces/llm_processor.py,sha256=oX5feXsQ5D-8q3Mf40Txltv3kkHKb2xBeNPwecy5yDQ,8442
+alita_sdk/langchain/interfaces/llm_processor.py,sha256=AbuWlZ54Cu5zy6AExRN0ou3r15i4pY3WSIVPJx8ltkA,8764
 alita_sdk/langchain/interfaces/loaders.py,sha256=li-O2dubiDNYn-qfVcDsuD4LqP_IZ61cV2vHUZAqeXc,3337
 alita_sdk/langchain/interfaces/splitters.py,sha256=tW65-Ejj9VYyxXFZNgPts_CKILQ18bWp_1bZ-24FKGc,3630
 alita_sdk/langchain/retrievers/AlitaRetriever.py,sha256=osChtJxUlfpsFESpJSE5mnJAkxTXnzgFZnC6l5mUlbo,6148
@@ -68,18 +68,19 @@ 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=ZYi-oZwezaqcZe6Ft6w2hlpbuCKYyDNjl3WRWViVT08,5074
 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
 alita_sdk/tools/artifact.py,sha256=uTa6K5d-NCDRnuLJVd6vA5TNIPH39onyPIyW5Thz4C0,6160
 alita_sdk/tools/datasource.py,sha256=pvbaSfI-ThQQnjHG-QhYNSTYRnZB0rYtZFpjCfpzxYI,2443
 alita_sdk/tools/echo.py,sha256=spw9eCweXzixJqHnZofHE1yWiSUa04L4VKycf3KCEaM,486
-alita_sdk/tools/function.py,sha256=g5VaLtgYHoEMgid1s1zU2NzavqT2IErjhSIWqqPw3Yk,2617
+alita_sdk/tools/function.py,sha256=wzPS5Y8ScmsXspmn73MQnkCcAghsllAg7BLh61ZNuVw,2543
 alita_sdk/tools/indexer_tool.py,sha256=P9S_omk5TZkizb6UXyxMO87Pzj4UCaye0CuXBgCnhTU,4258
-alita_sdk/tools/llm.py,sha256=oiaecQNvFLqokmImyfbBKQ_bNMD8hrD8lhTFpiUytdA,2779
+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.122.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.124.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.122.dist-info/METADATA,sha256=_tp0Oa-uXYLh9RmhhlYGM4dazECkrNy0vtUUFMce_Cg,7075
-alita_sdk-0.3.122.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-alita_sdk-0.3.122.dist-info/top_level.txt,sha256=SWRhxB7Et3cOy3RkE5hR7OIRnHoo3K8EXzoiNlkfOmc,25
-alita_sdk-0.3.122.dist-info/RECORD,,
+alita_sdk-0.3.124.dist-info/METADATA,sha256=SLr3I6hYo-6OFuiJYygEXnREyu4nt2GlWBoqh4jHw6M,7075
+alita_sdk-0.3.124.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+alita_sdk-0.3.124.dist-info/top_level.txt,sha256=SWRhxB7Et3cOy3RkE5hR7OIRnHoo3K8EXzoiNlkfOmc,25
+alita_sdk-0.3.124.dist-info/RECORD,,