alita-sdk 0.3.192__py3-none-any.whl → 0.3.194__py3-none-any.whl

alita_sdk/runtime/llms/alita.py

@@ -139,7 +139,7 @@ class AlitaChatModel(BaseChatModel):
             else:
                 message = _convert_delta_to_message_chunk(chunk, default_chunk_class)
                 finish_reason = None
-                generation_info = ()
+                generation_info = dict()
                 if stop:
                     for stop_word in stop:
                         if stop_word in message.content:

alita_sdk/tools/ado/test_plan/__init__.py

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

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__={

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__={

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, 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,15 +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[Literal[tuple(selected_tools)]],
-                             Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
-             __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):
@@ -50,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/tools/zephyr_squad/__init__.py

@@ -1,14 +1,13 @@
 from typing import List, Literal, Optional
 
-from langchain_community.agent_toolkits.base import BaseToolkit
-from langchain_core.tools import BaseTool
+from langchain_core.tools import BaseToolkit, BaseTool
 from pydantic import create_model, BaseModel, Field, SecretStr
 
 from .api_wrapper import ZephyrSquadApiWrapper
 from ..base.tool import BaseAction
 from ..utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
 
-name = "zephyr"
+name = "zephyr_squad"
 
 def get_tools(tool):
     return ZephyrSquadToolkit().get_toolkit(
@@ -28,10 +27,10 @@ class ZephyrSquadToolkit(BaseToolkit):
         selected_tools = {x['name']: x['args_schema'].schema() for x in ZephyrSquadApiWrapper.model_construct().get_available_tools()}
         ZephyrSquadToolkit.toolkit_max_length = get_max_toolkit_length(selected_tools)
         return create_model(
-            "zephyr_squad",
+            name,
             account_id=(str, Field(description="AccountID for the user that is going to be authenticating")),
-            access_key=(str, Field(description="Generated access key")),
-            secret_key=(SecretStr, Field(description="Generated secret key")),
+            access_key=(SecretStr, Field(description="Generated access key", json_schema_extra={'secret': True})),
+            secret_key=(SecretStr, Field(description="Generated secret key", json_schema_extra={'secret': True})),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
             __config__={'json_schema_extra': {'metadata': {"label": "Zephyr Squad", "icon_url": "zephyr.svg",
                             "categories": ["test management"],

alita_sdk/tools/zephyr_squad/api_wrapper.py

@@ -55,6 +55,42 @@ class ZephyrSquadApiWrapper(BaseToolApiWrapper):
         """Retrieve all possible statuses for test steps in Jira."""
         return self._client.get_all_test_step_statuses()
 
+    def get_bdd_content(self, issue_id):
+        """Retrieve BDD (Gherkin) content of an issue (feature or scenario)."""
+        return self._client.get_bdd_content(issue_id)
+
+    def update_bdd_content(self, issue_id, new_content):
+        """Replace BDD (Gherkin) content of an issue (feature or scenario)."""
+        return self._client.update_bdd_content(issue_id, new_content)
+
+    def delete_bdd_content(self, issue_id):
+        """Remove BDD (Gherkin) content of an issue (feature or scenario)."""
+        return self._client.delete_bdd_content(issue_id)
+
+    def create_new_cycle(self, json):
+        """Creates a Cycle from a JSON representation. If no VersionId is passed in the request, it will be defaulted to an unscheduled version"""
+        return self._client.create_new_cycle(json)
+
+    def create_folder(self, json):
+        """Creates a Folder from a JSON representation. Folder names within a cycle needs to be unique."""
+        return self._client.create_folder(json)
+
+    def add_test_to_cycle(self, cycle_id, json):
+        """Adds Tests(s) to a Cycle."""
+        return self._client.add_test_to_cycle(cycle_id, json)
+
+    def add_test_to_folder(self, folder_id, json):
+        """Adds Tests(s) to a Folder."""
+        return self._client.add_test_to_folder(folder_id, json)
+
+    def create_execution(self, json):
+        """Creates an execution from a JSON representation."""
+        return self._client.create_execution(json)
+
+    def get_execution(self, execution_id, issue_id, project_id):
+        """Retrieves Execution and ExecutionStatus by ExecutionId"""
+        return self._client.get_execution(execution_id, issue_id, project_id)
+
     def get_available_tools(self):
         return [
             {
@@ -92,14 +128,73 @@ class ZephyrSquadApiWrapper(BaseToolApiWrapper):
                 "description": self.get_all_test_step_statuses.__doc__,
                 "args_schema": create_model("NoInput"),
                 "ref": self.get_all_test_step_statuses,
+            },
+            {
+                "name": "get_bdd_content",
+                "description": self.get_bdd_content.__doc__,
+                "args_schema": Issue,
+                "ref": self.get_bdd_content,
+            },
+            {
+                "name": "update_bdd_content",
+                "description": self.update_bdd_content.__doc__,
+                "args_schema": UpdateBddContent,
+                "ref": self.update_bdd_content,
+            },
+            {
+                "name": "delete_bdd_content",
+                "description": self.delete_bdd_content.__doc__,
+                "args_schema": Issue,
+                "ref": self.delete_bdd_content,
+            },
+            {
+                "name": "create_new_cycle",
+                "description": self.create_new_cycle.__doc__,
+                "args_schema": CycleJson,
+                "ref": self.create_new_cycle,
+            },
+            {
+                "name": "create_folder",
+                "description": self.create_folder.__doc__,
+                "args_schema": FolderJson,
+                "ref": self.create_folder,
+            },
+            {
+                "name": "add_test_to_cycle",
+                "description": self.add_test_to_cycle.__doc__,
+                "args_schema": TestToCycle,
+                "ref": self.add_test_to_cycle,
+            },
+            {
+                "name": "add_test_to_folder",
+                "description": self.add_test_to_folder.__doc__,
+                "args_schema": TestToFolder,
+                "ref": self.add_test_to_folder,
+            },
+            {
+                "name": "create_execution",
+                "description": self.create_execution.__doc__,
+                "args_schema": ExecutionJson,
+                "ref": self.create_execution,
+            },
+            {
+                "name": "get_execution",
+                "description": self.get_execution.__doc__,
+                "args_schema": GetExecution,
+                "ref": self.get_execution,
             }
         ]
 
 
+Issue = create_model(
+    "Issue",
+    issue_id=(int, Field(description="Jira ticket id of test case."))
+)
+
 ProjectIssue = create_model(
     "ProjectIssue",
-    issue_id=(int, Field(description="Jira ticket id of test case to which the test step belongs.")),
-    project_id=(int, Field(description="Jira project id to which test case belongs."))
+    project_id=(int, Field(description="Jira project id to which test case belongs.")),
+    __base__=Issue
 )
 
 ProjectIssueStep = create_model(
@@ -133,3 +228,92 @@ CreateNewTestStep = create_model(
             "*IMPORTANT*: Use double quotes for all field names and string values."))),
     __base__=ProjectIssue
 )
+
+UpdateBddContent = create_model(
+    "UpdateTestStep",
+    new_content=(str, Field(description=(
+            "String containing a Gherkin scenario or a feature background.\n"
+            "New lines must be encoded as \\n or \\r\\n, and other characters that have special meaning in JSON strings must be escaped accordingly"))),
+    __base__=Issue
+)
+
+CycleJson = create_model(
+    "CycleJson",
+    json=(str, Field(description=(
+            "JSON body to create a Zephyr test cycle. Fields:\n"
+            "- name (string, required): Test cycle name. Example: \"Test Cycle\"\n"
+            "- build (string, optional): Build name. Example: \"build 1.0\"\n"
+            "- environment (string, optional): Environment name. Example: \"Bug Fix\"\n"
+            "- description (string, optional): Cycle description. Example: \"This contains the zephyr tests for a version\"\n"
+            "- startDate (long, optional): Start date as a Unix timestamp. Example: 1485278607\n"
+            "- endDate (long, optional): End date as a Unix timestamp. Example: 1485302400\n"
+            "- projectId (integer, required): Project ID. Example: 10100\n"
+            "- versionId (integer, required): Version ID. Example: 10000\n"
+            "*IMPORTANT*: Use double quotes for all field names and string values.")))
+)
+
+FolderJson = create_model(
+    "FolderJson",
+    json=(str, Field(description=(
+            "JSON body to create a Zephyr folder. Fields:\n"
+            "- name (string, required): Folder name. Example: \"Folder 01\"\n"
+            "- description (string, optional): Folder description. Example: \"Create New Folder\"\n"
+            "- cycleId (string, required): Cycle ID. Example: \"0001513838430954-242ac112-0001\"\n"
+            "- projectId (integer, required): Project ID. Example: 10100\n"
+            "- versionId (integer, required): Version ID. Example: 10000\n"
+            "*IMPORTANT*: Use double quotes for all field names and string values.")))
+)
+
+ExecutionJson = create_model(
+    "ExecutionJson",
+    json=(str, Field(description=(
+            "JSON body for Zephyr operation. Fields:\n"
+            "- status (object, optional): Status object. Example: {\"id\": 1}\n"
+            "- id (string, optional): Unique identifier. Example: \"0001456664462103-5aoee13a3fob-0001\"\n"
+            "- projectId (integer, required): Project ID. Example: 10000\n"
+            "- issueId (integer, required): Issue ID. Example: 10000\n"
+            "- cycleId (string, optional): Cycle ID. Example: \"0001456664262308-5a6ee13a3f6b-0001\"\n"
+            "- versionId (integer, required): Version ID. Example: -1\n"
+            "- assigneeType (string, optional): \"currentUser\" or \"assignee\". Example: \"assignee\"\n"
+            "- assignee (string, optional): Assignee name if assigneeType is \"assignee\". Example: \"jiraUserKey\"\n"
+            "*IMPORTANT*: Use double quotes for all field names and string values.")))
+)
+
+TestToCycle = create_model(
+    "TestToCycle",
+    step_id=(str, Field(description="Test step id to operate.")),
+    json=(str, Field(description=(
+            "JSON body for Zephyr operation. Fields:\n"
+            "- issues (array[string], required if method=1): List of Jira issue keys. Example: [\"TEST-1\"]\n"
+            "- jql (string, required if method=2): JQL query string. Example: \"project = DEMO AND type = Test AND reporter = admin\"\n"
+            "- versionId (integer, required): Version ID. Example: -1\n"
+            "- projectId (integer, required): Project ID. Example: 10000\n"
+            "- fromVersionId (integer, required if method=3): Source Version ID. Example: -1\n"
+            "- fromCycleId (string, required if method=3): Source Cycle ID. Example: \"-0001484006184518-242ac112-0001\"\n"
+            "- statuses (string, optional, used when method=3): Statuses. Example: \"-1\"\n"
+            "- priorities (string, optional, used when method=3): Priorities (comma-separated). Example: \"1,4\"\n"
+            "- labels (string, optional, used when method=3): Labels (comma-separated). Example: \"-High,dock\"\n"
+            "- method (string, required): Operation method. Example: \"1\"\n"
+            "*IMPORTANT*: Use double quotes for all field names and string values.")))
+)
+
+TestToFolder = create_model(
+    "TestToFolder",
+    step_id=(str, Field(description="folderId of Execution.")),
+    json=(str, Field(description=(
+            "JSON body to update a Zephyr test step. Fields:\n"
+            "- issues (array[string], required): List of Jira issue keys. Example: [\"FSC-2\", \"FSC-3\"]\n"
+            "- assigneeType (string, required): Type of assignee. Example: \"currentUser\"\n"
+            "- method (integer, required): Method identifier. Example: 1\n"
+            "- versionId (integer, required): Version ID. Example: 12829\n"
+            "- projectId (integer, required): Project ID. Example: 10930\n"
+            "- cycleId (string, required): Cycle ID. Example: \"0001513838430954-242ac112-0001\"\n"
+            "*IMPORTANT*: Use double quotes for all field names and string values.")))
+)
+
+GetExecution = create_model(
+    "GetExecution",
+    execution_id=(str, Field(description="executionId of Execution.")),
+    issue_id=(int, Field(description="issueId of Execution.")),
+    project_id=(int, Field(description="projectId of Execution."))
+)

alita_sdk/tools/zephyr_squad/zephyr_squad_cloud_client.py

@@ -1,4 +1,5 @@
 import hashlib
+import json
 import time
 
 import jwt
@@ -18,44 +19,79 @@ class ZephyrSquadCloud(object):
         self.base_url = "https://prod-api.zephyr4jiracloud.com/connect"
 
     def get_test_step(self, issue_id, step_id, project_id):
-        canonical_path = "/public/rest/api/1.0/teststep/issueId/id?projectId="
         api_path = f"/public/rest/api/1.0/teststep/{issue_id}/{step_id}?projectId={project_id}"
-        return self._do_request(method="GET", api_path=api_path, canonical_path=canonical_path)
+        return self._do_request(method="GET", api_path=api_path)
 
     def update_test_step(self, issue_id, step_id, project_id, json):
-        canonical_path = "/public/rest/api/1.0/teststep/issueId/id?projectId="
         api_path = f"/public/rest/api/1.0/teststep/{issue_id}/{step_id}?projectId={project_id}"
-        return self._do_request(method="PUT", api_path=api_path, canonical_path=canonical_path, json=json)
+        return self._do_request(method="PUT", api_path=api_path, json=json)
 
     def delete_test_step(self, issue_id, step_id, project_id):
-        canonical_path = "/public/rest/api/1.0/teststep/issueId/id?projectId="
         api_path = f"/public/rest/api/1.0/teststep/{issue_id}/{step_id}?projectId={project_id}"
-        return self._do_request(method="DELETE", api_path=api_path, canonical_path=canonical_path)
+        return self._do_request(method="DELETE", api_path=api_path)
 
     def create_new_test_step(self, issue_id, project_id, json):
-        canonical_path = "/public/rest/api/1.0/teststep/issueId?projectId="
         api_path = f"/public/rest/api/1.0/teststep/{issue_id}?projectId={project_id}"
-        return self._do_request(method="POST", api_path=api_path, canonical_path=canonical_path, json=json)
+        return self._do_request(method="POST", api_path=api_path, json=json)
 
     def get_all_test_steps(self, issue_id, project_id):
-        canonical_path = "/public/rest/api/2.0/teststep/issueId?projectId="
         api_path = f"/public/rest/api/2.0/teststep/{issue_id}?projectId={project_id}"
-        return self._do_request(method='GET', api_path=api_path, canonical_path=canonical_path)
+        return self._do_request(method='GET', api_path=api_path)
 
     def get_all_test_step_statuses(self):
         api_path = "/public/rest/api/1.0/teststep/statuses"
         return self._do_request(method='GET', api_path=api_path)
 
-    def _do_request(self, method, api_path, canonical_path=None, json=None):
+    def get_bdd_content(self, issue_id):
+        api_path = f"/public/rest/api/1.0/integration/bddcontent/{issue_id}"
+        return self._do_request(method='GET', api_path=api_path)
+
+    def update_bdd_content(self, issue_id, new_content: str):
+        api_path = f"/public/rest/api/1.0/integration/bddcontent/{issue_id}"
+        return self._do_request(method='POST', api_path=api_path, json=json.dumps({"content": new_content}))
+
+    def delete_bdd_content(self, issue_id):
+        api_path = f"/public/rest/api/1.0/integration/bddcontent/{issue_id}"
+        return self._do_request(method='DELETE', api_path=api_path, json="[]")
+
+    def create_new_cycle(self, json):
+        api_path = f"public/rest/api/1.0/cycle"
+        return self._do_request(method='POST', api_path=api_path, json=json)
+
+    def create_folder(self, json):
+        api_path = f"/public/rest/api/1.0/folder"
+        return self._do_request(method='POST', api_path=api_path, json=json)
+
+    def add_test_to_cycle(self, cycle_id, json):
+        api_path = f"/public/rest/api/1.0/executions/add/cycle/{cycle_id}"
+        return self._do_request(method='POST', api_path=api_path, json=json)
+
+    def add_test_to_folder(self, folder_id, json):
+        api_path = f"/public/rest/api/1.0/executions/add/folder/{folder_id}"
+        return self._do_request(method='POST', api_path=api_path, json=json)
+
+    def create_execution(self, json):
+        api_path = f"/public/rest/api/1.0/execution"
+        return self._do_request(method='POST', api_path=api_path, json=json)
+
+    def get_execution(self, execution_id, issue_id, project_id):
+        api_path = f"/public/rest/api/1.0/execution/{execution_id}?issueId={issue_id}&projectId={project_id}"
+        return self._do_request(method='GET', api_path=api_path)
+
+    def _do_request(self, method, api_path, json=None):
         url = f"{self.base_url}{api_path}"
         headers = {
-            "Authorization": f"JWT {self._generate_jwt_token(method, canonical_path or api_path)}",
-            "zapiAccessKey": self.access_key,
-            "Content-Type": "application/json"
+            "Authorization": f"JWT {self._generate_jwt_token(method, api_path)}",
+            "zapiAccessKey": self.access_key
         }
+        if json is not None:
+            headers["Content-Type"] = "application/json"
+        params = {"method": method, "url": url, "headers": headers}
+        if json is not None:
+            params["data"] = json
 
         try:
-            resp = requests.request(method=method, url=url, json=json, headers=headers)
+            resp = requests.request(**params)
 
             if resp.ok:
                 if resp.headers.get("Content-Type", "").startswith("application/json"):
@@ -68,12 +104,16 @@ class ZephyrSquadCloud(object):
             raise ToolException(f"Error performing request {method}:{api_path}: {e}")
 
     def _generate_jwt_token(self, method, path):
-        canonical_path = f"{method}&{path}&"
+        canonical_path = f"{method}&{path}".replace('?', '&')
+
+        if '&' not in path and '?' not in path:
+            canonical_path += '&'
+
         payload_token = {
             'sub': self.account_id,
             'qsh': hashlib.sha256(canonical_path.encode('utf-8')).hexdigest(),
             'iss': self.access_key,
-            'exp': int(time.time()) + 3600,
+            'exp': int(time.time()) + 300,
             'iat': int(time.time())
         }
         return jwt.encode(payload_token, self.secret_key, algorithm='HS256').strip()

{alita_sdk-0.3.192.dist-info → alita_sdk-0.3.194.dist-info}/METADATA

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

@@ -63,7 +63,7 @@ alita_sdk/runtime/langchain/tools/bdd_parser/bdd_parser.py,sha256=DiEEOqDef2Xo3x
 alita_sdk/runtime/langchain/tools/bdd_parser/feature_types.py,sha256=l3AdjSQnNv1CE1NuHi7wts6h6AsCiK-iPu0PnPf3jf0,399
 alita_sdk/runtime/langchain/tools/bdd_parser/parser.py,sha256=1H1Nd_OH5Wx8A5YV1zUghBxo613yPptZ7fqNo8Eg48M,17289
 alita_sdk/runtime/llms/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-alita_sdk/runtime/llms/alita.py,sha256=aKvFQgKocwgHm90hngRhkZ6aaHC0-vXKXDWKSdqM6sY,10126
+alita_sdk/runtime/llms/alita.py,sha256=SHKlQ3LVrzISHbDl3BLinvTjPu28EMHuOtvwSYtuQ6c,10130
 alita_sdk/runtime/llms/preloaded.py,sha256=3AaUbZK3d8fvxAQMjR3ftOoYa0SnkCOL1EvdvDCXIHE,11321
 alita_sdk/runtime/toolkits/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/runtime/toolkits/application.py,sha256=akqUuaIL9u7-SsUmS-XgN4qxDEnXFhsK9do4n8inpSo,2432
@@ -104,11 +104,11 @@ alita_sdk/tools/ado/__init__.py,sha256=mD6GHcYMTtffPJkJvFPe2rzvye_IRmXmWfI7xYuZh
 alita_sdk/tools/ado/utils.py,sha256=PTCludvaQmPLakF2EbCGy66Mro4-rjDtavVP-xcB2Wc,1252
 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=cFZLntARKrCbWKKOCNz8TK6DbuRu_fKIAGqZEWAjc-g,4567
+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=Y2miBgjM6qCF5rWVzxCEGBNOT2rbHvlAmlIcadsCwZE,4265
+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=2MUgrxr4m8ipefuS3rstx-z4NAp2ntIAhE3CukXdMOk,4363
+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=7dSx2_GGL8tYb2_p7r7s3_zDiRv_5HaKR4HfSxqEUXg,2643
-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
@@ -294,11 +294,11 @@ alita_sdk/tools/zephyr_enterprise/api_wrapper.py,sha256=Ir3zHljhbZQJRJJQOBzS_GL5
 alita_sdk/tools/zephyr_enterprise/zephyr_enterprise.py,sha256=hV9LIrYfJT6oYp-ZfQR0YHflqBFPsUw2Oc55HwK0H48,6809
 alita_sdk/tools/zephyr_scale/__init__.py,sha256=2NTcdrfkx4GSegqyXhsPLsEpc4FlACuDy85b0fk6cAo,4572
 alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=UHVQUVqcBc3SZvDfO78HSuBzwAsRw2cCDQa-xMOzndE,68663
-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.192.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.192.dist-info/METADATA,sha256=FlJapDgNO4bUez880xdcswrOtQVwxL5lvOpd0-d8K0U,18804
-alita_sdk-0.3.192.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.192.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.192.dist-info/RECORD,,
+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.194.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.194.dist-info/METADATA,sha256=OJ_VfIlIR53zBYnpZL7P5LbBiZPoD9hkBozgBPwQXGw,18804
+alita_sdk-0.3.194.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.194.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.194.dist-info/RECORD,,