alita-sdk 0.3.140b3__tar.gz → 0.3.140b4__tar.gz

{alita_sdk-0.3.140b3 → alita_sdk-0.3.140b4}/PKG-INFO

@@ -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 → alita_sdk-0.3.140b4}/pyproject.toml

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

{alita_sdk-0.3.140b3 → alita_sdk-0.3.140b4}/src/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 → alita_sdk-0.3.140b4}/src/alita_sdk.egg-info/PKG-INFO

@@ -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