alita-sdk 0.3.139__py3-none-any.whl → 0.3.140b1__py3-none-any.whl

alita_sdk/clients/client.py

@@ -176,17 +176,21 @@ class AlitaClient:
     def application(self, client: Any, application_id: int, application_version_id: int,
                     tools: Optional[list] = None, chat_history: Optional[List[Any]] = None,
                     app_type=None, memory=None, runtime='langchain',
-                    application_variables: Optional[dict] = None):
+                    application_variables: Optional[dict] = None,
+                    version_details: Optional[dict] = None):
         if tools is None:
             tools = []
         if chat_history is None:
             chat_history = []
-        try:
-            data = self.get_app_version_details(application_id, application_version_id)
-        except ApiDetailsRequestError as e:
-            error_msg = f"Failed to fetch application version details for {application_id}/{application_version_id}\nDetails: {e}"
-            logger.error(error_msg)
-            raise ToolException(error_msg)
+        if version_details:
+            data = version_details
+        else:
+            try:
+                data = self.get_app_version_details(application_id, application_version_id)
+            except ApiDetailsRequestError as e:
+                error_msg = f"Failed to fetch application version details for {application_id}/{application_version_id}\nDetails: {e}"
+                logger.error(error_msg)
+                raise ToolException(error_msg)
 
         if application_variables:
             for var in data.get('variables', {}):
@@ -204,15 +208,14 @@ class AlitaClient:
         elif app_type == 'autogen':
             app_type = "openai"
         if runtime == 'nonrunnable':
-            return LangChainAssistant(self, data, client, chat_history, app_type, 
+            return LangChainAssistant(self, data, client, chat_history, app_type,
                                       tools=tools, memory=memory)
         if runtime == 'langchain':
-            return LangChainAssistant(self, data, client, 
-                                      chat_history, app_type, 
+            return LangChainAssistant(self, data, client,
+                                      chat_history, app_type,
                                       tools=tools, memory=memory).runnable()
         elif runtime == 'llama':
             raise NotImplementedError("LLama runtime is not supported")
-        
 
     def datasource(self, datasource_id: int) -> AlitaDataSource:
         url = f"{self.datasources}/{datasource_id}"

alita_sdk/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,6 +41,22 @@ class McpServerTool(BaseTool):
                 field_type = float
             elif field_type == 'boolean':
                 field_type = bool
+            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'] 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
+                    None
+                )
+                if item_type is None:
+                    raise ValueError(f"Unsupported array item type: {item_schema['type']}")
+                field_type = list[item_type]
             else:
                 raise ValueError(f"Unsupported field type: {field_type}")
 

{alita_sdk-0.3.139.dist-info → alita_sdk-0.3.140b1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.139
+Version: 0.3.140b1
 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.139.dist-info → alita_sdk-0.3.140b1.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=iADTqIaVNmMefPBNH37zilPoX42CqcHZTtYBCS-lbtw,19413
+alita_sdk/clients/client.py,sha256=42g8Q7LosIxqGsxNRhNE0KF3VRPKCbVxkeQMVRnyP3Y,19562
 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
@@ -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=xcH9AiqfR2TYrwJ3Ixw-_A7XDodtJCnwmq1SsikXpYk,1930
+alita_sdk/tools/mcp_server_tool.py,sha256=8RcYSV_6FAEwFSh4KNpdb8rO9oFQ-RR6ufXiZOTgsHA,3476
 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.139.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.140b1.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.139.dist-info/METADATA,sha256=9ZEhLNU3RXXyKXJCqvTQIstIg22YoQYlRP24-ToFEB8,7076
-alita_sdk-0.3.139.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.139.dist-info/top_level.txt,sha256=SWRhxB7Et3cOy3RkE5hR7OIRnHoo3K8EXzoiNlkfOmc,25
-alita_sdk-0.3.139.dist-info/RECORD,,
+alita_sdk-0.3.140b1.dist-info/METADATA,sha256=mAGZFKCyqDPJoYal-kfCJV96nP9UYigBBybAWK8hxVI,7078
+alita_sdk-0.3.140b1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.140b1.dist-info/top_level.txt,sha256=SWRhxB7Et3cOy3RkE5hR7OIRnHoo3K8EXzoiNlkfOmc,25
+alita_sdk-0.3.140b1.dist-info/RECORD,,