alita-sdk 0.3.193__py3-none-any.whl → 0.3.195__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/bitbucket/cloud_api_wrapper.py

@@ -286,10 +286,7 @@ class BitbucketCloudApi(BitbucketApiAbstract):
         Returns:
             List[Dict[str, Any]]: List of commits in the pull request
         """
-        commits = self.repository.pullrequests.get(pr_id).get('commits', [])
-        if not isinstance(commits, list):
-            commits = list(commits)
-        return commits
+        return self.repository.pullrequests.get(pr_id).get('commits', {}).get('values', [])
     
     def get_pull_request(self, pr_id: str) -> Any:
         """ Get details of a pull request
@@ -322,18 +319,11 @@ class BitbucketCloudApi(BitbucketApiAbstract):
         """
         # Build the content dict for Bitbucket Cloud
         if isinstance(content, str):
-            content_dict = {"raw": content}
-        elif isinstance(content, dict):
-            # Only include allowed keys
-            content_dict = {k: v for k, v in content.items() if k in ("raw", "markup", "html")}
-            if not content_dict:
-                content_dict = {"raw": str(content)}
+            content_raw = content
+        elif isinstance(content, dict) and "raw" in content:
+            content_raw = content.get("raw")
         else:
-            content_dict = {"raw": str(content)}
-
-        data = {"content": content_dict}
-        if inline:
-            data["inline"] = inline
+            content_raw = str(content)
 
-        response = self.repository.pullrequests.comments.post(pr_id, data=data)
+        response = self.repository.pullrequests.get(pr_id).comment(content_raw)
         return response['links']['self']['href']

alita_sdk/tools/confluence/api_wrapper.py

@@ -201,6 +201,8 @@ GetPageAttachmentsInput = create_model(
     page_id=(str, Field(description="Confluence page ID from which attachments will be retrieved")),
     max_content_length=(int, Field(default=10000, description="Maximum number of characters to return for attachment content. Content will be truncated if longer. Default is 10000.")),
     custom_prompt=(Optional[str], Field(default=None, description="Custom prompt to use for LLM-based analysis of attachments (images, pdfs, etc). If not provided, a default prompt will be used.")),
+    allowed_extensions=(Optional[List[str]], Field(default=None, description="List of file extensions to include (e.g. ['pdf', 'docx']). If None, all extensions are included.", examples=[["pdf", "docx"]])),
+    name_pattern=(Optional[str], Field(default=None, description="Regex pattern to filter attachment names (e.g. '^report_.*\\.pdf$'). If None, all names are included.", examples=["^report_.*\\.pdf$"])),
 )
 
 
@@ -1427,8 +1429,8 @@ class ConfluenceAPIWrapper(BaseVectorStoreToolApiWrapper):
             stacktrace = traceback.format_exc()
             logger.error(f"Error processing page with images: {stacktrace}")
             return f"Error processing page with images: {str(e)}"
-
-    def get_page_attachments(self, page_id: str, max_content_length: int = 10000, custom_prompt: str = None):
+    
+    def get_page_attachments(self, page_id: str, max_content_length: int = 10000, custom_prompt: str = None, allowed_extensions: Optional[List[str]] = None, name_pattern: Optional[str] = None):
         """
         Retrieve all attachments for a Confluence page, including core metadata (with creator, created, updated), comments,
         file content, and LLM-based analysis for supported types.
@@ -1449,8 +1451,19 @@ class ConfluenceAPIWrapper(BaseVectorStoreToolApiWrapper):
                     logger.warning(f"Failed to fetch history for attachment {attachment.get('title', '')}: {str(e)}")
                     history_map[attachment['id']] = None
 
+            import re
             results = []
             for attachment in attachments['results']:
+                title = attachment.get('title', '')
+                file_ext = title.lower().split('.')[-1] if '.' in title else ''
+
+                # Filter by allowed_extensions
+                if allowed_extensions and file_ext not in allowed_extensions:
+                    continue
+                # Filter by name_pattern
+                if name_pattern and not re.match(name_pattern, title):
+                    continue
+
                 media_type = attachment.get('metadata', {}).get('mediaType', '')
                 # Core metadata extraction with history
                 hist = history_map.get(attachment['id']) or {}
@@ -1458,7 +1471,7 @@ class ConfluenceAPIWrapper(BaseVectorStoreToolApiWrapper):
                 created_date = hist.get('createdDate', '') if hist else attachment.get('created', '')
                 last_updated = hist.get('lastUpdated', {}).get('when', '') if hist else ''
                 metadata = {
-                    'name': attachment.get('title', ''),
+                    'name': title,
                     'size': attachment.get('extensions', {}).get('fileSize', None),
                     'creator': created_by,
                     'created': created_date,
@@ -1467,6 +1480,7 @@ class ConfluenceAPIWrapper(BaseVectorStoreToolApiWrapper):
                     'labels': [label['name'] for label in attachment.get('metadata', {}).get('labels', {}).get('results', [])],
                     'download_url': self.base_url.rstrip('/') + attachment['_links']['download'] if attachment.get('_links', {}).get('download') else None
                 }
+
                 # Fetch comments for the attachment
                 comments = []
                 try:
@@ -1480,16 +1494,13 @@ class ConfluenceAPIWrapper(BaseVectorStoreToolApiWrapper):
                                 'body': comment.get('body', {}).get('storage', {}).get('value', '')
                             })
                 except Exception as e:
-                    logger.warning(f"Failed to fetch comments for attachment {attachment.get('title', '')}: {str(e)}")
+                    logger.warning(f"Failed to fetch comments for attachment {title}: {str(e)}")
 
                 content = None
                 llm_analysis = None
-                title = attachment.get('title', '')
                 download_url = self.base_url.rstrip('/') + attachment['_links']['download']
 
                 # --- Begin: Raw content for xml, json, markdown, txt ---
-                # Check by media type or file extension
-                file_ext = title.lower().split('.')[-1] if '.' in title else ''
                 is_text_type = (
                     media_type in [
                         'application/xml', 'text/xml',
@@ -1661,7 +1672,7 @@ class ConfluenceAPIWrapper(BaseVectorStoreToolApiWrapper):
                             llm_analysis = self._process_image_with_llm(image_data, title, context_text, custom_prompt)
 
                 if llm_analysis and isinstance(llm_analysis, str) and len(llm_analysis) > max_content_length:
-                        llm_analysis = llm_analysis[:max_content_length] + f"\n...[truncated, showing first {max_content_length} characters]"
+                    llm_analysis = llm_analysis[:max_content_length] + f"\n...[truncated, showing first {max_content_length} characters]"
 
                 results.append({
                     'metadata': metadata,

alita_sdk/tools/slack/__init__.py

@@ -44,14 +44,20 @@ class SlackToolkit(BaseToolkit):
             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 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}}}
-        )
+             name,
+             name=(str, Field(description="Toolkit name", json_schema_extra={'toolkit_name': True,
+                                                                             'max_toolkit_length': SlackToolkit.toolkit_max_length,
+                                                                             'configuration': True,
+                                                                             'configuration_title': 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[Literal[tuple(selected_tools)]],
+                             Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
+             __config__={'json_schema_extra': {'metadata': {"label": "Slack", "icon_url": "slack-icon.svg"}}}
+         )
          model.check_connection = check_connection
          return model
 

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.193.dist-info → alita_sdk-0.3.195.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.193
+Version: 0.3.195
 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.193.dist-info → alita_sdk-0.3.195.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
@@ -120,7 +120,7 @@ alita_sdk/tools/base/tool.py,sha256=-N27AodZS49vdPCgFkU-bFS9bxoPopZBnNrmwInx3d0,
 alita_sdk/tools/bitbucket/__init__.py,sha256=mPdq25Ib4yP1SV4g1dgK3WD3y-lKyuaNCmcIZTo893w,4355
 alita_sdk/tools/bitbucket/api_wrapper.py,sha256=iyBaNotbjzhoAOR8WZ9Ue-8vs1J17R5BjnwR2zEbkvM,9721
 alita_sdk/tools/bitbucket/bitbucket_constants.py,sha256=UsbhQ1iEvrKoxceTFPWTYhaXS1zSxbmjs1TwY0-P4gw,462
-alita_sdk/tools/bitbucket/cloud_api_wrapper.py,sha256=KUAt6CB4941nJwvdJqExWedHd8kLWncu3PcAmnS960w,13279
+alita_sdk/tools/bitbucket/cloud_api_wrapper.py,sha256=_ownv-CJ9T15OP4HKCuNbuVQzvoO_8I30-xHqSwRNbc,12920
 alita_sdk/tools/bitbucket/tools.py,sha256=zKBUq7t9zLa1EvhlVZzyVcZSvwvdcbtz0oslgPFZeeo,15307
 alita_sdk/tools/browser/__init__.py,sha256=iByi9uMGjd6v44SagIPTm5fu1vWnxIkjn3xsx86uRwI,5249
 alita_sdk/tools/browser/crawler.py,sha256=jhE35dU94eQLURSM-D50tspOqEMsiGzMDbYNqNSR2mU,2279
@@ -186,7 +186,7 @@ alita_sdk/tools/code/loaders/codesearcher.py,sha256=XoXXZtIQZhvjIwZlnl_4wVGHC-3s
 alita_sdk/tools/code/sonar/__init__.py,sha256=u8wpgXJ_shToLl3G9-XEtGDor5dhmsnurIImh1-e-U0,3165
 alita_sdk/tools/code/sonar/api_wrapper.py,sha256=nNqxcWN_6W8c0ckj-Er9HkNuAdgQLoWBXh5UyzNutis,2653
 alita_sdk/tools/confluence/__init__.py,sha256=1wGSP4CjPbfpdZLsjC1SVftTV7XSvW3QCAMHuh9tIxA,6885
-alita_sdk/tools/confluence/api_wrapper.py,sha256=zP8S85oVulhqrom1wlgelN-TSuKtPuYxwwGAlOSn_I0,88776
+alita_sdk/tools/confluence/api_wrapper.py,sha256=WgqiNanj9B5ebmaAWmUT_U_excGUWEguUEFR96K7oYg,89491
 alita_sdk/tools/confluence/loader.py,sha256=aHqgdIQMqkyRry8feHAhyd-a_ASEyW3JrV6epTRG6-c,9162
 alita_sdk/tools/confluence/utils.py,sha256=Lxo6dBD0OlvM4o0JuK6qeB_4LV9BptiwJA9e1vqNcDw,435
 alita_sdk/tools/custom_open_api/__init__.py,sha256=9aT5SPNPWcJC6jMZEM-3rUCXVULj_3-qJLQKmnreKNo,2537
@@ -270,7 +270,7 @@ 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=Y2sXN1__1923icF8bZbAtKVg3G_EZbCAjnj93yHUwtY,3415
+alita_sdk/tools/slack/__init__.py,sha256=RfOgQqo1ImecVL3LQln2gC9dA9QowJVlAetL15vL5qo,3920
 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
@@ -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.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,,
+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.195.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.195.dist-info/METADATA,sha256=XUqYON3Yv7eheVEMka2q1dGZrBlyzZjamd3b8QTot5M,18804
+alita_sdk-0.3.195.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.195.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.195.dist-info/RECORD,,