alita-sdk 0.3.203__py3-none-any.whl → 0.3.203b1__py3-none-any.whl

alita_sdk/runtime/tools/mcp_server_tool.py

@@ -22,7 +22,13 @@ class McpServerTool(BaseTool):
     def create_pydantic_model_from_schema(schema: dict):
         fields = {}
         for field_name, field_info in schema['properties'].items():
-            field_type = field_info['type']
+            field_type = None
+
+            if 'type' in field_info:
+                field_type = field_info['type']
+            elif 'allOf' in field_info or 'anyOf' in field_info:  # ADO-MCP wit_update_work_item, testplan_add_test_cases_to_suite
+                field_type = 'array'
+
             field_description = field_info.get('description', '')
             if field_type == 'string':
                 if 'enum' in field_info:
@@ -35,16 +41,22 @@ class McpServerTool(BaseTool):
                 field_type = float
             elif field_type == 'boolean':
                 field_type = bool
-            elif field_type == 'object':#Dict[str, Any]
-                nested_model = McpServerTool.create_pydantic_model_from_schema(field_info)
+            elif field_type == 'object':  # Dict[str, Any] - for tool start_browser in selenium mcp
+                nested_model = McpServerTool.create_pydantic_model_from_schema(
+                    field_info) if 'properties' in field_info else dict  # for tool build_run_build in ADO-MCP
                 field_type = nested_model
             elif field_type == 'array':
-                item_schema = field_info['items']
-                item_type = McpServerTool.create_pydantic_model_from_schema(item_schema) if item_schema['type'] == 'object' else (
-                    str if item_schema['type'] == 'string' else
+                item_schema = field_info['items'] if 'items' in field_info else (
+                field_info.get('allOf', field_info.get('anyOf', [{'type': None}]))[
+                    -1])  # ADO-MCP wit_update_work_item, testplan_add_test_cases_to_suite
+                item_type = McpServerTool.create_pydantic_model_from_schema(item_schema) if item_schema[
+                                                                                                'type'] == 'object' else (
+                    str if item_schema['type'] == 'string' else  # support enum
                     int if item_schema['type'] == 'integer' else
                     float if item_schema['type'] == 'number' else
                     bool if item_schema['type'] == 'boolean' else
+                    list if item_schema['type'] == 'array' else  # ADO-MCP testplan_add_test_cases_to_suite
+                    str if item_schema['type'] == 'null' else  # for mcp-atlassian
                     None
                 )
                 if item_type is None:

{alita_sdk-0.3.203.dist-info → alita_sdk-0.3.203b1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.203
+Version: 0.3.203b1
 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>, Artem Dubrovskiy <ad13box@gmail.com>
 License-Expression: Apache-2.0

{alita_sdk-0.3.203.dist-info → alita_sdk-0.3.203b1.dist-info}/RECORD

@@ -84,7 +84,7 @@ alita_sdk/runtime/tools/indexer_tool.py,sha256=whSLPevB4WD6dhh2JDXEivDmTvbjiMV1M
 alita_sdk/runtime/tools/llm.py,sha256=NsrsP-SblyxDdzgMCn9_OBUL0sUGDVS5yqer49V7ciE,15069
 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=eI8QUt497xblwF4Zhbvi8wCg17yh2yoWjcw_AIzHwGE,2819
+alita_sdk/runtime/tools/mcp_server_tool.py,sha256=yFboYcLIORMbSrSi8QkrHHUNZT5FwwBN-gmBYcym3bM,3715
 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
@@ -297,8 +297,8 @@ alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=UHVQUVqcBc3SZvDfO78HSuBzwAsRw
 alita_sdk/tools/zephyr_squad/__init__.py,sha256=0AI_j27xVO5Gk5HQMFrqPTd4uvuVTpiZUicBrdfEpKg,2796
 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.203.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.203.dist-info/METADATA,sha256=5jKh4tiVrW5xNeR51J5Zsci1z6L2v0VtbBkhnfEPGYs,18804
-alita_sdk-0.3.203.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.203.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.203.dist-info/RECORD,,
+alita_sdk-0.3.203b1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.203b1.dist-info/METADATA,sha256=C8GQAab09i5o3skdjcxruSWPuxOl2qCg7ui1RvIZsjQ,18806
+alita_sdk-0.3.203b1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.203b1.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.203b1.dist-info/RECORD,,