alita-sdk 0.3.191__py3-none-any.whl → 0.3.193__py3-none-any.whl

alita_sdk/tools/ado/repos/__init__.py

@@ -85,7 +85,11 @@ class AzureDevOpsReposToolkit(BaseToolkit):
                     },
                     "categories": ["code repositories"],
                     "extra_categories": ["code", "repository", "version control"],
-                    "configuration_group": "ado_repos",
+                    "configuration_group": {
+                        "name": "ado_repos",
+                        "label": "Azure DevOps Repositories",
+                        "icon_url": "ado-repos-icon.svg",
+                    }
                 }}}
         )
 

alita_sdk/tools/ado/test_plan/__init__.py

@@ -51,7 +51,13 @@ class AzureDevOpsPlansToolkit(BaseToolkit):
                             ]
                         }
                     },
-                    "configuration_group": "ado",
+                    # connect different toolkits under the same configuration group with the same name,
+                    # label and icon
+                    "configuration_group": {
+                        "name": "ado",
+                        "label": "Azure DevOps",
+                        "icon_url": "azure-icon.svg",
+                    }
                 }
             }
             }

alita_sdk/tools/ado/wiki/__init__.py

@@ -26,9 +26,17 @@ class AzureDevOpsWikiToolkit(BaseToolkit):
                                  'max_toolkit_length': AzureDevOpsWikiToolkit.toolkit_max_length})
                   ),
             organization_url=(str, Field(title="Organization URL",
-                                                   description="ADO organization url")),
-            project=(str, Field(description="ADO project")),
-            token=(SecretStr, Field(description="ADO token", json_schema_extra={'secret': True})),
+                                         description="ADO organization url", json_schema_extra={
+                    'configuration': True,
+                })),
+            project=(str, Field(description="ADO project",
+                                json_schema_extra={
+                                    'configuration': True
+                                })),
+            token=(SecretStr,
+                   Field(description="ADO token",
+                         json_schema_extra={'secret': True, 'configuration': True }
+                         )),
             selected_tools=(List[Literal[tuple(selected_tools)]],
                             Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
             __config__={
@@ -49,7 +57,9 @@ class AzureDevOpsWikiToolkit(BaseToolkit):
                                 ]
                             }
                         },
-                        "configuration_group": "ado",
+                        "configuration_group": {
+                            "name": "ado",
+                        }
                     }
                 }
             }

alita_sdk/tools/ado/work_item/__init__.py

@@ -25,9 +25,16 @@ class AzureDevOpsWorkItemsToolkit(BaseToolkit):
                                  'max_toolkit_length': AzureDevOpsWorkItemsToolkit.toolkit_max_length})
                   ),
             organization_url=(str, Field(title="Organization URL",
-                                                   description="ADO organization url")),
-            project=(str, Field(description="ADO project")),
-            token=(SecretStr, Field(description="ADO token", json_schema_extra={'secret': True})),
+                                         description="ADO organization url",
+                                         json_schema_extra={
+                                             'configuration': True
+                                         })),
+            project=(str, Field(description="ADO project",
+                                json_schema_extra={
+                                    'configuration': True
+                                }
+                                )),
+            token=(SecretStr, Field(description="ADO token", json_schema_extra={'secret': True, 'configuration': True})),
             limit=(Optional[int], Field(description="ADO plans limit used for limitation of the list with results", default=5)),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
             __config__={
@@ -48,7 +55,9 @@ class AzureDevOpsWorkItemsToolkit(BaseToolkit):
                                 ]
                             }
                         },
-                        "configuration_group": "ado",
+                        "configuration_group": {
+                            "name": "ado",
+                        }
                     }
                 }
             }

alita_sdk/tools/memory/__init__.py

@@ -40,6 +40,7 @@ class MemoryToolkit(BaseToolkit):
                 'metadata': {
                     "label": "Memory",
                     "icon_url": "memory.svg",
+                    "hidden": True,
                     "categories": ["other"],
                     "extra_categories": ["long-term memory", "langmem"],
                 }

alita_sdk/tools/slack/__init__.py

@@ -1,11 +1,17 @@
-from typing import List, Optional
+from typing import List, Optional, Literal
+
+import logging
+
+logger = logging.getLogger(__name__)
 
 from langchain_core.tools import BaseToolkit, BaseTool
 from pydantic import create_model, BaseModel, Field, SecretStr
 from ..base.tool import BaseAction
 
 from .api_wrapper import SlackApiWrapper
-from ..utils import TOOLKIT_SPLITTER, clean_string, get_max_toolkit_length
+from ..utils import TOOLKIT_SPLITTER, clean_string, get_max_toolkit_length, check_connection_response
+from slack_sdk.errors import SlackApiError
+from slack_sdk import WebClient
 
 name = "slack"
 
@@ -24,13 +30,30 @@ class SlackToolkit(BaseToolkit):
     def toolkit_config_schema() -> BaseModel:
          selected_tools = {x['name']: x['args_schema'].schema() for x in SlackApiWrapper.model_construct().get_available_tools()}
          SlackToolkit.toolkit_max_length = get_max_toolkit_length(selected_tools)
-         return create_model(
+
+         @check_connection_response
+         def check_connection(self):
+            """
+            Checks the connection to Slack using the provided token.
+            Returns the response from Slack's auth.test endpoint.
+            """
+            try:
+                response = WebClient(token=self.slack_token.get_secret_value()).auth_test()
+                logger.info("Slack connection successful: %s", response)
+                return {"success": True, "response": response}
+            except SlackApiError as e:
+                logger.error(f"Slack connection failed: {e.response['error']}")
+                return {"success": False, "error": e.response['error']}
+            
+         model = create_model(
             name,
-            slack_token=(SecretStr, Field( description="Slack Bot/User OAuth Token like XOXB-*****-*****-*****-*****")),
-            channel_id=(str, Field(title="Channel ID", description="Channel ID, user ID, or conversation ID to send the message to. (like C12345678 for public channels, D12345678 for DMs)")),
-            selected_tools=(list[str], Field(title="Selected Tools", description="List of tools to enable", default=[])),
-             __config__={'json_schema_extra': {'metadata': {"label": "slack", "icon_url": None, "hidden": True}}}
+            slack_token=(SecretStr, Field(description="Slack Token like XOXB-*****-*****-*****-*****", json_schema_extra={'secret': True, 'configuration': True})),
+            channel_id=(str, Field( description="Channel ID", json_schema_extra={'configuration': True})),
+            selected_tools=(list[str], Field( description="List of tools to enable", default=[],json_schema_extra={'args_schemas': selected_tools})),
+            __config__={'json_schema_extra': {'metadata': {"label": "Slack", "icon_url": None}}}
         )
+         model.check_connection = check_connection
+         return model
 
     @classmethod
     def get_toolkit(cls, selected_tools: Optional[List[str]] = None, toolkit_name: Optional[str] = None, **kwargs):
@@ -48,7 +71,6 @@ class SlackToolkit(BaseToolkit):
                 name=prefix + tool["name"],
                 description=tool["description"],
                 args_schema=tool["args_schema"],
-                func=tool["ref"]
             ))
         return cls(tools=tools)
 

alita_sdk/tools/slack/api_wrapper.py

@@ -38,27 +38,19 @@ class SlackApiWrapper(BaseToolApiWrapper):
     slack_token: Optional[SecretStr] = Field(default=None,description="Slack Bot/User OAuth Token like XOXB-*****-*****-*****-*****")
     channel_id: Optional[str] = Field(default=None, description="Channel ID, user ID, or conversation ID to send the message to. (like C12345678 for public channels, D12345678 for DMs)")
 
-    @model_validator(mode="after")
+    @model_validator(mode="before")
     @classmethod
     def validate_toolkit(cls, values):
-        token = values.slack_token.get_secret_value() if values.slack_token else None
+        token = values.get("slack_token")
         if not token:
-            logging.error("Slack token is required.")
-            raise ValueError("Slack token is required.")
-        try:
-            cls._client = WebClient(token=token)
-            logging.info("Authenticated with Slack token.")
-        except Exception as e:
-            logging.error(f"Failed to authenticate with Slack: {str(e)}")
-            raise ValueError(f"Failed to authenticate with Slack: {str(e)}")
+            logger.error("Slack token is required for authentication.")
+            raise ValueError("Slack token is required for authentication.")
         return values
-
     def _get_client(self):
-        if not self._client:
+        if not hasattr(self, "_client") or self._client is None:
             self._client = WebClient(token=self.slack_token.get_secret_value())
         return self._client
-
-
+    
     def send_message(self, message: str):
         """
         Sends a message to a specified Slack channel, user, or conversation.
@@ -85,6 +77,7 @@ class SlackApiWrapper(BaseToolApiWrapper):
         try:
 
             client = self._get_client()
+            logger.info(f"auth test: {client.auth_test()}")
             # Fetch conversation history
             response = client.conversations_history(
                 channel=self.channel_id,

{alita_sdk-0.3.191.dist-info → alita_sdk-0.3.193.dist-info}/METADATA

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

@@ -102,13 +102,13 @@ alita_sdk/tools/__init__.py,sha256=UztP-wrR-MuJI2V0gcJzNoMhsNJV6mre07TJNPChAtM,1
 alita_sdk/tools/elitea_base.py,sha256=p1xuAsCe3qUwsmmgp_rdi6QipGeOKXA8wdSwX4MRNk4,21292
 alita_sdk/tools/ado/__init__.py,sha256=mD6GHcYMTtffPJkJvFPe2rzvye_IRmXmWfI7xYuZhO4,912
 alita_sdk/tools/ado/utils.py,sha256=PTCludvaQmPLakF2EbCGy66Mro4-rjDtavVP-xcB2Wc,1252
-alita_sdk/tools/ado/repos/__init__.py,sha256=bzVSEAPwBoH4sY3cNj5_FNXIC3yY8lkofaonNKhwhRk,6286
+alita_sdk/tools/ado/repos/__init__.py,sha256=saNovhnawWFY3xwKA6FssR7cakPQzTNp_UjkpgCnyF0,6462
 alita_sdk/tools/ado/repos/repos_wrapper.py,sha256=_OWKAls7VFfFtEPTwqj_DxE1MSvpC0ivxdTIULEz3Tk,48206
-alita_sdk/tools/ado/test_plan/__init__.py,sha256=K-jhCT7D7bxXbiat3GCP_qDVg_ipcBlnh9r3JhmAubM,4408
+alita_sdk/tools/ado/test_plan/__init__.py,sha256=bVywTYTvdm1rUeP2krVVMRN-xDCY--ze7NFdTxJP9ow,4708
 alita_sdk/tools/ado/test_plan/test_plan_wrapper.py,sha256=p1Mptd_1J6bmkyrvf2M-FB79s8THzEesBlfgaOnRXb8,18152
-alita_sdk/tools/ado/wiki/__init__.py,sha256=yyqTlzc14TC9w9OapKAmTjrZSuwRpsIdMkV9HUZGRlI,4201
+alita_sdk/tools/ado/wiki/__init__.py,sha256=WCIKOisU2h3E4SNDvGfWCMZ3nRMxfH_ZhIffmSHH3XI,4576
 alita_sdk/tools/ado/wiki/ado_wrapper.py,sha256=l4bc2QoKSUXg9UqNcx0ylv7YL9JPPQd35Ti5MXyEgC4,12690
-alita_sdk/tools/ado/work_item/__init__.py,sha256=0vaynh82-stscb-K9MJsPt1okOZ2h9O5pr7efJx3Kno,4299
+alita_sdk/tools/ado/work_item/__init__.py,sha256=k6gZ6pEE7gvNWvCDoDV05jltzbqxC_NPm06CEr5Wwcs,4726
 alita_sdk/tools/ado/work_item/ado_wrapper.py,sha256=aLB-aSNQST0FCwP7I01OXanCpZHKVarZZB1u9j2H1LA,26253
 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
@@ -223,7 +223,7 @@ alita_sdk/tools/llm/llm_utils.py,sha256=v3_lWP_Nk6tJLkj0BYohOun0OWNfvzqLjPdPAMl-
 alita_sdk/tools/localgit/__init__.py,sha256=NScO0Eu-wl-rc63jjD5Qv1RXXB1qukSIJXx-yS_JQLI,2529
 alita_sdk/tools/localgit/local_git.py,sha256=gsAftNcK7nMCd8VsIkwDLs2SoG0MgpYdkQG5tmoynkA,18074
 alita_sdk/tools/localgit/tool.py,sha256=It_B24rMvFPurB355Oy5IShg2BsZTASsEoSS8hu2SXw,998
-alita_sdk/tools/memory/__init__.py,sha256=kMa3sDe6CF_vq0xMWhadBClCfY_W8BIbIliatenHu7U,2323
+alita_sdk/tools/memory/__init__.py,sha256=8Q02h-PUvIw3bNbWbfklSJZe3qj0zAjfahq9C5XjyzE,2359
 alita_sdk/tools/ocr/__init__.py,sha256=pvslKVXyJmK0q23FFDNieuc7RBIuzNXTjTNj-GqhGb0,3335
 alita_sdk/tools/ocr/api_wrapper.py,sha256=08UF8wj1sR8DcW0z16pw19bgLatLkBF8dySW-Ds8iRk,29649
 alita_sdk/tools/ocr/text_detection.py,sha256=1DBxt54r3_HdEi93QynSIVta3rH3UpIvy799TPtDTtk,23825
@@ -270,8 +270,8 @@ alita_sdk/tools/sharepoint/__init__.py,sha256=HqKQDFboab1AYh20uJvHxs9HFLJSqVfVTj
 alita_sdk/tools/sharepoint/api_wrapper.py,sha256=qCHCIH4FRDtgdpIK22ewFhZJeOaTv9hT9BVivslxxlE,7119
 alita_sdk/tools/sharepoint/authorization_helper.py,sha256=n-nL5dlBoLMK70nHu7P2RYCb8C6c9HMA_gEaw8LxuhE,2007
 alita_sdk/tools/sharepoint/utils.py,sha256=fZ1YzAu5CTjKSZeslowpOPH974902S8vCp1Wu7L44LM,446
-alita_sdk/tools/slack/__init__.py,sha256=_s4Ogro3O92AxHdUgWCp0SlrnBChRhCimZLbcDvNzeA,2550
-alita_sdk/tools/slack/api_wrapper.py,sha256=yZWrJtXBowhq1fmF7YrrBVyChdnxkYF1XfwHxe8-_D8,7555
+alita_sdk/tools/slack/__init__.py,sha256=Y2sXN1__1923icF8bZbAtKVg3G_EZbCAjnj93yHUwtY,3415
+alita_sdk/tools/slack/api_wrapper.py,sha256=dkMcS3xozEhoJ-A1ycEn6OqbIftxmVHjyYttTy3D2JI,7343
 alita_sdk/tools/sql/__init__.py,sha256=9Lh8YHKO8zD5eeolpR4O9swTUsjpXj9LVDn8fM-T5IM,3506
 alita_sdk/tools/sql/api_wrapper.py,sha256=Rky0_CX9HWDQ2mClHGAgP3LHjYVX4iymPuilZMtaDlQ,3687
 alita_sdk/tools/sql/models.py,sha256=AKJgSl_kEEz4fZfw3kbvdGHXaRZ-yiaqfJOB6YOj3i0,641
@@ -297,8 +297,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.191.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.191.dist-info/METADATA,sha256=7-ulNxjrYbxmdKjgcTMdwamNuNbUTtHq-6Dcqth1WVM,18804
-alita_sdk-0.3.191.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.191.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.191.dist-info/RECORD,,
+alita_sdk-0.3.193.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.193.dist-info/METADATA,sha256=_ciaBDZRw_ISqVr3oVPTmZPfE4mkrHSotJeAHvHPiq4,18804
+alita_sdk-0.3.193.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.193.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.193.dist-info/RECORD,,