alita-sdk 0.3.168__py3-none-any.whl → 0.3.170__py3-none-any.whl

alita_sdk/runtime/clients/client.py

@@ -162,7 +162,9 @@ class AlitaClient:
         resp = requests.patch(url, headers=self.headers, verify=False, json={'configurations': configs})
         if resp.ok:
             return resp.json()
-        raise ApiDetailsRequestError
+        logger.error(f"Failed to fetch application version details: {resp.status_code} - {resp.text}."
+                     f" Application ID: {application_id}, Version ID: {application_version_id}")
+        raise ApiDetailsRequestError(f"Failed to fetch application version details for {application_id}/{application_version_id}.")
 
     def get_integration_details(self, integration_id: str, format_for_model: bool = False):
         url = f"{self.integration_details}/{integration_id}"

alita_sdk/tools/__init__.py

@@ -54,7 +54,7 @@ _safe_import_tool('zephyr_enterprise', 'zephyr_enterprise', 'get_tools', 'Zephyr
 _safe_import_tool('ado', 'ado', 'get_tools')
 _safe_import_tool('ado_repos', 'ado.repos', 'get_tools', 'AzureDevOpsReposToolkit')
 _safe_import_tool('ado_plans', 'ado.test_plan', None, 'AzureDevOpsPlansToolkit')
-_safe_import_tool('ado_work_items', 'ado.work_item', None, 'AzureDevOpsWorkItemsToolkit')
+_safe_import_tool('ado_boards', 'ado.work_item', None, 'AzureDevOpsWorkItemsToolkit')
 _safe_import_tool('ado_wiki', 'ado.wiki', None, 'AzureDevOpsWikiToolkit')
 _safe_import_tool('rally', 'rally', 'get_tools', 'RallyToolkit')
 _safe_import_tool('sql', 'sql', 'get_tools', 'SQLToolkit')
@@ -97,17 +97,16 @@ def get_tools(tools_list, alita, llm, store: Optional[BaseStore] = None, *args,
         tool['settings']['llm'] = llm
         tool['settings']['store'] = store
         tool_type = tool['type']
-        
+
+        # Handle special cases for ADO tools
+        if tool_type in ['ado_boards', 'ado_wiki', 'ado_plans']:
+            tools.extend(AVAILABLE_TOOLS['ado']['get_tools'](tool_type, tool))
+
         # Check if tool is available and has get_tools function
-        if tool_type in AVAILABLE_TOOLS and 'get_tools' in AVAILABLE_TOOLS[tool_type]:
+        elif tool_type in AVAILABLE_TOOLS and 'get_tools' in AVAILABLE_TOOLS[tool_type]:
             try:
                 get_tools_func = AVAILABLE_TOOLS[tool_type]['get_tools']
-                
-                # Handle special cases for ADO tools
-                if tool_type in ['ado_boards', 'ado_wiki', 'ado_plans']:
-                    tools.extend(get_tools_func(tool_type, tool))
-                else:
-                    tools.extend(get_tools_func(tool))
+                tools.extend(get_tools_func(tool))
                     
             except Exception as e:
                 logger.error(f"Error getting tools for {tool_type}: {e}")

alita_sdk/tools/ado/repos/__init__.py

@@ -3,6 +3,8 @@ from typing import List, Literal, Optional
 from langchain_core.tools import BaseTool, BaseToolkit
 from pydantic import BaseModel, Field, create_model, SecretStr
 
+import requests
+from alita_tools.utils import check_connection_response
 from ...base.tool import BaseAction
 from .repos_wrapper import ReposApiWrapper
 from ...utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
@@ -43,7 +45,7 @@ class AzureDevOpsReposToolkit(BaseToolkit):
     def toolkit_config_schema() -> BaseModel:
         selected_tools = {x['name']: x['args_schema'].schema() for x in ReposApiWrapper.model_construct().get_available_tools()}
         AzureDevOpsReposToolkit.toolkit_max_length = get_max_toolkit_length(selected_tools)
-        return create_model(
+        m = create_model(
             name,
             organization_url=(Optional[str], Field(default="", title="Organization URL", description="ADO organization url")),
             project=(Optional[str], Field(default="", title="Project", description="ADO project")),
@@ -78,6 +80,18 @@ class AzureDevOpsReposToolkit(BaseToolkit):
                 }}}
         )
 
+        @check_connection_response
+        def check_connection(self):
+            response = requests.get(
+                f'{self.organization_url}/{self.project}/_apis/git/repositories/{self.repository_id}?api-version=7.0',
+                headers = {'Authorization': f'Bearer {self.token}'},
+                timeout=5
+            )
+            return response
+
+        m.check_connection = check_connection
+        return m
+
     @classmethod
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
         from os import environ

alita_sdk/tools/ado/repos/repos_wrapper.py

@@ -191,10 +191,14 @@ class ArgsSchema(Enum):
         "CreatePullRequest",
         pull_request_title=(str, Field(description="Title of the pull request")),
         pull_request_body=(str, Field(description="Body of the pull request")),
-        branch_name=(
+        target_branch=(
             str,
-            Field(description="The name of the branch, e.g. `my_branch`."),
+            Field(description="The name of the target branch, e.g. `my_branch`."),
         ),
+        source_branch=(
+            str,
+            Field(description="The name of the source branch, e.g. `feature_branch`."),
+        )
     )
     GetCommits = create_model(
         "GetCommits",
@@ -1041,7 +1045,7 @@ class ReposApiWrapper(BaseCodeToolApiWrapper):
             return ToolException(msg)
 
     def create_pr(
-            self, pull_request_title: str, pull_request_body: str, branch_name: str
+            self, pull_request_title: str, pull_request_body: str, target_branch: str, source_branch: str
     ) -> str:
         """
         Creates a pull request in Azure DevOps from the active branch to the base branch mentioned in params.
@@ -1049,18 +1053,19 @@ class ReposApiWrapper(BaseCodeToolApiWrapper):
         Parameters:
             pull_request_title (str): Title of the pull request.
             pull_request_body (str): Description/body of the pull request.
-            branch_name (str): The name of the branch which is used as target branch for pull request.
+            target_branch (str): The name of the branch which is used as target branch for pull request.
+            source_branch (str): The name of the source branch which is used as source branch for pull request.
 
         Returns:
             str: A success or failure message.
         """
-        if self.active_branch == branch_name:
-            return f"Cannot create a pull request because the source branch '{self.active_branch}' is the same as the target branch '{branch_name}'"
+        if source_branch == target_branch:
+            return f"Cannot create a pull request because the source branch '{source_branch}' is the same as the target branch '{target_branch}'"
 
         try:
             pull_request = {
-                "sourceRefName": f"refs/heads/{self.active_branch}",
-                "targetRefName": f"refs/heads/{branch_name}",
+                "sourceRefName": f"refs/heads/{source_branch}",
+                "targetRefName": f"refs/heads/{target_branch}",
                 "title": pull_request_title,
                 "description": pull_request_body,
                 "reviewers": [],

alita_sdk/tools/ado/work_item/__init__.py

@@ -6,8 +6,7 @@ from pydantic import create_model, BaseModel, Field, SecretStr
 from ...base.tool import BaseAction
 from ...utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
 
-name = "azure_devops_boards"
-name_alias = 'ado_boards'
+name = "ado_boards"
 
 class AzureDevOpsWorkItemsToolkit(BaseToolkit):
     tools: List[BaseTool] = []
@@ -18,7 +17,7 @@ class AzureDevOpsWorkItemsToolkit(BaseToolkit):
         selected_tools = {x['name']: x['args_schema'].schema() for x in AzureDevOpsApiWrapper.model_construct().get_available_tools()}
         AzureDevOpsWorkItemsToolkit.toolkit_max_length = get_max_toolkit_length(selected_tools)
         return create_model(
-            name_alias,
+            name,
             name=(str, Field(description="Toolkit name",
                              json_schema_extra={
                                  'toolkit_name': True,

{alita_sdk-0.3.168.dist-info → alita_sdk-0.3.170.dist-info}/METADATA

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

@@ -44,7 +44,7 @@ alita_sdk/community/deep_researcher/utils/os.py,sha256=Q1xX7c7_p7EmuzzXIAY9TDmra
 alita_sdk/runtime/__init__.py,sha256=4W0UF-nl3QF2bvET5lnah4o24CoTwSoKXhuN0YnwvEE,828
 alita_sdk/runtime/clients/__init__.py,sha256=BdehU5GBztN1Qi1Wul0cqlU46FxUfMnI6Vq2Zd_oq1M,296
 alita_sdk/runtime/clients/artifact.py,sha256=4N2t5x3GibyXLq3Fvrv2o_VA7Z000yNfc-UN4eGsHZg,2679
-alita_sdk/runtime/clients/client.py,sha256=jbC_M72CybwZgFfMRL6paj-NmICrSuk1vVnVTm_u-kc,19734
+alita_sdk/runtime/clients/client.py,sha256=6ezOJ92CSw6b2PVs4uFMQKQdp40uT1awoFEqWAfBH_A,20029
 alita_sdk/runtime/clients/datasource.py,sha256=HAZovoQN9jBg0_-lIlGBQzb4FJdczPhkHehAiVG3Wx0,1020
 alita_sdk/runtime/clients/prompt.py,sha256=li1RG9eBwgNK_Qf0qUaZ8QNTmsncFrAL2pv3kbxZRZg,1447
 alita_sdk/runtime/langchain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -129,17 +129,17 @@ alita_sdk/runtime/utils/logging.py,sha256=svPyiW8ztDfhqHFITv5FBCj8UhLxz6hWcqGIY6
 alita_sdk/runtime/utils/save_dataframe.py,sha256=i-E1wp-t4wb17Zq3nA3xYwgSILjoXNizaQAA9opWvxY,1576
 alita_sdk/runtime/utils/streamlit.py,sha256=z4J_bdxkA0zMROkvTB4u379YBRFCkKh-h7PD8RlnZWQ,85644
 alita_sdk/runtime/utils/utils.py,sha256=dM8whOJAuFJFe19qJ69-FLzrUp6d2G-G6L7d4ss2XqM,346
-alita_sdk/tools/__init__.py,sha256=iuEQDwXHY8MQvHt6qIkYpXD7VC_3no-iVgDCpHqwIy4,10018
+alita_sdk/tools/__init__.py,sha256=48DhEi14KkaYhNb-KvXuM9XJ4WGC-v9sRcWfN7GFWd4,9963
 alita_sdk/tools/elitea_base.py,sha256=NQaIxPX6DVIerHCb18jwUR6maZxxk73NZaTsFHkBQWE,21119
 alita_sdk/tools/ado/__init__.py,sha256=mD6GHcYMTtffPJkJvFPe2rzvye_IRmXmWfI7xYuZhO4,912
 alita_sdk/tools/ado/utils.py,sha256=PTCludvaQmPLakF2EbCGy66Mro4-rjDtavVP-xcB2Wc,1252
-alita_sdk/tools/ado/repos/__init__.py,sha256=U8MA5VA3hDRomcbbZ2bS8zzwnGYnAfOlLN3jfIOdWMU,5217
-alita_sdk/tools/ado/repos/repos_wrapper.py,sha256=RfhHr7aL1peIubW1FgL9r7p5WExzFfXzotuZmeCzlO8,47931
+alita_sdk/tools/ado/repos/__init__.py,sha256=1gieQi1YWGT8-1lXCLnYTqDoAD8Zlx09H1KzlLV9KpA,5715
+alita_sdk/tools/ado/repos/repos_wrapper.py,sha256=_OWKAls7VFfFtEPTwqj_DxE1MSvpC0ivxdTIULEz3Tk,48206
 alita_sdk/tools/ado/test_plan/__init__.py,sha256=ctn2CUkH_xS0Wkv2gLrO3FHQ36BvNCc0VcpMuddvU8E,3404
 alita_sdk/tools/ado/test_plan/test_plan_wrapper.py,sha256=oIvVhLUMP5ZGctoAtK6sU0y6Si9gNv9-mbLqcWtw3gY,12525
 alita_sdk/tools/ado/wiki/__init__.py,sha256=92AIAXVYSEYKnNvEG08W2YmR2lC35Bn92CiXJ8T3vpA,3736
 alita_sdk/tools/ado/wiki/ado_wrapper.py,sha256=l4bc2QoKSUXg9UqNcx0ylv7YL9JPPQd35Ti5MXyEgC4,12690
-alita_sdk/tools/ado/work_item/__init__.py,sha256=ZF6x0IWESIuwJ7jggM4o5koaZycks4lMrwdWlEzBEp0,3873
+alita_sdk/tools/ado/work_item/__init__.py,sha256=HcPX18wODKCn8oCnhs0DRXYKaRxiQ6GvEGgd1emHud8,3832
 alita_sdk/tools/ado/work_item/ado_wrapper.py,sha256=t0D9xubU0yy_JmRJ_zEtRCxwFLyanT1StbIrtHGaqpw,26108
 alita_sdk/tools/advanced_jira_mining/__init__.py,sha256=pUTzECqGvYaR5qWY3JPUhrImrZgc7pCXuqSe5eWIE80,4604
 alita_sdk/tools/advanced_jira_mining/data_mining_wrapper.py,sha256=nZPtuwVWp8VeHw1B8q9kdwf-6ZvHnlXTOGdcIMDkKpw,44211
@@ -326,8 +326,8 @@ alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=UHVQUVqcBc3SZvDfO78HSuBzwAsRw
 alita_sdk/tools/zephyr_squad/__init__.py,sha256=rq4jOb3lRW2GXvAguk4H1KinO5f-zpygzhBJf-E1Ucw,2773
 alita_sdk/tools/zephyr_squad/api_wrapper.py,sha256=iOMxyE7vOc_LwFB_nBMiSFXkNtvbptA4i-BrTlo7M0A,5854
 alita_sdk/tools/zephyr_squad/zephyr_squad_cloud_client.py,sha256=IYUJoMFOMA70knLhLtAnuGoy3OK80RuqeQZ710oyIxE,3631
-alita_sdk-0.3.168.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.168.dist-info/METADATA,sha256=IwdlbGZD0vCpKWXvl9TfWgiRZ7MovliOOva3Emi2mm4,18757
-alita_sdk-0.3.168.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.168.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.168.dist-info/RECORD,,
+alita_sdk-0.3.170.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.170.dist-info/METADATA,sha256=Djs8vf5jNVfyDSg3HQxk9usbGV1nfetOVO_JZyWOcW0,18757
+alita_sdk-0.3.170.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.170.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.170.dist-info/RECORD,,