alita-sdk 0.3.260__py3-none-any.whl → 0.3.262__py3-none-any.whl

alita_sdk/runtime/clients/client.py

@@ -20,6 +20,7 @@ from .prompt import AlitaPrompt
 from .datasource import AlitaDataSource
 from .artifact import Artifact
 from ..langchain.chat_message_template import Jinja2TemplatedChatMessagesTemplate
+from ..utils.utils import TOOLKIT_SPLITTER
 from ...tools import get_available_toolkit_models
 
 logger = logging.getLogger(__name__)
@@ -762,7 +763,7 @@ class AlitaClient:
                             base_available_tools.append(base_name)
 
                         # Track full names separately
-                        if '___' in tool_name_attr:
+                        if TOOLKIT_SPLITTER in tool_name_attr:
                             full_available_tools.append(tool_name_attr)
 
                 # Create comprehensive error message

alita_sdk/runtime/langchain/assistant.py

@@ -5,6 +5,7 @@ from typing import Any, Optional
 from langchain.agents import (
     AgentExecutor, create_openai_tools_agent,
     create_json_chat_agent)
+from langgraph.graph.state import CompiledStateGraph
 from langgraph.store.base import BaseStore
 from .agents.xml_chat import create_xml_chat_agent
 from .langraph_agent import create_graph
@@ -61,11 +62,11 @@ class Assistant:
         #     )
         #     self.client = target_cls(**model_params)
         # validate agents compatibility: non-pipeline agents cannot have pipelines as toolkits
-        if app_type not in ["pipeline", "predict"]:
-            tools_to_check = data.get('tools', [])
-            if any(tool['agent_type'] == 'pipeline' for tool in tools_to_check):
-                raise ToolException("Non-pipeline agents cannot have pipelines as a toolkits. "
-                                    "Review toolkits configuration or use pipeline as master agent.")
+        # if app_type not in ["pipeline", "predict"]:
+        #     tools_to_check = data.get('tools', [])
+        #     if any(tool['agent_type'] == 'pipeline' for tool in tools_to_check):
+        #         raise ToolException("Non-pipeline agents cannot have pipelines as a toolkits. "
+        #                             "Review toolkits configuration or use pipeline as master agent.")
 
         # configure memory store if memory tool is defined (not needed for predict agents)
         if app_type != "predict":
@@ -148,20 +149,20 @@ class Assistant:
 
     def getAgentExecutor(self):
         # Exclude compiled graph runnables from simple tool agents
-        simple_tools = [t for t in self.tools if isinstance(t, BaseTool)]
+        simple_tools = [t for t in self.tools if isinstance(t, (BaseTool, CompiledStateGraph))]
         agent = create_json_chat_agent(llm=self.client, tools=simple_tools, prompt=self.prompt)
         return self._agent_executor(agent)
 
 
     def getXMLAgentExecutor(self):
         # Exclude compiled graph runnables from simple tool agents
-        simple_tools = [t for t in self.tools if isinstance(t, BaseTool)]
+        simple_tools = [t for t in self.tools if isinstance(t, (BaseTool, CompiledStateGraph))]
         agent = create_xml_chat_agent(llm=self.client, tools=simple_tools, prompt=self.prompt)
         return self._agent_executor(agent)
 
     def getOpenAIToolsAgentExecutor(self):
         # Exclude compiled graph runnables from simple tool agents
-        simple_tools = [t for t in self.tools if isinstance(t, BaseTool)]
+        simple_tools = [t for t in self.tools if isinstance(t, (BaseTool, CompiledStateGraph))]
         agent = create_openai_tools_agent(llm=self.client, tools=simple_tools, prompt=self.prompt)
         return self._agent_executor(agent)
 
@@ -171,7 +172,7 @@ class Assistant:
         This creates a proper LangGraphAgentRunnable with modern tool support.
         """
         # Exclude compiled graph runnables from simple tool agents
-        simple_tools = [t for t in self.tools if isinstance(t, BaseTool)]
+        simple_tools = [t for t in self.tools if isinstance(t, (BaseTool, CompiledStateGraph))]
         
         # Set up memory/checkpointer if available
         checkpointer = None

alita_sdk/runtime/toolkits/tools.py

@@ -77,7 +77,6 @@ def get_tools(tools_list: list, alita_client, llm, memory_store: BaseStore = Non
                 connection_string=tool['settings'].get('connection_string', None),
                 collection_name=tool.get('toolkit_name'),
                 embedding_model=tool['settings'].get('embedding_model'),
-                embedding_model_params=tool['settings'].get('embedding_model_params', None),
                 vectorstore_type="PGVector"
             ).get_tools())
         elif tool['type'] == 'vectorstore':

alita_sdk/runtime/tools/vectorstore_base.py

@@ -421,7 +421,7 @@ class VectorStoreWrapperBase(BaseToolApiWrapper):
                         for result in text_results:
                             doc_id = result['id']
                             text_score = result['text_score']
-                            
+
                             if doc_id in doc_map:
                                 # Document exists in vector results, combine scores
                                 doc, vector_score = doc_map[doc_id]
@@ -452,7 +452,7 @@ class VectorStoreWrapperBase(BaseToolApiWrapper):
         
         # Apply cutoff filter
         if cut_off:
-            combined_items = [item for item in combined_items if abs(item[1]) >= cut_off]
+            combined_items = [item for item in combined_items if abs(item[1]) <= cut_off]
         
         # Sort by score and limit results
         # DISABLED: for chroma we want ascending order (lower score is better), for others descending

alita_sdk/tools/bitbucket/__init__.py

@@ -3,7 +3,6 @@ from typing import Dict, List, Literal, Optional
 from requests.auth import HTTPBasicAuth
 
 from .api_wrapper import BitbucketAPIWrapper
-from .tools import __all__
 from langchain_core.tools import BaseToolkit
 from langchain_core.tools import BaseTool
 from pydantic import BaseModel, Field, ConfigDict, create_model
@@ -41,10 +40,8 @@ class AlitaBitbucketToolkit(BaseToolkit):
 
     @staticmethod
     def toolkit_config_schema() -> BaseModel:
-        selected_tools = {}
-        for t in __all__:
-            default = t['tool'].__pydantic_fields__['args_schema'].default
-            selected_tools[t['name']] = default.schema() if default else default
+        selected_tools = {x['name']: x['args_schema'].schema() for x in
+                          BitbucketAPIWrapper.model_construct().get_available_tools()}
         AlitaBitbucketToolkit.toolkit_max_length = get_max_toolkit_length(selected_tools)
         m = create_model(
             name,
@@ -98,7 +95,7 @@ class AlitaBitbucketToolkit(BaseToolkit):
             **(kwargs.get('pgvector_configuration') or {}),
         }
         bitbucket_api_wrapper = BitbucketAPIWrapper(**wrapper_payload)
-        available_tools: List[Dict] = __all__
+        available_tools: List[Dict] = bitbucket_api_wrapper.get_available_tools()
         prefix = clean_string(toolkit_name, cls.toolkit_max_length) + TOOLKIT_SPLITTER if toolkit_name else ''
         tools = []
         for tool in available_tools:

alita_sdk/tools/bitbucket/api_wrapper.py

@@ -2,21 +2,119 @@
 from __future__ import annotations
 
 import logging
-from typing import TYPE_CHECKING, Any, Dict, List, Optional
+from typing import Any, Dict, List, Optional
+import fnmatch
 
 from langchain_core.tools import ToolException
-from pydantic import model_validator, SecretStr
+from pydantic import model_validator, SecretStr, create_model, Field
 from .bitbucket_constants import create_pr_data
 from .cloud_api_wrapper import BitbucketCloudApi, BitbucketServerApi
 from pydantic.fields import PrivateAttr
 
-from ..elitea_base import BaseCodeToolApiWrapper
+from ..elitea_base import BaseCodeToolApiWrapper, extend_with_vector_tools
 
 logger = logging.getLogger(__name__)
 
+# Pydantic model definitions for tool arguments
+CreateBranchModel = create_model(
+    "CreateBranchModel",
+    branch_name=(str, Field(description="The name of the branch, e.g. `my_branch`.")),
+)
 
-if TYPE_CHECKING:
-    pass
+CreatePullRequestModel = create_model(
+    "CreatePullRequestModel",
+    pr_json_data=(str, Field(description=create_pr_data)),
+)
+
+CreateFileModel = create_model(
+    "CreateFileModel",
+    file_path=(str, Field(description="The path of the file")),
+    file_contents=(str, Field(description="The contents of the file")),
+    branch=(str, Field(description="The branch to create the file in")),
+)
+
+UpdateFileModel = create_model(
+    "UpdateFileModel",
+    file_path=(str, Field(description="The path of the file")),
+    update_query=(str, Field(description="Contains the file contents required to be updated. "
+                                       "The old file contents is wrapped in OLD <<<< and >>>> OLD. "
+                                       "The new file contents is wrapped in NEW <<<< and >>>> NEW")),
+    branch=(str, Field(description="The branch to update the file in")),
+)
+
+ReadFileModel = create_model(
+    "ReadFileModel",
+    file_path=(str, Field(description="The path of the file")),
+    branch=(str, Field(description="The branch to read the file from")),
+)
+
+SetActiveBranchModel = create_model(
+    "SetActiveBranchModel",
+    branch_name=(str, Field(description="The name of the branch, e.g. `my_branch`.")),
+)
+
+ListBranchesInRepoModel = create_model(
+    "ListBranchesInRepoModel",
+    limit=(Optional[int], Field(default=20, description="Maximum number of branches to return. If not provided, all branches will be returned.")),
+    branch_wildcard=(Optional[str], Field(default=None, description="Wildcard pattern to filter branches by name. If not provided, all branches will be returned."))
+)
+
+ListFilesModel = create_model(
+    "ListFilesModel",
+    path=(Optional[str], Field(description="The path to list files from")),
+    recursive=(bool, Field(description="Whether to list files recursively", default=True)),
+    branch=(Optional[str], Field(description="The branch to list files from")),
+)
+
+GetPullRequestsCommitsModel = create_model(
+    "GetPullRequestsCommitsModel",
+    pr_id=(str, Field(description="The ID of the pull request to get commits from")),
+)
+
+GetPullRequestModel = create_model(
+    "GetPullRequestModel",
+    pr_id=(str, Field(description="The ID of the pull request to get details from")),
+)
+
+GetPullRequestsChangesModel = create_model(
+    "GetPullRequestsChangesModel",
+    pr_id=(str, Field(description="The ID of the pull request to get changes from")),
+)
+
+AddPullRequestCommentModel = create_model(
+    "AddPullRequestCommentModel",
+    pr_id=(str, Field(description="The ID of the pull request to add a comment to")),
+    content=(str, Field(description="The comment content")),
+    inline=(Optional[dict], Field(default=None, description="Inline comment details. Example: {'from': 57, 'to': 122, 'path': '<string>'}"))
+)
+
+DeleteFileModel = create_model(
+    "DeleteFileModel",
+    file_path=(str, Field(description="The path of the file")),
+    branch=(str, Field(description="The branch to delete the file from")),
+    commit_message=(str, Field(default=None, description="Commit message for deleting the file. Optional.")),
+)
+
+AppendFileModel = create_model(
+    "AppendFileModel",
+    file_path=(str, Field(description="The path of the file")),
+    content=(str, Field(description="The content to append to the file")),
+    branch=(str, Field(description="The branch to append the file in")),
+)
+
+GetIssuesModel = create_model(
+    "GetIssuesModel",
+)
+
+GetIssueModel = create_model(
+    "GetIssueModel",
+    issue_number=(int, Field(description="The number of the issue")),
+)
+
+CommentOnIssueModel = create_model(
+    "CommentOnIssueModel",
+    comment_query=(str, Field(description="The comment query string")),
+)
 
 
 class BitbucketAPIWrapper(BaseCodeToolApiWrapper):
@@ -41,18 +139,6 @@ class BitbucketAPIWrapper(BaseCodeToolApiWrapper):
     """Bitbucket installation type: true for cloud, false for server.
     """
 
-    llm: Optional[Any] = None
-    # Alita instance
-    alita: Optional[Any] = None
-
-    # Vector store configuration
-    connection_string: Optional[SecretStr] = None
-    collection_name: Optional[str] = None
-    doctype: Optional[str] = 'code'
-    embedding_model: Optional[str] = "HuggingFaceEmbeddings"
-    embedding_model_params: Optional[Dict[str, Any]] = {"model_name": "sentence-transformers/all-MiniLM-L6-v2"}
-    vectorstore_type: Optional[str] = "PGVector"
-
     @model_validator(mode='before')
     @classmethod
     def validate_env(cls, values: Dict) -> Dict:
@@ -83,14 +169,34 @@ class BitbucketAPIWrapper(BaseCodeToolApiWrapper):
         cls._active_branch = values.get('branch')
         return values
 
-    def set_active_branch(self, branch: str) -> None:
+    def set_active_branch(self, branch_name: str) -> str:
         """Set the active branch for the bot."""
-        self._active_branch = branch
-        return f"Active branch set to `{branch}`"
+        self._active_branch = branch_name
+        return f"Active branch set to `{branch_name}`"
 
-    def list_branches_in_repo(self) -> List[str]:
-        """List all branches in the repository."""
-        return self._bitbucket.list_branches()
+    def list_branches_in_repo(self, limit: Optional[int] = 20, branch_wildcard: Optional[str] = None) -> List[str]:
+        """
+        Lists branches in the repository with optional limit and wildcard filtering.
+
+        Parameters:
+            limit (Optional[int]): Maximum number of branches to return
+            branch_wildcard (Optional[str]): Wildcard pattern to filter branches (e.g., '*dev')
+
+        Returns:
+            List[str]: List containing names of branches
+        """
+        try:
+            branches = self._bitbucket.list_branches()
+
+            if branch_wildcard:
+                branches = [branch for branch in branches if fnmatch.fnmatch(branch, branch_wildcard)]
+
+            if limit is not None:
+                branches = branches[:limit]
+
+            return branches
+        except Exception as e:
+            return f"Failed to list branches: {str(e)}"
 
     def create_branch(self, branch_name: str) -> None:
         """Create a new branch in the repository."""
@@ -266,3 +372,106 @@ class BitbucketAPIWrapper(BaseCodeToolApiWrapper):
             return self._bitbucket.get_file(file_path=file_path, branch=branch)
         except Exception as e:
             raise ToolException(f"Can't extract file content (`{file_path}`) due to error:\n{str(e)}")
+
+    def list_files(self, path: str = None, recursive: bool = True, branch: str = None) -> List[str]:
+        """List files in the repository with optional path, recursive search, and branch."""
+        branch = branch if branch else self._active_branch
+        try:
+            files_str = self._get_files(path, branch)
+            # Parse the string response to extract file paths
+            # This is a simplified implementation - might need adjustment based on actual response format
+            import ast
+            try:
+                files_list = ast.literal_eval(files_str)
+                if isinstance(files_list, list):
+                    return files_list
+                else:
+                    return [str(files_list)]
+            except:
+                return [files_str] if files_str else []
+        except Exception as e:
+            return f"Failed to list files: {str(e)}"
+
+    def read_file(self, file_path: str, branch: str) -> str:
+        """Read the contents of a file in the repository."""
+        try:
+            return self._read_file(file_path, branch)
+        except Exception as e:
+            return f"Failed to read file {file_path}: {str(e)}"
+
+    @extend_with_vector_tools
+    def get_available_tools(self):
+        return [
+            {
+                "name": "create_branch",
+                "ref": self.create_branch,
+                "description": self.create_branch.__doc__ or "Create a new branch in the repository.",
+                "args_schema": CreateBranchModel,
+            },
+            {
+                "name": "list_branches_in_repo",
+                "ref": self.list_branches_in_repo,
+                "description": self.list_branches_in_repo.__doc__ or "List branches in the repository with optional limit and wildcard filtering.",
+                "args_schema": ListBranchesInRepoModel,
+            },
+            {
+                "name": "list_files",
+                "ref": self.list_files,
+                "description": self.list_files.__doc__ or "List files in the repository with optional path, recursive search, and branch.",
+                "args_schema": ListFilesModel,
+            },
+            {
+                "name": "create_pull_request",
+                "ref": self.create_pull_request,
+                "description": self.create_pull_request.__doc__ or "Create a pull request in the repository.",
+                "args_schema": CreatePullRequestModel,
+            },
+            {
+                "name": "create_file",
+                "ref": self.create_file,
+                "description": self.create_file.__doc__ or "Create a new file in the repository.",
+                "args_schema": CreateFileModel,
+            },
+            {
+                "name": "read_file",
+                "ref": self.read_file,
+                "description": self.read_file.__doc__ or "Read the contents of a file in the repository.",
+                "args_schema": ReadFileModel,
+            },
+            {
+                "name": "update_file",
+                "ref": self.update_file,
+                "description": self.update_file.__doc__ or "Update the contents of a file in the repository.",
+                "args_schema": UpdateFileModel,
+            },
+            {
+                "name": "set_active_branch",
+                "ref": self.set_active_branch,
+                "description": self.set_active_branch.__doc__ or "Set the active branch in the repository.",
+                "args_schema": SetActiveBranchModel,
+            },
+            {
+                "name": "get_pull_requests_commits",
+                "ref": self.get_pull_requests_commits,
+                "description": self.get_pull_requests_commits.__doc__ or "Get commits from a pull request in the repository.",
+                "args_schema": GetPullRequestsCommitsModel,
+            },
+            {
+                "name": "get_pull_request",
+                "ref": self.get_pull_request,
+                "description": self.get_pull_request.__doc__ or "Get details of a pull request in the repository.",
+                "args_schema": GetPullRequestModel,
+            },
+            {
+                "name": "get_pull_requests_changes",
+                "ref": self.get_pull_requests_changes,
+                "description": self.get_pull_requests_changes.__doc__ or "Get changes from a pull request in the repository.",
+                "args_schema": GetPullRequestsChangesModel,
+            },
+            {
+                "name": "add_pull_request_comment",
+                "ref": self.add_pull_request_comment,
+                "description": self.add_pull_request_comment.__doc__ or "Add a comment to a pull request in the repository.",
+                "args_schema": AddPullRequestCommentModel,
+            },
+        ]

alita_sdk/tools/utils/content_parser.py

@@ -142,7 +142,7 @@ def load_content(file_path: str, extension: str = None, loader_extra_config: dic
         if "file_path" in loader_kwargs:
             del loader_kwargs["file_path"]
 
-        loader = loader_cls(file_path, **loader_kwargs)
+        loader = loader_cls(file_path=file_path, **loader_kwargs)
         documents = loader.load()
 
         page_contents = [doc.page_content for doc in documents]

{alita_sdk-0.3.260.dist-info → alita_sdk-0.3.262.dist-info}/METADATA

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

@@ -26,11 +26,11 @@ alita_sdk/configurations/zephyr_enterprise.py,sha256=5W1QEcv62Y5Rk_kApI2QmOwvWZe
 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=H3pJAh5G-zWVyJ6YbqHGk4jA8U6HfacQduiTivpJZ3Y,3210
-alita_sdk/runtime/clients/client.py,sha256=ZkXP3-e785EyYoJLNowAGV3bRqv3fs1XqD_iuqqkg9Y,43513
+alita_sdk/runtime/clients/client.py,sha256=13_Ht9MnRO-9V1uP9_YTxM07g7mpm2uVevVh2wDPOnE,43567
 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
-alita_sdk/runtime/langchain/assistant.py,sha256=Bn9vUyZlFAP-D9Bh3zc2G1ZQkh5rr2c2g5t1WAQW6Hw,13388
+alita_sdk/runtime/langchain/assistant.py,sha256=suBFEt24t9bLyBHMzkR3Mkgd9HIrGBq_bzcMiENiEr4,13539
 alita_sdk/runtime/langchain/chat_message_template.py,sha256=kPz8W2BG6IMyITFDA5oeb5BxVRkHEVZhuiGl4MBZKdc,2176
 alita_sdk/runtime/langchain/constants.py,sha256=eHVJ_beJNTf1WJo4yq7KMK64fxsRvs3lKc34QCXSbpk,3319
 alita_sdk/runtime/langchain/indexer.py,sha256=0ENHy5EOhThnAiYFc7QAsaTNp9rr8hDV_hTK8ahbatk,37592
@@ -88,7 +88,7 @@ alita_sdk/runtime/toolkits/configurations.py,sha256=kIDAlnryPQfbZyFxV-9SzN2-Vefz
 alita_sdk/runtime/toolkits/datasource.py,sha256=qk78OdPoReYPCWwahfkKLbKc4pfsu-061oXRryFLP6I,2498
 alita_sdk/runtime/toolkits/prompt.py,sha256=WIpTkkVYWqIqOWR_LlSWz3ug8uO9tm5jJ7aZYdiGRn0,1192
 alita_sdk/runtime/toolkits/subgraph.py,sha256=ZYqI4yVLbEPAjCR8dpXbjbL2ipX598Hk3fL6AgaqFD4,1758
-alita_sdk/runtime/toolkits/tools.py,sha256=kOoWkG3MQtphc7WHqtbf7WTTZ1e-kjmlrdjcAEeoHnw,7999
+alita_sdk/runtime/toolkits/tools.py,sha256=2WJNELUOjEL1v0LKANJ4crB2Kow3pfI03X1n30zUtTM,7906
 alita_sdk/runtime/toolkits/vectorstore.py,sha256=BGppQADa1ZiLO17fC0uCACTTEvPHlodEDYEzUcBRbAA,2901
 alita_sdk/runtime/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/runtime/tools/agent.py,sha256=m98QxOHwnCRTT9j18Olbb5UPS8-ZGeQaGiUyZJSyFck,3162
@@ -107,7 +107,7 @@ alita_sdk/runtime/tools/prompt.py,sha256=nJafb_e5aOM1Rr3qGFCR-SKziU9uCsiP2okIMs9
 alita_sdk/runtime/tools/router.py,sha256=wCvZjVkdXK9dMMeEerrgKf5M790RudH68pDortnHSz0,1517
 alita_sdk/runtime/tools/tool.py,sha256=lE1hGi6qOAXG7qxtqxarD_XMQqTghdywf261DZawwno,5631
 alita_sdk/runtime/tools/vectorstore.py,sha256=yl6FKJGVQDevftSkxWTkMbqjIskIFz69vXELdEGp9u4,34780
-alita_sdk/runtime/tools/vectorstore_base.py,sha256=Ko-82amS4mBoJqX-v6h-iUrQ5LNzMvM978q2i8TcfVw,27409
+alita_sdk/runtime/tools/vectorstore_base.py,sha256=OM9nMUzQ7SgfQD8QYlzGYLXzKuGMZ1onpqSrBx9vMKk,27381
 alita_sdk/runtime/utils/AlitaCallback.py,sha256=E4LlSBuCHWiUq6W7IZExERHZY0qcmdjzc_rJlF2iQIw,7356
 alita_sdk/runtime/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/runtime/utils/constants.py,sha256=Xntx1b_uxUzT4clwqHA_U6K8y5bBqf_4lSQwXdcWrp4,13586
@@ -144,11 +144,10 @@ alita_sdk/tools/azure_ai/search/__init__.py,sha256=FVWNSW4LlOXKt34fVUgXut5oZcok9
 alita_sdk/tools/azure_ai/search/api_wrapper.py,sha256=E4p6HPDlwgxfT_i6cvg9rN4Vn_47CVAyNBAKLIGq3mU,7265
 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=_ywYlgYoE6gtJlLR94MHcS4EPWIaCFU_Mxy2Iha8nCE,5385
-alita_sdk/tools/bitbucket/api_wrapper.py,sha256=xKa2dQ-gw2YbLJx7P1xrc3JUfgBkXkMsEG-s0mzh3KI,11023
+alita_sdk/tools/bitbucket/__init__.py,sha256=YhD74yn5pC8Kjm2nRfH3Xot00Nk2kpdvZxaLmMRzmKA,5343
+alita_sdk/tools/bitbucket/api_wrapper.py,sha256=ajnF9-LYYy4w-tXagYtnvKicnQIZE0JUQPGlwOVOy7E,19983
 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=o1hwSEbSmjd5YgF-AsWls2ZyrewUuegBn9S6xTwsT74,15326
 alita_sdk/tools/browser/__init__.py,sha256=iByi9uMGjd6v44SagIPTm5fu1vWnxIkjn3xsx86uRwI,5249
 alita_sdk/tools/browser/crawler.py,sha256=jhE35dU94eQLURSM-D50tspOqEMsiGzMDbYNqNSR2mU,2279
 alita_sdk/tools/browser/duck_duck_go_search.py,sha256=iKws923v34o-ySXohJw-8xTDBWlj3fMsnzC_ZRuPugE,2002
@@ -314,7 +313,7 @@ alita_sdk/tools/testrail/__init__.py,sha256=0kETjWKLU7R6mugBWsjwEUsh10pipbAeNSGJ
 alita_sdk/tools/testrail/api_wrapper.py,sha256=5T-QyTzt-J0rI32xc_E684lCdgyWeHSyeTYiwQwtGyg,32275
 alita_sdk/tools/utils/__init__.py,sha256=155xepXPr4OEzs2Mz5YnjXcBpxSv1X2eznRUVoPtyK0,3268
 alita_sdk/tools/utils/available_tools_decorator.py,sha256=IbrdfeQkswxUFgvvN7-dyLMZMyXLiwvX7kgi3phciCk,273
-alita_sdk/tools/utils/content_parser.py,sha256=zqeyuxZqZqVFq5M5sZM-falMdlOw48FyZnp3Z0XUpCw,9868
+alita_sdk/tools/utils/content_parser.py,sha256=a8m5kSpEuI4d3YIJlBqSrHtEc-igAnOUDI_uRyo4Sls,9878
 alita_sdk/tools/vector_adapters/VectorStoreAdapter.py,sha256=a6FAsiix_EvATIKUf5YT6vHh5LDyJ5uSP3LJqoxFo04,17367
 alita_sdk/tools/vector_adapters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/tools/xray/__init__.py,sha256=BnG2StSfX44CUMtrjHTcSCDWxxse5tCZqwyaZSkBKIc,4230
@@ -336,8 +335,8 @@ alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=HOt9ShtJI_1tVPcwd3Rwk-VS0SMLq
 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.260.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.260.dist-info/METADATA,sha256=dRh-jEpVK5p3pK7KvE4h_JEX6qfMkaHe8F4NkOdHs8c,18897
-alita_sdk-0.3.260.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.260.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.260.dist-info/RECORD,,
+alita_sdk-0.3.262.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.262.dist-info/METADATA,sha256=jMCRTQ5YTuRDgHondlpPBMyNcZ9IH5M7jebxHxoZjTY,18897
+alita_sdk-0.3.262.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.262.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.262.dist-info/RECORD,,

alita_sdk/tools/bitbucket/tools.py

@@ -1,304 +0,0 @@
-import logging
-import traceback
-from typing import Type, Optional, List
-
-from langchain_core.tools import BaseTool
-from pydantic import create_model, Field, BaseModel
-from .bitbucket_constants import create_pr_data
-
-from .api_wrapper import BitbucketAPIWrapper
-from ..elitea_base import LoaderSchema
-
-logger = logging.getLogger(__name__)
-
-branchInput = create_model(
-    "BranchInput",
-    branch_name=(str, Field(description="The name of the branch, e.g. `my_branch`.")))
-
-
-class CreateBitbucketBranchTool(BaseTool):
-    api_wrapper: BitbucketAPIWrapper = Field(default_factory=BitbucketAPIWrapper)
-    name: str = "create_branch"
-    description: str = """This tool is a wrapper for the Bitbucket API to create a new branch in the repository."""
-    args_schema: Type[BaseModel] = branchInput
-
-    def _run(self, branch_name: str):
-        try:
-            logger.info(f"Creating branch {branch_name} in the repository.")
-            return self.api_wrapper.create_branch(branch_name)
-        except Exception:
-            stacktrace = traceback.format_exc()
-            logger.error(f"Unable to create a branch: {stacktrace}")
-            return f"Unable to create a branch: {stacktrace}"
-
-
-class CreatePRTool(BaseTool):
-    api_wrapper: BitbucketAPIWrapper = Field(default_factory=BitbucketAPIWrapper)
-    name: str = "create_pull_request"
-    description: str = """This tool is a wrapper for the Bitbucket API to create a new pull request in a GitLab repository.
-    Strictly follow and provide input parameters based on context.
-    """
-    args_schema: Type[BaseModel] = create_model(
-        "CreatePRInput",
-        pr_json_data=(str, Field(description=create_pr_data)))
-
-    def _run(self, pr_json_data: str):
-        try:
-            base_branch = self.api_wrapper.branch
-            logger.info(f"Creating pull request using data: base_branch: {pr_json_data}")
-            return self.api_wrapper.create_pull_request(pr_json_data)
-        except Exception as e:
-            stacktrace = traceback.format_exc()
-            logger.error(f"Unable to create PR: {stacktrace}")
-            return f"Unable to create PR: {stacktrace}"
-
-
-class CreateFileTool(BaseTool):
-    api_wrapper: BitbucketAPIWrapper = Field(default_factory=BitbucketAPIWrapper)
-    name: str = "create_file"
-    description: str = """This tool is a wrapper for the Bitbucket API, useful when you need to create a file in a Bitbucket repository.
-    """
-    args_schema: Type[BaseModel] = create_model(
-        "CreateFileInput",
-        file_path=(str, Field(
-            description="File path of file to be created. e.g. `src/agents/developer/tools/git/bitbucket.py`. **IMPORTANT**: the path must not start with a slash")),
-        file_contents=(str, Field(description="""
-    Full file content to be created. It must be without any escapes, just raw content to CREATE in GIT.
-    Generate full file content for this field without any additional texts, escapes, just raw code content. 
-    You MUST NOT ignore, skip or comment any details, PROVIDE FULL CONTENT including all content based on all best practices.
-    """)),
-        branch=(str, Field(
-            description="branch - name of the branch file should be read from. e.g. `feature-1`. **IMPORTANT**: if branch not specified, try to determine from the chat history or clarify with user."))
-    )
-
-    def _run(self, file_path: str, file_contents: str, branch: str):
-        logger.info(f"Create file in the repository {file_path} with content: {file_contents}")
-        return self.api_wrapper.create_file(file_path, file_contents, branch)
-
-
-class UpdateFileTool(BaseTool):
-    api_wrapper: BitbucketAPIWrapper = Field(default_factory=BitbucketAPIWrapper)
-    name: str = "update_file"
-    description: str = """
-        Updates a file with new content.
-        Parameters:
-            file_path (str): Path to the file to be updated 
-            branch (str): The name of the branch where update the file.
-            update_query(str): Contains file contents.
-                The old file contents is wrapped in OLD <<<< and >>>> OLD
-                The new file contents is wrapped in NEW <<<< and >>>> NEW
-                For example:
-                /test/hello.txt
-                OLD <<<<
-                Hello Earth!
-                >>>> OLD
-                NEW <<<<
-                Hello Mars!
-                >>>> NEW
-                """
-    args_schema: Type[BaseModel] = create_model(
-        "UpdateFileTool",
-        file_path=(str, Field(
-            description="File path of file to be updated. e.g. `src/agents/developer/tools/git/bitbucket.py`. **IMPORTANT**: the path must not start with a slash")),
-        update_query=(str, Field(description="File path followed by the old and new contents")),
-        branch=(str, Field(
-            description="branch - the name of the branch where file that has to be updated is located. e.g. `feature-1`. **IMPORTANT**: if branch not specified, try to determine from the chat history or clarify with user."))
-    )
-
-    def _run(self, file_path: str, update_query: str, branch: str):
-        logger.info(f"Update file in the repository {file_path} with content: {update_query} and branch {branch}")
-        return self.api_wrapper.update_file(file_path, update_query, branch)
-
-
-class SetActiveBranchTool(BaseTool):
-    api_wrapper: BitbucketAPIWrapper = Field(default_factory=BitbucketAPIWrapper)
-    name: str = "set_active_branch"
-    description: str = """
-    This tool is a wrapper for the Bitbucket API and set the active branch in the repository, similar to `git checkout <branch_name>` and `git switch -c <branch_name>`."""
-    args_schema: Type[BaseModel] = branchInput
-
-    def _run(self, branch_name: str):
-        try:
-            logger.info(f"Set active branch {branch_name} in the repository.")
-            return self.api_wrapper.set_active_branch(branch=branch_name)
-        except Exception as e:
-            stacktrace = traceback.format_exc()
-            logger.error(f"Unable to set active branch: {stacktrace}")
-            return f"Unable to set active branch: {stacktrace}"
-
-
-class ListBranchesTool(BaseTool):
-    """Tool for interacting with the Bitbucket API."""
-
-    api_wrapper: BitbucketAPIWrapper = Field(default_factory=BitbucketAPIWrapper)
-    name: str = "list_branches_in_repo"
-    description: str = """This tool is a wrapper for the Bitbucket API to fetch a list of all branches in the repository. 
-    It will return the name of each branch. No input parameters are required."""
-    args_schema: Type[BaseModel] = create_model("NoInput")
-
-    def _run(self):
-        try:
-            logger.debug("List branches in the repository.")
-            return self.api_wrapper.list_branches_in_repo()
-        except Exception as e:
-            stacktrace = traceback.format_exc()
-            logger.error(f"Unable to list branches: {stacktrace}")
-            return f"Unable to list branches: {stacktrace}"
-
-
-class ReadFileTool(BaseTool):
-    api_wrapper: BitbucketAPIWrapper = Field(default_factory=BitbucketAPIWrapper)
-    name: str = "read_file"
-    description: str = """This tool is a wrapper for the GitLab API, useful when you need to read a file in a GitLab repository.
-    Simply pass in the full
-    file path of the file you would like to read. **IMPORTANT**: the path must not start with a slash"""
-    args_schema: Type[BaseModel] = create_model(
-        "ReadFileInput",
-        file_path=(str, Field(
-            description="File path of file to be read. e.g. `src/agents/developer/tools/git/github_tools.py`. **IMPORTANT**: the path must not start with a slash")),
-        branch=(str, Field(
-            description="branch - name of the branch file should be read from. e.g. `feature-1`. **IMPORTANT**: if branch not specified, try to determine from the chat history or clarify with user."))
-    )
-
-    def _run(self, file_path: str, branch: str):
-        try:
-            return self.api_wrapper._read_file(file_path, branch)
-        except Exception:
-            stacktrace = traceback.format_exc()
-            logger.error(f"Unable to read file: {stacktrace}")
-            return f"Unable to read file: {stacktrace}"
-
-
-class GetFilesListTool(BaseTool):
-    api_wrapper: BitbucketAPIWrapper = Field(default_factory=BitbucketAPIWrapper)
-    name: str = "get_files"
-    description: str = """
-    This tool returns list of files.
-    IMPORTANT: *User may leave it empty if he wants to read files from the root of the repository and using active branch.*
-
-    Args:
-        file_path (Optional[str], default: None): Package path to read files from. e.g. `src/agents/developer/tools/git/`. **IMPORTANT**: the path must not start with a slash.
-        branch (Optional[str], default: None): branch - name of the branch file should be read from. e.g. `feature-1`. **IMPORTANT**: if branch not specified, try to determine from the chat history, leave it empty.
-    """
-    args_schema: Type[BaseModel] = create_model(
-        "GetFilesListModel",
-        file_path=(Optional[str], Field(
-            description="Package path to read files from. e.g. `src/agents/developer/tools/git/`. Default: None. "
-                        "**IMPORTANT**: the path must not start with a slash",
-            default=None)),
-        branch=(Optional[str], Field(
-            description="branch - name of the branch file should be read from. e.g. `feature-1`. Default: None. "
-                        "**IMPORTANT**: if branch not specified, try to determine from the chat history or use active branch.",
-            default=None))
-    )
-
-    def _run(self, file_path: Optional[str] = None, branch: Optional[str] = None):
-        try:
-            return self.api_wrapper._get_files(file_path, branch)
-        except Exception:
-            stacktrace = traceback.format_exc()
-            logger.error(f"Unable to read file: {stacktrace}")
-            return f"Unable to read file: {stacktrace}"
-
-
-class LoaderTool(BaseTool):
-    api_wrapper: BitbucketAPIWrapper = Field(default_factory=BitbucketAPIWrapper)
-    name: str = "loader"
-    description: str = """This tool is a wrapper for the Bitbucket API, useful when you need to create a file in a Bitbucket repository.
-    """
-    args_schema: Type[BaseModel] = LoaderSchema
-
-    def _run(self,
-             branch: Optional[str] = None,
-             whitelist: Optional[List[str]] = None,
-             blacklist: Optional[List[str]] = None):
-        return self.api_wrapper.loader(branch, whitelist, blacklist)
-
-
-class GetPullRequestsCommitsTool(BaseTool):
-    api_wrapper: BitbucketAPIWrapper = Field(default_factory=BitbucketAPIWrapper)
-    name: str = "get_pull_requests_commits"
-    description: str = """This tool is a wrapper for the Bitbucket API to get commits from pull requests."""
-    args_schema: Type[BaseModel] = create_model(
-        "GetPullRequestsCommitsInput",
-        pr_id=(str, Field(description="The ID of the pull request to get commits from.")))
-
-    def _run(self, pr_id: str):
-        try:
-            logger.debug(f"Get pull request commits for {pr_id}.")
-            return self.api_wrapper.get_pull_requests_commits(pr_id)
-        except Exception as e:
-            stacktrace = traceback.format_exc()
-            logger.error(f"Can't get commits from pull request `{pr_id}` due to error:\n{stacktrace}")
-            return f"Can't get commits from pull request `{pr_id}` due to error:\n{stacktrace}"
-
-class GetPullRequestTool(BaseTool):
-    api_wrapper: BitbucketAPIWrapper = Field(default_factory=BitbucketAPIWrapper)
-    name: str = "get_pull_request"
-    description: str = """This tool is a wrapper for the Bitbucket API to get details of a pull request."""
-    args_schema: Type[BaseModel] = create_model(
-        "GetPullRequestInput",
-        pr_id=(str, Field(description="The ID of the pull request to get details from.")))
-
-    def _run(self, pr_id: str):
-        try:
-            logger.debug(f"Get pull request details for {pr_id}.")
-            return self.api_wrapper.get_pull_request(pr_id)
-        except Exception as e:
-            stacktrace = traceback.format_exc()
-            logger.error(f"Can't get pull request `{pr_id}` due to error:\n{stacktrace}")
-            return f"Can't get pull request `{pr_id}` due to error:\n{stacktrace}"
-
-class GetPullRequestsChangesTool(BaseTool):
-    api_wrapper: BitbucketAPIWrapper = Field(default_factory=BitbucketAPIWrapper)
-    name: str = "get_pull_requests_changes"
-    description: str = """This tool is a wrapper for the Bitbucket API to get changes of a pull request."""
-    args_schema: Type[BaseModel] = create_model(
-        "GetPullRequestsChangesInput",
-        pr_id=(str, Field(description="The ID of the pull request to get changes from.")))
-
-    def _run(self, pr_id: str):
-        try:
-            logger.debug(f"Get pull request changes for {pr_id}.")
-            return self.api_wrapper.get_pull_requests_changes(pr_id)
-        except Exception as e:
-            stacktrace = traceback.format_exc()
-            logger.error(f"Can't get changes from pull request `{pr_id}` due to error:\n{stacktrace}")
-            return f"Can't get changes from pull request `{pr_id}` due to error:\n{stacktrace}"
-
-class AddPullRequestCommentTool(BaseTool):
-    api_wrapper: BitbucketAPIWrapper = Field(default_factory=BitbucketAPIWrapper)
-    name: str = "add_pull_request_comment"
-    description: str = """This tool is a wrapper for the Bitbucket API to add a comment to a pull request.
-    Supports multiple content types and inline comments."""
-    args_schema: Type[BaseModel] = create_model(
-        "AddPullRequestCommentInput",
-        pr_id=(str, Field(description="The ID of the pull request to add a comment to.")),
-        content=(str, Field(description="The comment content. Can be a string (raw text) or a dict with keys 'raw', 'markup', 'html'.")),
-        inline=(Optional[dict], Field(default=None, description="Inline comment details. Example: {'from': 57, 'to': 122, 'path': '<string>'}"))
-    )
-
-    def _run(self, pr_id: str, content: str, inline: Optional[dict] = None):
-        try:
-            logger.debug(f"Add comment to pull request {pr_id}.")
-            return self.api_wrapper.add_pull_request_comment(pr_id, content, inline)
-        except Exception as e:
-            stacktrace = traceback.format_exc()
-            logger.error(f"Can't add comment to pull request `{pr_id}` due to error:\n{stacktrace}")
-            return f"Can't add comment to pull request `{pr_id}` due to error:\n{stacktrace}"
-
-__all__ = [
-    {"name": "create_branch", "tool": CreateBitbucketBranchTool},
-    {"name": "create_pull_request", "tool": CreatePRTool},
-    {"name": "create_file", "tool": CreateFileTool},
-    {"name": "set_active_branch", "tool": SetActiveBranchTool},
-    {"name": "list_branches_in_repo", "tool": ListBranchesTool},
-    {"name": "read_file", "tool": ReadFileTool},
-    {"name": "get_files", "tool": GetFilesListTool},
-    {"name": "update_file", "tool": UpdateFileTool},
-    {"name": "loader", "tool": LoaderTool},
-    {"name": "get_pull_requests_commits", "tool": GetPullRequestsCommitsTool},
-    {"name": "get_pull_request", "tool": GetPullRequestTool},
-    {"name": "get_pull_requests_changes", "tool": GetPullRequestsChangesTool},
-    {"name": "add_pull_request_comment", "tool": AddPullRequestCommentTool}
-]