alita-sdk 0.3.173__py3-none-any.whl → 0.3.174__py3-none-any.whl

alita_sdk/tools/azure_ai/search/__init__.py

@@ -4,7 +4,8 @@ from .api_wrapper import AzureSearchApiWrapper
 from ...base.tool import BaseAction
 from langchain_core.tools import BaseToolkit, BaseTool
 from pydantic import create_model, BaseModel, ConfigDict, Field, SecretStr
-from ...utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
+from ...utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length, check_connection_response
+import requests
 
 logger = getLogger(__name__)
 
@@ -35,12 +36,12 @@ class AzureSearchToolkit(BaseToolkit):
     def toolkit_config_schema() -> BaseModel:
         selected_tools = {x['name']: x['args_schema'].schema() for x in AzureSearchApiWrapper.model_construct().get_available_tools()}
         AzureSearchToolkit.toolkit_max_length = get_max_toolkit_length(selected_tools)
-        return create_model(
+        m = create_model(
             name,
-            api_key=(SecretStr, Field(description="API key", json_schema_extra={'secret': True})),
+            api_key=(SecretStr, Field(description="API key", json_schema_extra={'secret': True, 'configuration': True})),
             endpoint=(str, Field(description="Azure Search endpoint")),
             index_name=(str, Field(description="Azure Search index name")),
-            api_base=(Optional[str], Field(description="Azure OpenAI base URL", default=None, json_schema_extra={'toolkit_name': True, 'max_toolkit_length': AzureSearchToolkit.toolkit_max_length})),
+            api_base=(Optional[str], Field(description="Azure OpenAI base URL", default=None, json_schema_extra={'toolkit_name': True, 'max_toolkit_length': AzureSearchToolkit.toolkit_max_length, 'configuration': True})),
             api_version=(Optional[str], Field(description="API version", default=None)),
             openai_api_key=(Optional[str], Field(description="Azure OpenAI API Key", default=None, json_schema_extra={'secret': True})),
             model_name=(str, Field(description="Model name for Embeddings model", default=None)),
@@ -54,6 +55,16 @@ class AzureSearchToolkit(BaseToolkit):
             })
         )
 
+        @check_connection_response
+        def check_connection(self):
+            response = requests.get(f"{self.api_base}/openai/deployments", headers={
+                "api-key": self.api_key
+            })
+            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):
         if selected_tools is None:

alita_sdk/tools/bitbucket/__init__.py

@@ -1,10 +1,14 @@
 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, SecretStr
-from ..utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
+from ..utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length, check_connection_response
+import requests
 
 
 name = "bitbucket"
@@ -35,14 +39,14 @@ class AlitaBitbucketToolkit(BaseToolkit):
             default = t['tool'].__pydantic_fields__['args_schema'].default
             selected_tools[t['name']] = default.schema() if default else default
         AlitaBitbucketToolkit.toolkit_max_length = get_max_toolkit_length(selected_tools)
-        return create_model(
+        m = create_model(
             name,
-            url=(str, Field(description="Bitbucket URL")),
-            project=(str, Field(description="Project/Workspace")),
-            repository=(str, Field(description="Repository", json_schema_extra={'toolkit_name': True, 'max_toolkit_length': AlitaBitbucketToolkit.toolkit_max_length})),
+            url=(str, Field(description="Bitbucket URL", json_schema_extra={'configuration': True})),
+            project=(str, Field(description="Project/Workspace", json_schema_extra={'configuration': True})),
+            repository=(str, Field(description="Repository", json_schema_extra={'toolkit_name': True, 'max_toolkit_length': AlitaBitbucketToolkit.toolkit_max_length, 'configuration': True})),
             branch=(str, Field(description="Main branch", default="main")),
-            username=(str, Field(description="Username")),
-            password=(SecretStr, Field(description="GitLab private token", json_schema_extra={'secret': True})),
+            username=(str, Field(description="Username", json_schema_extra={'configuration': True})),
+            password=(SecretStr, Field(description="GitLab private token", json_schema_extra={'secret': True, 'configuration': True})),
             cloud=(Optional[bool], Field(description="Hosting Option", default=None)),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
             __config__=ConfigDict(json_schema_extra=
@@ -56,6 +60,18 @@ class AlitaBitbucketToolkit(BaseToolkit):
             })
         )
 
+        @check_connection_response
+        def check_connection(self):
+            if self.cloud:
+                request_url = f"{self.url}/2.0/repositories/{self.project}/{self.repository}"
+            else:
+                request_url = f"{self.url}/rest/api/1.0/projects/{self.project}/repos/{self.repository}"
+            response = requests.get(request_url, auth=HTTPBasicAuth(self.username, self.password))
+            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):
         if selected_tools is None:

{alita_sdk-0.3.173.dist-info → alita_sdk-0.3.174.dist-info}/METADATA

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

@@ -144,11 +144,11 @@ alita_sdk/tools/ado/work_item/ado_wrapper.py,sha256=t0D9xubU0yy_JmRJ_zEtRCxwFLya
 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
 alita_sdk/tools/azure_ai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-alita_sdk/tools/azure_ai/search/__init__.py,sha256=FFSMBINRfODSjs7hNalwRx9BrroHUnfv6Jw8nN0w2Y8,3808
+alita_sdk/tools/azure_ai/search/__init__.py,sha256=CYqGc6jPkXb94_dNve1xY3rI8WTm4Gw5aEJlkMYV3U8,4198
 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=987vZZvFnrmKeaefXtXs4DjF6OFVEOKAG9F5ZHf1A_4,3562
+alita_sdk/tools/bitbucket/__init__.py,sha256=pYwgq-74AwZWeTIzdUKn-7tnuUgCqpOmz_2p-dBgEDs,4326
 alita_sdk/tools/bitbucket/api_wrapper.py,sha256=HAmxY43VwfI08u0J6Z3ViBvADGNKM6LKAj1Iq5DNvMU,7133
 alita_sdk/tools/bitbucket/bitbucket_constants.py,sha256=UsbhQ1iEvrKoxceTFPWTYhaXS1zSxbmjs1TwY0-P4gw,462
 alita_sdk/tools/bitbucket/cloud_api_wrapper.py,sha256=9pvU--c02EB2JVkXuMI4cehWU657HsFpywAmtLC3yVs,7731
@@ -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.173.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.173.dist-info/METADATA,sha256=XTue8f1fKyX4zdPdddEyLCNsSDbFo-VdK7pLuIbZ5mI,18757
-alita_sdk-0.3.173.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.173.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.173.dist-info/RECORD,,
+alita_sdk-0.3.174.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.174.dist-info/METADATA,sha256=tbM4oo5f1XOezac1kgzmW98CVraz5Yky6q6N82k_vKg,18757
+alita_sdk-0.3.174.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.174.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.174.dist-info/RECORD,,