alita-sdk 0.3.225__py3-none-any.whl → 0.3.227__py3-none-any.whl

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"""
@@ -161,7 +161,7 @@ class BitbucketAPIWrapper(BaseCodeToolApiWrapper):
             return result if isinstance(result, ToolException) else f"File has been updated: {file_path}."
         except Exception as e:
             return ToolException(f"File was not updated due to error: {str(e)}")
-    
+
     def get_pull_requests_commits(self, pr_id: str) -> List[Dict[str, Any]]:
         """
         Get commits from a pull request
@@ -175,7 +175,7 @@ class BitbucketAPIWrapper(BaseCodeToolApiWrapper):
             return result
         except Exception as e:
             return ToolException(f"Can't get commits from pull request `{pr_id}` due to error:\n{str(e)}")
-        
+
     def get_pull_requests(self) -> List[Dict[str, Any]]:
         """
         Get pull requests from the repository
@@ -183,7 +183,7 @@ class BitbucketAPIWrapper(BaseCodeToolApiWrapper):
             List[Dict[str, Any]]: List of pull requests in the repository
         """
         return self._bitbucket.get_pull_requests()
-    
+
     def get_pull_request(self, pr_id: str) -> Dict[str, Any]:
         """
         Get details of a pull request
@@ -196,7 +196,7 @@ class BitbucketAPIWrapper(BaseCodeToolApiWrapper):
             return self._bitbucket.get_pull_request(pr_id=pr_id)
         except Exception as e:
             return ToolException(f"Can't get pull request `{pr_id}` due to error:\n{str(e)}")
-        
+
     def get_pull_requests_changes(self, pr_id: str) -> Dict[str, Any]:
         """
         Get changes of a pull request
@@ -209,7 +209,7 @@ class BitbucketAPIWrapper(BaseCodeToolApiWrapper):
             return self._bitbucket.get_pull_requests_changes(pr_id=pr_id)
         except Exception as e:
             return ToolException(f"Can't get changes from pull request `{pr_id}` due to error:\n{str(e)}")
-        
+
     def add_pull_request_comment(self, pr_id: str, content, inline=None) -> str:
         """
         Add a comment to a pull request. Supports multiple content types and inline comments.

alita_sdk/tools/elitea_base.py

@@ -555,7 +555,7 @@ class BaseCodeToolApiWrapper(BaseVectorStoreToolApiWrapper):
         """
         from .chunkers.code.codeparser import parse_code_files_for_db
 
-        _files = self.__handle_get_files("", branch or self.active_branch)
+        _files = self.__handle_get_files("", branch or self.active_branch or self._active_branch)
 
         logger.info(f"Files in branch: {_files}")
 
@@ -573,8 +573,8 @@ class BaseCodeToolApiWrapper(BaseVectorStoreToolApiWrapper):
             for file in _files:
                 if is_whitelisted(file) and not is_blacklisted(file):
                     yield {"file_name": file,
-                           "file_content": self._read_file(file, branch=branch or self.active_branch),
-                           "commit_hash": self._file_commit_hash(file, branch=branch or self.active_branch)}
+                           "file_content": self._read_file(file, branch=branch or self.active_branch or self._active_branch),
+                           "commit_hash": self._file_commit_hash(file, branch=branch or self.active_branch or self._active_branch)}
 
         return parse_code_files_for_db(file_content_generator())
     

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
 
@@ -430,7 +429,6 @@ class GitLabAPIWrapper(BaseCodeToolApiWrapper):
             for commit in commits
         ]
 
-    @extend_with_vector_tools
     def get_available_tools(self):
         return [
             {

{alita_sdk-0.3.225.dist-info → alita_sdk-0.3.227.dist-info}/METADATA

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

@@ -94,7 +94,7 @@ alita_sdk/runtime/utils/toolkit_runtime.py,sha256=MU63Fpxj0b5_r1IUUc0Q3-PN9VwL7r
 alita_sdk/runtime/utils/toolkit_utils.py,sha256=I9QFqnaqfVgN26LUr6s3XlBlG6y0CoHURnCzG7XcwVs,5311
 alita_sdk/runtime/utils/utils.py,sha256=CpEl3LCeLbhzQySz08lkKPm7Auac6IiLF7WB8wmArMI,589
 alita_sdk/tools/__init__.py,sha256=1AHqP2xyLjn92xVm70l9XIke6FkfHkLo5OoQVe4BuP8,10421
-alita_sdk/tools/elitea_base.py,sha256=anEPuN9dtOsSTJQlHpap5ho_hf-iPmBQpw4tmNUTP5k,30475
+alita_sdk/tools/elitea_base.py,sha256=AmOks-hZfHFI2G0_fBh7Yyz3wjohyTA1sLxLEqW87EQ,30544
 alita_sdk/tools/ado/__init__.py,sha256=2NMQwt2pjIukSC9nSZ7CLocdGpK7002x7ixKr_wunxk,1313
 alita_sdk/tools/ado/utils.py,sha256=PTCludvaQmPLakF2EbCGy66Mro4-rjDtavVP-xcB2Wc,1252
 alita_sdk/tools/ado/repos/__init__.py,sha256=VvOapCG5okdyfCNdeTfOdcgRvl54cB1-sx_BF0TBWOw,6858
@@ -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=OU55KjtFalYIZ4ioeBck0zjqTewB6BdwQuAS3Kud4R0,10847
 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=KYCRO2pF8EPTLhWuEj64XsHPCYSucsf8S3R_ofJttrA,22301
 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
@@ -307,8 +307,8 @@ alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=JAeWf-RXohsxheUpT0iMDClc_izj-
 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.225.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.225.dist-info/METADATA,sha256=aEZ-H5sC3z9syNzA4TBqKOQuGhQzuwMc9OzBnttEu4E,18917
-alita_sdk-0.3.225.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.225.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.225.dist-info/RECORD,,
+alita_sdk-0.3.227.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.227.dist-info/METADATA,sha256=Rdq53_hxX1XkPsuzWiYs9_74Sxo6BzniJu9PvooRS9A,18917
+alita_sdk-0.3.227.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.227.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.227.dist-info/RECORD,,