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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.193
+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.193 → alita_sdk-0.3.194}/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.194}/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.194/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.194}/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.194}/alita_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.193
+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.193 → alita_sdk-0.3.194}/pyproject.toml

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

alita_sdk-0.3.193/alita_sdk/tools/zephyr_squad/api_wrapper.py

@@ -1,135 +0,0 @@
-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_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,
-            }
-        ]
-
-
-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."))
-)
-
-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
-)