alita-sdk 0.3.140b3__py3-none-any.whl → 0.3.140b4__py3-none-any.whl

alita_sdk/tools/mcp_server_tool.py

@@ -19,7 +19,7 @@ class McpServerTool(BaseTool):
 
 
     @staticmethod
-    def create_pydantic_model_from_schema(schema: dict):
+    def create_pydantic_model_from_schema(schema: dict, model_name: str = "ArgsSchema"):
         def parse_type(field: dict, name: str = "Field") -> Any:
             if "allOf" in field:
                 merged = {}
@@ -67,7 +67,7 @@ class McpServerTool(BaseTool):
             if t == "boolean":
                 return bool
             if t == "object":
-                return McpServerTool.create_pydantic_model_from_schema(field)
+                return McpServerTool.create_pydantic_model_from_schema(field, name.capitalize())
             if t == "array":
                 items = field.get("items", {})
                 return List[parse_type(items, name + "Item")]
@@ -85,7 +85,16 @@ class McpServerTool(BaseTool):
             if "format" in prop:
                 field_args["format"] = prop["format"]
             fields[name] = (typ, Field(default, **field_args))
-        return create_model("DynamicModel", **fields)
+        return create_model(model_name, **fields)
+
+    def _to_args_and_kwargs(
+        self, tool_input: Union[str, dict], tool_call_id: Optional[str]
+    ) -> tuple[tuple, dict]:
+        # just return input as is without any transformation to/from pydentic model
+        if isinstance(tool_input, str):
+            return (tool_input,), {}
+        else:
+            return (), tool_input
 
     def _run(self, *args, **kwargs):
         call_data = {

{alita_sdk-0.3.140b3.dist-info → alita_sdk-0.3.140b4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.140b3
+Version: 0.3.140b4
 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 <lifedjik@gmail.com>
 Project-URL: Homepage, https://projectalita.ai

{alita_sdk-0.3.140b3.dist-info → alita_sdk-0.3.140b4.dist-info}/RECORD

@@ -88,7 +88,7 @@ alita_sdk/tools/indexer_tool.py,sha256=P9S_omk5TZkizb6UXyxMO87Pzj4UCaye0CuXBgCnh
 alita_sdk/tools/llm.py,sha256=UUd4U4CuCNImHxVJAF5daYbew-7X9H8k21bUWy2PDvg,3502
 alita_sdk/tools/loop.py,sha256=uds0WhZvwMxDVFI6MZHrcmMle637cQfBNg682iLxoJA,8335
 alita_sdk/tools/loop_output.py,sha256=U4hO9PCQgWlXwOq6jdmCGbegtAxGAPXObSxZQ3z38uk,8069
-alita_sdk/tools/mcp_server_tool.py,sha256=SWuCGz6ZEYbB0jWElhHIWEGUzNekz3gViKQ99SurZrk,4109
+alita_sdk/tools/mcp_server_tool.py,sha256=WIDh8i6pJX-ZbC3uHGSHS5gD9FUPFgvrY1epNzqntV0,4496
 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
@@ -101,13 +101,13 @@ alita_sdk/utils/logging.py,sha256=hBE3qAzmcLMdamMp2YRXwOOK9P4lmNaNhM76kntVljs,31
 alita_sdk/utils/save_dataframe.py,sha256=tNwnaVCvN4_B0oi3F4Y3Z13elbIRtwKmdKHsMcj658g,1465
 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.140b3.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.140b4.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
 tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/test_ado_analysis.py,sha256=wsxB4B2Ycxoiykthh6YbPQ9hqsDbPFie8D9ZK1i_6kg,3311
 tests/test_github_analysis.py,sha256=ulR4CEGmiMRPydJuX7aQcglzvhC7kFOAtZRLLBB9F_M,3148
 tests/test_gitlab_analysis.py,sha256=J7Y2mNi5Sj8-rH2PMRmVbT3uwZ17YeR9pcs0MDIyNW4,3352
 tests/test_jira_analysis.py,sha256=6F3Elikt02L28N6sS_AKDy9lgqgD81_hr979NcdZeg4,3359
-alita_sdk-0.3.140b3.dist-info/METADATA,sha256=9OBtknQ__fZR7o8DPh0QKuvcd2eMHY7R-TF5no3SZks,7078
-alita_sdk-0.3.140b3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.140b3.dist-info/top_level.txt,sha256=SWRhxB7Et3cOy3RkE5hR7OIRnHoo3K8EXzoiNlkfOmc,25
-alita_sdk-0.3.140b3.dist-info/RECORD,,
+alita_sdk-0.3.140b4.dist-info/METADATA,sha256=kPo52ujJYFCH8RFmRn2d8PW3fBsSfewCbqOwjFBhsqI,7078
+alita_sdk-0.3.140b4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.140b4.dist-info/top_level.txt,sha256=SWRhxB7Et3cOy3RkE5hR7OIRnHoo3K8EXzoiNlkfOmc,25
+alita_sdk-0.3.140b4.dist-info/RECORD,,