alita-sdk 0.3.191__tar.gz → 0.3.193__tar.gz

{alita_sdk-0.3.191 → alita_sdk-0.3.193}/PKG-INFO

@@ -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 → alita_sdk-0.3.193}/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-0.3.191 → alita_sdk-0.3.193}/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-0.3.191 → alita_sdk-0.3.193}/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-0.3.191 → alita_sdk-0.3.193}/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-0.3.191 → alita_sdk-0.3.193}/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-0.3.191 → alita_sdk-0.3.193}/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-0.3.191 → alita_sdk-0.3.193}/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 → alita_sdk-0.3.193}/alita_sdk.egg-info/PKG-INFO

@@ -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 → alita_sdk-0.3.193}/pyproject.toml

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