alita-sdk 0.3.225__tar.gz → 0.3.226__tar.gz

{alita_sdk-0.3.225 → alita_sdk-0.3.226}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.225
+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.225 → alita_sdk-0.3.226}/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-0.3.225 → alita_sdk-0.3.226}/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-0.3.225 → alita_sdk-0.3.226}/alita_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.225
+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.225 → alita_sdk-0.3.226}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "alita_sdk"
-version = "0.3.225"
+version = "0.3.226"
 description = "SDK for building langchain agents using resources from Alita"
 readme = "README.md"
 requires-python = ">=3.10"