alita-sdk 0.3.331__py3-none-any.whl → 0.3.333__py3-none-any.whl

alita_sdk/runtime/toolkits/tools.py

@@ -15,6 +15,7 @@ from ..tools.mcp_server_tool import McpServerTool
 # Import community tools
 from ...community import get_toolkits as community_toolkits, get_tools as community_tools
 from ...tools.memory import MemoryToolkit
+from ...tools.utils import TOOLKIT_SPLITTER
 
 logger = logging.getLogger(__name__)
 
@@ -150,11 +151,11 @@ def _mcp_tools(tools_list, alita):
         tools = []
         #
         for selected_toolkit in tools_list:
-            toolkit_name = selected_toolkit['type']
-            toolkit_conf = toolkit_lookup.get(toolkit_name)
+            server_toolkit_name = selected_toolkit['type']
+            toolkit_conf = toolkit_lookup.get(server_toolkit_name)
             #
             if not toolkit_conf:
-                logger.debug(f"Toolkit '{toolkit_name}' not found in available MCP toolkits. Skipping...")
+                logger.debug(f"Toolkit '{server_toolkit_name}' not found in available MCP toolkits. Skipping...")
                 continue
             #
             available_tools = toolkit_conf.get("tools", [])
@@ -162,7 +163,7 @@ def _mcp_tools(tools_list, alita):
             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']):
+                    if server_tool := _init_single_mcp_tool(server_toolkit_name, selected_toolkit["name"], available_tool, alita, selected_toolkit['settings']):
                         tools.append(server_tool)
         return tools
     except Exception:
@@ -170,9 +171,10 @@ def _mcp_tools(tools_list, alita):
         return []
 
 
-def _init_single_mcp_tool(toolkit_name, available_tool, alita, toolkit_settings):
+def _init_single_mcp_tool(server_toolkit_name, toolkit_name, available_tool, alita, toolkit_settings):
     try:
-        tool_name = available_tool["name"]
+
+        tool_name = f'{toolkit_name}{TOOLKIT_SPLITTER}{available_tool["name"]}'
         return McpServerTool(
             name=tool_name,
             description=f"MCP for a tool '{tool_name}': {available_tool.get("description", "")}",
@@ -180,9 +182,9 @@ def _init_single_mcp_tool(toolkit_name, available_tool, alita, toolkit_settings)
                 available_tool.get("inputSchema", {})
             ),
             client=alita,
-            server=toolkit_name,
+            server=server_toolkit_name,
             tool_timeout_sec=toolkit_settings.get("timeout", 90)
         )
     except Exception as e:
-        logger.error(f"Failed to create McpServerTool for '{toolkit_name}.{tool_name}': {e}")
+        logger.error(f"Failed to create McpServerTool ('{server_toolkit_name}') for '{toolkit_name}.{tool_name}': {e}")
         return None

alita_sdk/runtime/tools/llm.py

@@ -31,26 +31,25 @@ def create_llm_input_with_messages(
     
     # Build the input messages
     input_messages = []
+    messages = params.get('messages', [])
     
     # Add system message from prompt if available
     if prompt:
         try:
             # Format the system message using the prompt template or value and params
             prompt_str = prompt['template'] if 'template' in prompt else prompt['value']
-            system_content = prompt_str.format(**params) if params else prompt_str
-            input_messages.append(SystemMessage(content=system_content))
+            prompt_content = prompt_str.format(**params) if params else prompt_str
+            # if user hasn't specified chat history, add system message as HumanMessage
+            input_messages.append(SystemMessage(content=prompt_content) if messages else HumanMessage(content=prompt_content))
         except KeyError as e:
             error_msg = f"KeyError in prompt formatting: {e}. Available params: {list(params.keys())}"
             logger.error(error_msg)
             raise ToolException(error_msg)
     
     # Add the chat history messages
-    messages = params.get('messages', [])
+
     if messages:
         input_messages.extend(messages)
-    else:
-        # conditionally add a default human message if no chat history
-        input_messages.extend([HumanMessage(content="Reply to this message.")])
 
     return input_messages
 

alita_sdk/runtime/tools/mcp_server_tool.py

@@ -5,6 +5,8 @@ from typing import Any, Type, Literal, Optional, Union, List
 from langchain_core.tools import BaseTool
 from pydantic import BaseModel, Field, create_model, EmailStr, constr
 
+from ...tools.utils import TOOLKIT_SPLITTER
+
 logger = getLogger(__name__)
 
 
@@ -93,7 +95,7 @@ class McpServerTool(BaseTool):
             "tool_timeout_sec": self.tool_timeout_sec,
             "tool_call_id": str(uuid.uuid4()),
             "params": {
-                "name": self.name,
+                "name": self.name.rsplit(TOOLKIT_SPLITTER)[1],
                 "arguments": kwargs
             }
         }

{alita_sdk-0.3.331.dist-info → alita_sdk-0.3.333.dist-info}/METADATA

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

@@ -101,7 +101,7 @@ alita_sdk/runtime/toolkits/configurations.py,sha256=kIDAlnryPQfbZyFxV-9SzN2-Vefz
 alita_sdk/runtime/toolkits/datasource.py,sha256=qk78OdPoReYPCWwahfkKLbKc4pfsu-061oXRryFLP6I,2498
 alita_sdk/runtime/toolkits/prompt.py,sha256=WIpTkkVYWqIqOWR_LlSWz3ug8uO9tm5jJ7aZYdiGRn0,1192
 alita_sdk/runtime/toolkits/subgraph.py,sha256=wwUK8JjPXkGzyVZ3tAukmvST6eGbqx_U11rpnmbrvtg,2105
-alita_sdk/runtime/toolkits/tools.py,sha256=Ea3LO6voPNysdzVB7jYMZIqSMtda7LCq_6fkVfb8C54,7764
+alita_sdk/runtime/toolkits/tools.py,sha256=_8NtB11XGWc3YGp01dyqJMLQJUSGJOdKzGEanlVRO0M,7954
 alita_sdk/runtime/toolkits/vectorstore.py,sha256=BGppQADa1ZiLO17fC0uCACTTEvPHlodEDYEzUcBRbAA,2901
 alita_sdk/runtime/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/runtime/tools/agent.py,sha256=m98QxOHwnCRTT9j18Olbb5UPS8-ZGeQaGiUyZJSyFck,3162
@@ -112,10 +112,10 @@ alita_sdk/runtime/tools/echo.py,sha256=spw9eCweXzixJqHnZofHE1yWiSUa04L4VKycf3KCE
 alita_sdk/runtime/tools/function.py,sha256=0iZJ-UxaPbtcXAVX9G5Vsn7vmD7lrz3cBG1qylto1gs,2844
 alita_sdk/runtime/tools/graph.py,sha256=MbnZYqdmvZY7SGDp43lOVVIjUt5ARHSgj43mdtBjSjQ,3092
 alita_sdk/runtime/tools/indexer_tool.py,sha256=whSLPevB4WD6dhh2JDXEivDmTvbjiMV1MrPl9cz5eLA,4375
-alita_sdk/runtime/tools/llm.py,sha256=lVg8t785JLrB2kwIQjpigfhZZP9-BRu-SEUQw6DQyh8,15128
+alita_sdk/runtime/tools/llm.py,sha256=qd8P7YPqx5uRkMgEEGvDvmV75Gf5kG23ankrXe7Ml2M,15110
 alita_sdk/runtime/tools/loop.py,sha256=uds0WhZvwMxDVFI6MZHrcmMle637cQfBNg682iLxoJA,8335
 alita_sdk/runtime/tools/loop_output.py,sha256=U4hO9PCQgWlXwOq6jdmCGbegtAxGAPXObSxZQ3z38uk,8069
-alita_sdk/runtime/tools/mcp_server_tool.py,sha256=trGraI8-AwdbNmTKMjfmlBxgTDMTE4-21heCVtd_lz0,4156
+alita_sdk/runtime/tools/mcp_server_tool.py,sha256=E7-_rdvlXITLzJ3xeKTPWxditHAUt-40O_5QFvBqD0w,4229
 alita_sdk/runtime/tools/pgvector_search.py,sha256=NN2BGAnq4SsDHIhUcFZ8d_dbEOM8QwB0UwpsWCYruXU,11692
 alita_sdk/runtime/tools/prompt.py,sha256=nJafb_e5aOM1Rr3qGFCR-SKziU9uCsiP2okIMs9PppM,741
 alita_sdk/runtime/tools/router.py,sha256=wCvZjVkdXK9dMMeEerrgKf5M790RudH68pDortnHSz0,1517
@@ -349,8 +349,8 @@ alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=kT0TbmMvuKhDUZc0i7KO18O38JM9S
 alita_sdk/tools/zephyr_squad/__init__.py,sha256=0ne8XLJEQSLOWfzd2HdnqOYmQlUliKHbBED5kW_Vias,2895
 alita_sdk/tools/zephyr_squad/api_wrapper.py,sha256=kmw_xol8YIYFplBLWTqP_VKPRhL_1ItDD0_vXTe_UuI,14906
 alita_sdk/tools/zephyr_squad/zephyr_squad_cloud_client.py,sha256=R371waHsms4sllHCbijKYs90C-9Yu0sSR3N4SUfQOgU,5066
-alita_sdk-0.3.331.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.331.dist-info/METADATA,sha256=EvsLkYCZntZIiWjFFEhnZJhZiY_vqbOmoMLPZxK4fIA,18835
-alita_sdk-0.3.331.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.331.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.331.dist-info/RECORD,,
+alita_sdk-0.3.333.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.333.dist-info/METADATA,sha256=4WHbu-7dc9jxF4SicrImaSmW4E0EbRqIa0F_a2-zcfQ,18835
+alita_sdk-0.3.333.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.333.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.333.dist-info/RECORD,,