alita-sdk 0.3.224__py3-none-any.whl → 0.3.226__py3-none-any.whl

alita_sdk/tools/ado/test_plan/test_plan_wrapper.py

@@ -364,7 +364,7 @@ class TestPlanApiWrapper(BaseVectorStoreToolApiWrapper):
             logger.error(f"Error getting test cases: {e}")
             return ToolException(f"Error getting test cases: {e}")
 
-    def _base_loader(self, plan_id: str, suite_ids: Optional[list[str]] = []) -> Generator[Document, None, None]:
+    def _base_loader(self, plan_id: str, suite_ids: Optional[list[str]] = [], **kwargs) -> Generator[Document, None, None]:
         cases = []
         for sid in suite_ids:
             cases.extend(self.get_test_cases(plan_id, sid))

alita_sdk/tools/ado/wiki/ado_wrapper.py

@@ -229,7 +229,7 @@ class AzureDevOpsApiWrapper(BaseVectorStoreToolApiWrapper):
             logger.error(f"Unable to modify wiki page: {str(e)}")
             return ToolException(f"Unable to modify wiki page: {str(e)}")
 
-    def _base_loader(self, wiki_identifier: str) -> Generator[Document, None, None]:
+    def _base_loader(self, wiki_identifier: str, **kwargs) -> Generator[Document, None, None]:
         pages = self._client.get_pages_batch(pages_batch_request={}, project=self.project, wiki_identifier=wiki_identifier)
         #
         for page in pages:

alita_sdk/tools/bitbucket/api_wrapper.py

@@ -23,7 +23,7 @@ class BitbucketAPIWrapper(BaseCodeToolApiWrapper):
     """Wrapper for Bitbucket API."""
 
     _bitbucket: Any = PrivateAttr()
-    _active_branch: Any = PrivateAttr() 
+    active_branch: Any = PrivateAttr()
     url: str = ''
     project: str = ''
     """The key of the project this repo belongs to"""
@@ -78,12 +78,12 @@ class BitbucketAPIWrapper(BaseCodeToolApiWrapper):
             project=values['project'],
             repository=values['repository']
         )
-        cls._active_branch = values.get('branch')
+        cls.active_branch = values.get('branch')
         return values
 
     def set_active_branch(self, branch: str) -> None:
         """Set the active branch for the bot."""
-        self._active_branch = branch
+        self.active_branch = branch
         return f"Active branch set to `{branch}`"
 
     def list_branches_in_repo(self) -> List[str]:
@@ -93,15 +93,15 @@ class BitbucketAPIWrapper(BaseCodeToolApiWrapper):
     def create_branch(self, branch_name: str) -> None:
         """Create a new branch in the repository."""
         try:
-            self._bitbucket.create_branch(branch_name, self._active_branch)
+            self._bitbucket.create_branch(branch_name, self.active_branch)
         except Exception as e:
             if "not permitted to access this resource" in str(e):
                 return f"Please, verify you token/password: {str}"
             if "already exists" in str(e):
-                self._active_branch = branch_name
+                self.active_branch = branch_name
                 return f"Branch {branch_name} already exists. set it as active"
             return f"Unable to create branch due to error:\n{e}"
-        self._active_branch = branch_name
+        self.active_branch = branch_name
         return f"Branch {branch_name} created successfully and set as active"
 
     def create_pull_request(self, pr_json_data: str) -> str:
@@ -234,7 +234,7 @@ class BitbucketAPIWrapper(BaseCodeToolApiWrapper):
         Returns:
             str: List of the files
         """
-        return str(self._bitbucket.get_files_list(file_path=path if path else '', branch=branch if branch else self._active_branch))
+        return str(self._bitbucket.get_files_list(file_path=path if path else '', branch=branch if branch else self.active_branch))
 
     # TODO: review this method, it may not work as expected
     # def _file_commit_hash(self, file_path: str, branch: str):

alita_sdk/tools/gitlab/api_wrapper.py

@@ -1,7 +1,6 @@
 # api_wrapper.py
 from typing import Any, Dict, List, Optional
 import fnmatch
-from alita_sdk.tools.elitea_base import extend_with_vector_tools
 from alita_sdk.tools.elitea_base import BaseCodeToolApiWrapper
 from pydantic import create_model, Field, model_validator, SecretStr, PrivateAttr
 
@@ -105,7 +104,7 @@ class GitLabAPIWrapper(BaseCodeToolApiWrapper):
     branch: Optional[str] = 'main'
     _git: Any = PrivateAttr()
     _repo_instance: Any = PrivateAttr()
-    _active_branch: Any = PrivateAttr()
+    active_branch: Any = PrivateAttr()
 
     llm: Optional[Any] = None
     # Alita instance
@@ -139,11 +138,11 @@ class GitLabAPIWrapper(BaseCodeToolApiWrapper):
         g.auth()
         cls._repo_instance = g.projects.get(values.get('repository'))
         cls._git = g
-        cls._active_branch = values.get('branch')
+        cls.active_branch = values.get('branch')
         return values
 
     def set_active_branch(self, branch_name: str) -> str:
-        self._active_branch = branch_name
+        self.active_branch = branch_name
         self._repo_instance.default_branch = branch_name
         return f"Active branch set to {branch_name}"
 
@@ -173,19 +172,19 @@ class GitLabAPIWrapper(BaseCodeToolApiWrapper):
             return f"Failed to list branches: {str(e)}"
 
     def list_files(self, path: str = None, recursive: bool = True, branch: str = None) -> List[str]:
-        branch = branch if branch else self._active_branch
+        branch = branch if branch else self.active_branch
         files = self._get_all_files(path, recursive, branch)
         paths = [file['path'] for file in files if file['type'] == 'blob']
         return paths
 
     def list_folders(self, path: str = None, recursive: bool = True, branch: str = None) -> List[str]:
-        branch = branch if branch else self._active_branch
+        branch = branch if branch else self.active_branch
         files = self._get_all_files(path, recursive, branch)
         paths = [file['path'] for file in files if file['type'] == 'tree']
         return paths
 
     def _get_all_files(self, path: str = None, recursive: bool = True, branch: str = None):
-        branch = branch if branch else self._active_branch
+        branch = branch if branch else self.active_branch
         return self._repo_instance.repository_tree(path=path, ref=branch, recursive=recursive, all=True)
 
     # overrided for indexer
@@ -211,15 +210,15 @@ class GitLabAPIWrapper(BaseCodeToolApiWrapper):
             self._repo_instance.branches.create(
                 {
                     'branch': branch_name,
-                    'ref': self._active_branch,
+                    'ref': self.active_branch,
                 }
             )
         except Exception as e:
             if "Branch already exists" in str(e):
-                self._active_branch = branch_name
+                self.active_branch = branch_name
                 return f"Branch {branch_name} already exists. set it as active"
             return f"Unable to create branch due to error:\n{e}"
-        self._active_branch = branch_name
+        self.active_branch = branch_name
         return f"Branch {branch_name} created successfully and set as active"
 
     def parse_issues(self, issues: List[Any]) -> List[dict]:
@@ -430,7 +429,6 @@ class GitLabAPIWrapper(BaseCodeToolApiWrapper):
             for commit in commits
         ]
 
-    @extend_with_vector_tools
     def get_available_tools(self):
         return [
             {

alita_sdk/tools/zephyr_enterprise/api_wrapper.py

@@ -153,10 +153,10 @@ class ZephyrApiWrapper(BaseVectorStoreToolApiWrapper):
         Returns a list of fields for index_data args schema.
         """
         return {
-            "zql": (str, Field(description=zql_description, examples=["folder=\"TestToolkit\"", "name~\"TestToolkit5\""))
+            "zql": (str, Field(description=zql_description, examples=["folder=\"TestToolkit\"", "name~\"TestToolkit5\""]))
         }
 
-    def _base_loader(self, zql: str) -> Generator[Document, None, None]:
+    def _base_loader(self, zql: str, **kwargs) -> Generator[Document, None, None]:
         test_cases = self.get_testcases_by_zql(zql)
         for test_case in test_cases:
             metadata = {

alita_sdk/tools/zephyr_essential/api_wrapper.py

@@ -229,7 +229,7 @@ class ZephyrEssentialApiWrapper(BaseVectorStoreToolApiWrapper):
         except json.JSONDecodeError as e:
             raise ValueError(f"Invalid JSON string: {str(e)}")
 
-    def _base_loader(self) -> Generator[Document, None, None]:
+    def _base_loader(self, **kwargs) -> Generator[Document, None, None]:
         try:
             test_cases = self.list_test_cases()
         except Exception as e:

alita_sdk/tools/zephyr_scale/api_wrapper.py

@@ -1213,7 +1213,7 @@ class ZephyrScaleApiWrapper(BaseVectorStoreToolApiWrapper):
             """))
         }
 
-    def _base_loader(self, project_key: str, jql: str) -> Generator[Document, None, None]:
+    def _base_loader(self, project_key: str, jql: str, **kwargs) -> Generator[Document, None, None]:
         for test_case_doc in self._get_test_cases_docs(project_key, jql):
             yield test_case_doc
         for folder_doc in self._get_folders_docs(project_key):

{alita_sdk-0.3.224.dist-info → alita_sdk-0.3.226.dist-info}/METADATA

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

@@ -100,9 +100,9 @@ alita_sdk/tools/ado/utils.py,sha256=PTCludvaQmPLakF2EbCGy66Mro4-rjDtavVP-xcB2Wc,
 alita_sdk/tools/ado/repos/__init__.py,sha256=VvOapCG5okdyfCNdeTfOdcgRvl54cB1-sx_BF0TBWOw,6858
 alita_sdk/tools/ado/repos/repos_wrapper.py,sha256=SgnEUX1lTn5ncZI8EFfPibk0_gHstkApDSqTCxYsrUk,50097
 alita_sdk/tools/ado/test_plan/__init__.py,sha256=QUNSOCVheUbILve4CxjBzBHKcIA28jZjoA6rRb8tD_Y,5483
-alita_sdk/tools/ado/test_plan/test_plan_wrapper.py,sha256=MIEgTcjXQ_Hmx9Thzb1VXrviiB-nusFe-ym3BYlwa_4,20175
+alita_sdk/tools/ado/test_plan/test_plan_wrapper.py,sha256=WWSWQx97sQfkM_CALybj9BPmPelozdRlxwefbkTgxis,20185
 alita_sdk/tools/ado/wiki/__init__.py,sha256=3cPNqUyOFwnZ2g5QW8wsdBC9sB0u39F-q4KKVjQqTN8,5351
-alita_sdk/tools/ado/wiki/ado_wrapper.py,sha256=JfEbq2OBVtQe-6igYj09ucxwHEl34SozPIJ6gRGWcU8,14157
+alita_sdk/tools/ado/wiki/ado_wrapper.py,sha256=HoM2U2G8sKM8Hh6HpTFHdTiKAu4vLigB8VwE0_QtMhw,14167
 alita_sdk/tools/ado/work_item/__init__.py,sha256=dBDjP8CJnKe9xBbJg-EG1AvRo06mcfS2f0AVeXMaBmk,5507
 alita_sdk/tools/ado/work_item/ado_wrapper.py,sha256=7YrqNIDnVYYm8__joABAr3iIk3IvYoQ7NZ9Vi-iaICc,28233
 alita_sdk/tools/advanced_jira_mining/__init__.py,sha256=pUTzECqGvYaR5qWY3JPUhrImrZgc7pCXuqSe5eWIE80,4604
@@ -118,7 +118,7 @@ alita_sdk/tools/azure_ai/search/api_wrapper.py,sha256=E4p6HPDlwgxfT_i6cvg9rN4Vn_
 alita_sdk/tools/base/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/tools/base/tool.py,sha256=-N27AodZS49vdPCgFkU-bFS9bxoPopZBnNrmwInx3d0,864
 alita_sdk/tools/bitbucket/__init__.py,sha256=EEJUp965OT0JijI6x7U7DP-_5sgRC3Q3TbNnxky6hPo,5458
-alita_sdk/tools/bitbucket/api_wrapper.py,sha256=wZp00LAxRs-nsMAduXUHTGGV41ayThpAKqGKmm4g27U,10880
+alita_sdk/tools/bitbucket/api_wrapper.py,sha256=Sww4j67Ib5owH5-bKcSadikIQ9BJpzQXMVToku3Wpc0,10872
 alita_sdk/tools/bitbucket/bitbucket_constants.py,sha256=UsbhQ1iEvrKoxceTFPWTYhaXS1zSxbmjs1TwY0-P4gw,462
 alita_sdk/tools/bitbucket/cloud_api_wrapper.py,sha256=VELi65tLXvszwCGQSqVfyVal0ylx9DgAmAGpRQL_Zkg,15522
 alita_sdk/tools/bitbucket/tools.py,sha256=zKBUq7t9zLa1EvhlVZzyVcZSvwvdcbtz0oslgPFZeeo,15307
@@ -203,7 +203,7 @@ alita_sdk/tools/github/schemas.py,sha256=yFsqivfjCPRk9GxFJrL8sTz6nnjFCZ0j5DIfPtG
 alita_sdk/tools/github/tool.py,sha256=Jnnv5lenV5ds8AAdyo2m8hSzyJ117HZBjzHC6T1ck-M,1037
 alita_sdk/tools/github/tool_prompts.py,sha256=y6ZW_FpUCE87Uop3WuQAZVRnzxO5t7xjBOI5bCqiluw,30194
 alita_sdk/tools/gitlab/__init__.py,sha256=DftTt4uVaRG4g2RFvLKyghG_A28ukGPHK9G2jojEGnM,4853
-alita_sdk/tools/gitlab/api_wrapper.py,sha256=kAGHmne6Wh8WrQQUdZ1d8bhgzgmN_fAz49WZsX7K1Ps,22396
+alita_sdk/tools/gitlab/api_wrapper.py,sha256=SG8YL92ksTqM-lHgJL3XyycbvMb-DDkaCqdh0iwaY1Y,22292
 alita_sdk/tools/gitlab/tools.py,sha256=vOGTlSaGaFmWn6LS6YFP-FuTqUPun9vnv1VrUcUHAZQ,16500
 alita_sdk/tools/gitlab/utils.py,sha256=Z2XiqIg54ouqqt1to-geFybmkCb1I6bpE91wfnINH1I,2320
 alita_sdk/tools/gitlab_org/__init__.py,sha256=_DJ5y92E6GuNBtCuaEZakGNInxbuFtvLYZMTMCDf3Js,3713
@@ -297,18 +297,18 @@ alita_sdk/tools/zephyr/__init__.py,sha256=8B2Ibz5QTmB5WkV0q8Sq4kuj92FFaFWZLrT877
 alita_sdk/tools/zephyr/api_wrapper.py,sha256=lJCYPG03ej0qgdpLflnS7LFB4HSAfGzIvTjAJt07CQs,6244
 alita_sdk/tools/zephyr/rest_client.py,sha256=7vSD3oYIX-3KbAFed-mphSQif_VRuXrq5O07ryNQ7Pk,6208
 alita_sdk/tools/zephyr_enterprise/__init__.py,sha256=XAWLj4WgMpEz3xnP8VW1lzHVW48K3CxIXdgeubjMvaw,3885
-alita_sdk/tools/zephyr_enterprise/api_wrapper.py,sha256=arPAvOGrurEsZaICvdVrcbspXBdDQl4XuNdfZSmm1lg,11204
+alita_sdk/tools/zephyr_enterprise/api_wrapper.py,sha256=p9EpkO5tif3JJzprz2_VuLsQ1yET7TwwBfPOKJGwt9c,11215
 alita_sdk/tools/zephyr_enterprise/zephyr_enterprise.py,sha256=hV9LIrYfJT6oYp-ZfQR0YHflqBFPsUw2Oc55HwK0H48,6809
 alita_sdk/tools/zephyr_essential/__init__.py,sha256=-VS-smytUSjqNLLi6HmqbsU3NfmNFTrzPjzNiEtOi5c,3628
-alita_sdk/tools/zephyr_essential/api_wrapper.py,sha256=zjZnAx1Fl1fYx5Vc1Q8JOc_pxUEkCngbRzzmea4sj7Y,36785
+alita_sdk/tools/zephyr_essential/api_wrapper.py,sha256=TpNov35XPgjM9eymCEFqv22mbpdVvLMBTb9WVqUcvNA,36795
 alita_sdk/tools/zephyr_essential/client.py,sha256=bfNcUKNqj9MFWTludGbbqD4qZlxrBaC2JtWsCfZMqSY,9722
 alita_sdk/tools/zephyr_scale/__init__.py,sha256=2NTcdrfkx4GSegqyXhsPLsEpc4FlACuDy85b0fk6cAo,4572
-alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=ZpHVRpeWDE-kxOf491vHgNesozuwEVS4oB0cnPs4XMw,78005
+alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=JAeWf-RXohsxheUpT0iMDClc_izj-zxMwafXCW4jtC0,78015
 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.224.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.224.dist-info/METADATA,sha256=lmtfgiN8O2B_M8Sld4D1uGYOr5g5aOyESKCjt-_b9uc,18917
-alita_sdk-0.3.224.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.224.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.224.dist-info/RECORD,,
+alita_sdk-0.3.226.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.226.dist-info/METADATA,sha256=mbKuIG_1KR0BDC7Q8-c34XlXzugfo5k1Z2L7duhJebs,18917
+alita_sdk-0.3.226.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.226.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.226.dist-info/RECORD,,