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

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

@@ -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 → alita_sdk-0.3.195}/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-0.3.193 → alita_sdk-0.3.195}/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-0.3.193 → alita_sdk-0.3.195}/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-0.3.193 → alita_sdk-0.3.195}/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-0.3.193 → alita_sdk-0.3.195}/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-0.3.195/alita_sdk/tools/zephyr_squad/api_wrapper.py

@@ -0,0 +1,319 @@
+from typing import List, Literal
+
+from pydantic import model_validator, create_model, Field, SecretStr, BaseModel, PrivateAttr
+
+from .zephyr_squad_cloud_client import ZephyrSquadCloud
+from ..elitea_base import BaseToolApiWrapper
+
+
+class ZephyrSquadApiWrapper(BaseToolApiWrapper):
+    account_id: str
+    access_key: str
+    secret_key: SecretStr
+    _client: ZephyrSquadCloud = PrivateAttr()
+
+    @model_validator(mode='before')
+    @classmethod
+    def validate_toolkit(cls, values):
+        account_id = values.get("account_id", None)
+        access_key = values.get("access_key", None)
+        secret_key = values.get("secret_key", None)
+        if not account_id:
+            raise ValueError("account_id is required.")
+        if not access_key:
+            raise ValueError("access_key is required.")
+        if not secret_key:
+            raise ValueError("secret_key is required.")
+        cls._client = ZephyrSquadCloud(
+            account_id=account_id,
+            access_key=access_key,
+            secret_key=secret_key
+        )
+        return values
+
+    def get_test_step(self, issue_id, step_id, project_id):
+        """Retrieve details for a specific test step in a Jira test case."""
+        return self._client.get_test_step(issue_id, step_id, project_id)
+
+    def update_test_step(self, issue_id, step_id, project_id, json):
+        """Update the content or a specific test step in a Jira test case."""
+        return self._client.update_test_step(issue_id, step_id, project_id, json)
+
+    def delete_test_step(self, issue_id, step_id, project_id):
+        """Remove a specific test step from a Jira test case."""
+        return self._client.delete_test_step(issue_id, step_id, project_id)
+
+    def create_new_test_step(self, issue_id, project_id, json):
+        """Add a new test step to a Jira test case."""
+        return self._client.create_new_test_step(issue_id, project_id, json)
+
+    def get_all_test_steps(self, issue_id, project_id):
+        """List all test steps associated with a Jira test case."""
+        return self._client.get_all_test_steps(issue_id, project_id)
+
+    def get_all_test_step_statuses(self):
+        """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 [
+            {
+                "name": "get_test_step",
+                "description": self.get_test_step.__doc__,
+                "args_schema": ProjectIssueStep,
+                "ref": self.get_test_step,
+            },
+            {
+                "name": "update_test_step",
+                "description": self.update_test_step.__doc__,
+                "args_schema": UpdateTestStep,
+                "ref": self.update_test_step,
+            },
+            {
+                "name": "delete_test_step",
+                "description": self.delete_test_step.__doc__,
+                "args_schema": ProjectIssueStep,
+                "ref": self.delete_test_step,
+            },
+            {
+                "name": "create_new_test_step",
+                "description": self.create_new_test_step.__doc__,
+                "args_schema": CreateNewTestStep,
+                "ref": self.create_new_test_step,
+            },
+            {
+                "name": "get_all_test_steps",
+                "description": self.get_all_test_steps.__doc__,
+                "args_schema": ProjectIssue,
+                "ref": self.get_all_test_steps,
+            },
+            {
+                "name": "get_all_test_step_statuses",
+                "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",
+    project_id=(int, Field(description="Jira project id to which test case belongs.")),
+    __base__=Issue
+)
+
+ProjectIssueStep = create_model(
+    "ProjectIssueStep",
+    step_id=(str, Field(description="Test step id to operate.")),
+    __base__=ProjectIssue
+)
+
+UpdateTestStep = create_model(
+    "UpdateTestStep",
+    json=(str, Field(description=(
+            "JSON body to update a Zephyr test step. Fields:\n"
+            "- id (string, required): Unique identifier for the test step. Example: \"0001481146115453-3a0480a3ffffc384-0001\"\n"
+            "- step (string, required): Description of the test step. Example: \"Sample Test Step\"\n"
+            "- data (string, optional): Test data used in this step. Example: \"Sample Test Data\"\n"
+            "- result (string, optional): Expected result after executing the step. Example: \"Expected Test Result\"\n"
+            "- customFieldValues (array[object], optional): List of custom field values for the test step. Each object contains:\n"
+            "    - customFieldId (string, required): ID of the custom field. Example: \"3ce1c679-7c43-4d37-89f6-757603379e31\"\n"
+            "    - value (object, required): Value for the custom field. Example: {\"value\": \"08/21/2018\"}\n"
+            "*IMPORTANT*: Use double quotes for all field names and string values."))),
+    __base__=ProjectIssueStep
+)
+
+CreateNewTestStep = create_model(
+    "CreateNewTestStep",
+    json=(str, Field(description=(
+            "JSON body to create a Zephyr test step. Fields:\n"
+            "- step (string, required): Description of the test step. Example: \"Sample Test Step\"\n"
+            "- data (string, optional): Test data used in this step. Example: \"Sample Test Data\"\n"
+            "- result (string, optional): Expected result after executing the step. Example: \"Expected Test Result\"\n"
+            "*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-0.3.193 → alita_sdk-0.3.195}/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 → alita_sdk-0.3.195}/alita_sdk.egg-info/PKG-INFO

@@ -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