alita-sdk 0.3.256__py3-none-any.whl → 0.3.258__py3-none-any.whl

alita_sdk/configurations/github.py

@@ -1,6 +1,6 @@
 from typing import Optional
 
-from pydantic import BaseModel, ConfigDict, Field, SecretStr
+from pydantic import BaseModel, ConfigDict, Field, SecretStr, model_validator
 
 
 class GithubConfiguration(BaseModel):
@@ -35,6 +35,7 @@ class GithubConfiguration(BaseModel):
             }
         }
     )
+
     base_url: Optional[str] = Field(description="Base API URL", default="https://api.github.com")
     app_id: Optional[str] = Field(description="Github APP ID", default=None)
     app_private_key: Optional[SecretStr] = Field(description="Github APP private key", default=None)
@@ -43,3 +44,44 @@ class GithubConfiguration(BaseModel):
 
     username: Optional[str] = Field(description="Github Username", default=None)
     password: Optional[SecretStr] = Field(description="Github Password", default=None)
+
+    @model_validator(mode='before')
+    @classmethod
+    def validate_auth_sections(cls, data):
+        if not isinstance(data, dict):
+            return data
+
+        has_token = bool(data.get('access_token') and str(data.get('access_token')).strip())
+        has_password = bool(
+            data.get('username') and str(data.get('username')).strip() and
+            data.get('password') and str(data.get('password')).strip()
+        )
+        has_app_key = bool(
+            data.get('app_id') and str(data.get('app_id')).strip() and
+            data.get('app_private_key') and str(data.get('app_private_key')).strip()
+        )
+
+        # If any method is partially configured, raise exception
+        if (
+                (data.get('username') and not data.get('password')) or
+                (data.get('password') and not data.get('username')) or
+                (data.get('app_id') and not data.get('app_private_key')) or
+                (data.get('app_private_key') and not data.get('app_id'))
+        ):
+            raise ValueError(
+                "Authentication is misconfigured: both username and password, or both app_id and app_private_key, must be provided together."
+            )
+
+        # If all are missing, allow anonymous
+        if not (has_token or has_password or has_app_key):
+            return data
+
+        # If any method is fully configured
+        if has_token or has_password or has_app_key:
+            return data
+
+        raise ValueError(
+            "Authentication is misconfigured: provide either Token (access_token), "
+            "Password (username + password), App private key (app_id + app_private_key), "
+            "or leave all blank for anonymous access."
+        )

alita_sdk/runtime/clients/client.py

@@ -364,8 +364,8 @@ class AlitaClient:
         return data.content
 
     def delete_artifact(self, bucket_name, artifact_name):
-        url = f'{self.artifact_url}/{bucket_name}/{quote(artifact_name)}'
-        data = requests.delete(url, headers=self.headers, verify=False)
+        url = f'{self.artifact_url}/{bucket_name}'
+        data = requests.delete(url, headers=self.headers, verify=False, params={'filename': quote(artifact_name)})
         return self._process_requst(data)
 
     def _prepare_messages(self, messages: list[BaseMessage]):

alita_sdk/tools/ado/__init__.py

@@ -14,11 +14,12 @@ def get_tools(tool_type, tool):
         "limit": tool['settings'].get('limit', 5),
         "toolkit_name": tool.get('toolkit_name', ''),
         # indexer settings
+        "alita": tool['settings'].get('alita', None),
         "llm": tool['settings'].get('llm', None),
         "pgvector_configuration": tool['settings'].get('pgvector_configuration', {}),
         "collection_name": tool['toolkit_name'],
         "doctype": 'doc',
-        "embedding_model": tool['settings'].get('embedding_configuration', {}).get('name', None),
+        "embedding_model": tool['settings'].get('embedding_model'),
         "vectorstore_type": "PGVector"
     }
     if tool_type == 'ado_plans':

alita_sdk/tools/ado/repos/__init__.py

@@ -23,10 +23,9 @@ def _get_toolkit(tool) -> BaseToolkit:
         active_branch=tool['settings'].get('active_branch', ""),
         toolkit_name=tool['settings'].get('toolkit_name', ""),
         pgvector_configuration=tool['settings'].get('pgvector_configuration', {}),
-        collection_name=tool['toolkit_name'],
-        doctype='code',
         embedding_model=tool['settings'].get('embedding_model'),
-        vectorstore_type="PGVector",
+        collection_name=tool['toolkit_name'],
+        alita=tool['settings'].get('alita', None),
     )
 
 def get_toolkit():
@@ -62,11 +61,6 @@ class AzureDevOpsReposToolkit(BaseToolkit):
                     "icon_url": "ado-repos-icon.svg",
                     "categories": ["code repositories"],
                     "extra_categories": ["code", "repository", "version control"],
-                    # "configuration_group": {
-                    #     "name": "ado_repos",
-                    #     "label": "Azure DevOps Repositories",
-                    #     "icon_url": "ado-repos-icon.svg",
-                    # }
                 }}}
         )
 
@@ -95,6 +89,7 @@ class AzureDevOpsReposToolkit(BaseToolkit):
             **kwargs,
             # TODO use ado_repos_configuration fields
             **kwargs['ado_repos_configuration'],
+            **kwargs['ado_repos_configuration']['ado_configuration'],
             **(kwargs.get('pgvector_configuration') or {}),
         }
         azure_devops_repos_wrapper = ReposApiWrapper(**wrapper_payload)

alita_sdk/tools/ado/repos/repos_wrapper.py

@@ -251,15 +251,6 @@ class ReposApiWrapper(BaseCodeToolApiWrapper):
     token: Optional[SecretStr]
     _client: Optional[GitClient] = PrivateAttr()
 
-    llm: Optional[Any] = None
-    # Vector store configuration
-    connection_string: Optional[SecretStr] = None
-    collection_name: Optional[str] = None
-    doctype: Optional[str] = 'code'
-    embedding_model: Optional[str] = "HuggingFaceEmbeddings"
-    embedding_model_params: Optional[dict] = {"model_name": "sentence-transformers/all-MiniLM-L6-v2"}
-    vectorstore_type: Optional[str] = "PGVector"
-
     class Config:
         arbitrary_types_allowed = True
 

alita_sdk/tools/ado/test_plan/test_plan_wrapper.py

@@ -16,11 +16,6 @@ from pydantic.fields import FieldInfo as Field
 from ..work_item import AzureDevOpsApiWrapper
 from ...elitea_base import BaseVectorStoreToolApiWrapper, extend_with_vector_tools
 
-try:
-    from alita_sdk.runtime.langchain.interfaces.llm_processor import get_embeddings
-except ImportError:
-    from alita_sdk.langchain.interfaces.llm_processor import get_embeddings
-
 logger = logging.getLogger(__name__)
 
 # Input models for Test Plan operations

alita_sdk/tools/ado/wiki/ado_wrapper.py

@@ -17,11 +17,6 @@ from pydantic.fields import Field
 
 from ...elitea_base import BaseVectorStoreToolApiWrapper, extend_with_vector_tools
 
-try:
-    from alita_sdk.runtime.langchain.interfaces.llm_processor import get_embeddings
-except ImportError:
-    from alita_sdk.langchain.interfaces.llm_processor import get_embeddings
-
 logger = logging.getLogger(__name__)
 
 GetWikiInput = create_model(
@@ -68,13 +63,6 @@ class AzureDevOpsApiWrapper(BaseVectorStoreToolApiWrapper):
     _client: Optional[WikiClient] = PrivateAttr()  # Private attribute for the wiki client
     _core_client: Optional[CoreClient] = PrivateAttr()  # Private attribute for the CoreClient client
 
-    llm: Any = None
-    connection_string: Optional[SecretStr] = None
-    collection_name: Optional[str] = None
-    embedding_model: Optional[str] = "HuggingFaceEmbeddings"
-    embedding_model_params: Optional[Dict[str, Any]] = {"model_name": "sentence-transformers/all-MiniLM-L6-v2"}
-    vectorstore_type: Optional[str] = "PGVector"
-
     class Config:
         arbitrary_types_allowed = True  # Allow arbitrary types (e.g., WorkItemTrackingClient)
 

alita_sdk/tools/ado/work_item/ado_wrapper.py

@@ -16,11 +16,6 @@ from pydantic.fields import Field
 
 from alita_sdk.tools.non_code_indexer_toolkit import NonCodeIndexerToolkit
 
-try:
-    from alita_sdk.runtime.langchain.interfaces.llm_processor import get_embeddings
-except ImportError:
-    from alita_sdk.langchain.interfaces.llm_processor import get_embeddings
-
 logger = logging.getLogger(__name__)
 
 create_wi_field = """JSON of the work item fields to create in Azure DevOps, i.e.

alita_sdk/tools/base_indexer_toolkit.py

@@ -5,7 +5,6 @@ from typing import Any, Optional, List, Literal, Dict, Generator
 from langchain_core.documents import Document
 from pydantic import create_model, Field, SecretStr
 
-# from alita_sdk.runtime.langchain.interfaces.llm_processor import get_embeddings
 from .utils.content_parser import process_content_by_type
 from .vector_adapters.VectorStoreAdapter import VectorStoreAdapterFactory
 from ..runtime.tools.vectorstore_base import VectorStoreWrapperBase
@@ -95,7 +94,6 @@ BaseIndexDataParams = create_model(
                          description="Optional step size for progress reporting during indexing")),
     clean_index=(Optional[bool], Field(default=False,
                        description="Optional flag to enforce clean existing index before indexing new data")),
-    chunking_tool=(Literal[None,'markdown', 'statistical', 'proposal'], Field(description="Name of chunking tool", default=None)),
     chunking_config=(Optional[dict], Field(description="Chunking tool configuration", default_factory=dict)),
 )
 
@@ -162,7 +160,7 @@ class BaseIndexerToolkit(VectorStoreWrapperBase):
         chunking_config = kwargs.get("chunking_config")
         #
         if clean_index:
-            self._clean_index()
+            self._clean_index(collection_suffix)
         #
         documents = self._base_loader(**kwargs)
         documents = self._reduce_duplicates(documents, collection_suffix)
@@ -173,7 +171,7 @@ class BaseIndexerToolkit(VectorStoreWrapperBase):
         return self._save_index(list(documents), collection_suffix=collection_suffix, progress_step=progress_step)
     
     def _apply_loaders_chunkers(self, documents: Generator[Document, None, None], chunking_tool: str=None, chunking_config=None) -> Generator[Document, None, None]:
-        from alita_sdk.tools.chunkers import __confluence_chunkers__ as chunkers
+        from alita_sdk.tools.chunkers import __all__ as chunkers
 
         if chunking_config is None:
             chunking_config = {}

alita_sdk/tools/bitbucket/tools.py

@@ -134,7 +134,7 @@ class ListBranchesTool(BaseTool):
     name: str = "list_branches_in_repo"
     description: str = """This tool is a wrapper for the Bitbucket API to fetch a list of all branches in the repository. 
     It will return the name of each branch. No input parameters are required."""
-    args_schema: Type[BaseModel] = None
+    args_schema: Type[BaseModel] = create_model("NoInput")
 
     def _run(self):
         try:

alita_sdk/tools/chunkers/__init__.py

@@ -2,13 +2,15 @@ from .code.codeparser import parse_code_files_for_db
 from .sematic.statistical_chunker import statistical_chunker
 from .sematic.markdown_chunker import markdown_chunker
 from .sematic.proposal_chunker import proposal_chunker
+from .sematic.json_chunker import json_chunker
 from .models import StatisticalChunkerConfig, MarkdownChunkerConfig, ProposalChunkerConfig
 
 __all__ = {
     'code_parser': parse_code_files_for_db,
     'statistical': statistical_chunker,
     'markdown': markdown_chunker,
-    'proposal': proposal_chunker
+    'proposal': proposal_chunker,
+    'json': json_chunker
 }
 
 __confluence_chunkers__ = {

alita_sdk/tools/chunkers/sematic/json_chunker.py

@@ -0,0 +1,24 @@
+import json
+import logging
+from typing import Generator
+from langchain_text_splitters import RecursiveJsonSplitter
+from langchain_core.documents import Document
+
+def json_chunker(file_content_generator: Generator[Document, None, None], config: dict, *args, **kwargs) -> Generator[Document, None, None]:
+    max_tokens = config.get("max_tokens", 512)
+    for doc in file_content_generator:
+        try:
+            data_dict = json.loads(doc.page_content)
+            chunks = RecursiveJsonSplitter(max_chunk_size=max_tokens).split_json(json_data=data_dict, convert_lists=True)
+            if len(chunks) == 1:
+                yield doc
+                continue
+            chunk_id = 1
+            for chunk in chunks:
+                metadata = doc.metadata.copy()
+                metadata['chunk_id'] = chunk_id
+                chunk_id += 1
+                yield Document(page_content=json.dumps(chunk), metadata=metadata)
+        except Exception as e:
+            logging.error(f"Failed to chunk document: {e}")
+            yield doc

alita_sdk/tools/jira/api_wrapper.py

@@ -1237,11 +1237,11 @@ class JiraApiWrapper(BaseVectorStoreToolApiWrapper):
         jql = kwargs.get('jql')
         fields_to_extract = kwargs.get('fields_to_extract')
         fields_to_index = kwargs.get('fields_to_index')
-        include_attachments = kwargs.get('include_attachments', False)
         max_total_issues = kwargs.get('max_total_issues', 1000)
 
-        # set values for skipped attachment extensions
+        # set values for skipped attachment extension
         self._skipped_attachment_extensions = kwargs.get('skip_attachment_extensions', [])
+        self._include_attachments = kwargs.get('include_attachments', False)
         self._included_fields = fields_to_extract.copy() if fields_to_extract else []
 
         try:
@@ -1252,7 +1252,7 @@ class JiraApiWrapper(BaseVectorStoreToolApiWrapper):
             if fields_to_extract:
                 fields.extend(fields_to_extract)
 
-            if include_attachments:
+            if self._include_attachments:
                 fields.append('attachment')
 
             # Use provided JQL query or default to all issues
@@ -1292,36 +1292,36 @@ class JiraApiWrapper(BaseVectorStoreToolApiWrapper):
 
         issue_key = base_document.metadata.get('issue_key')
         # get attachments content
-
-        issue = self._client.issue(issue_key, fields="attachment")
-        attachments = issue.get('fields', {}).get('attachment', [])
-        for attachment in attachments:
-            # get extension
-            ext = f".{attachment['filename'].split('.')[-1].lower()}"
-            if ext not in self._skipped_attachment_extensions:
-                attachment_id = f"attach_{attachment['id']}"
-                base_document.metadata.setdefault(IndexerKeywords.DEPENDENT_DOCS.value, []).append(attachment_id)
-                try:
-                    attachment_content = self._client.get_attachment_content(attachment['id'])
-                except Exception as e:
-                    logger.error(f"Failed to download attachment {attachment['filename']} for issue {issue_key}: {str(e)}")
-                    attachment_content = self._client.get(path=f"secure/attachment/{attachment['id']}/{attachment['filename']}", not_json_response=True)
-                content = load_content_from_bytes(attachment_content, ext, llm=self.llm) if ext not in '.pdf' \
-                    else parse_file_content(file_content=attachment_content, file_name=attachment['filename'], llm=self.llm, is_capture_image=True)
-                if not content:
-                    continue
-                yield Document(page_content=content,
-                               metadata={
-                                   'id': attachment_id,
-                                   'issue_key': issue_key,
-                                   'source': f"{self.base_url}/browse/{issue_key}",
-                                   'filename': attachment['filename'],
-                                   'created': attachment['created'],
-                                   'mimeType': attachment['mimeType'],
-                                   'author': attachment.get('author', {}).get('name'),
-                                   IndexerKeywords.PARENT.value: base_document.metadata.get('id', None),
-                                   'type': 'attachment',
-                               })
+        if self._include_attachments:
+            issue = self._client.issue(issue_key, fields="attachment")
+            attachments = issue.get('fields', {}).get('attachment', [])
+            for attachment in attachments:
+                # get extension
+                ext = f".{attachment['filename'].split('.')[-1].lower()}"
+                if ext not in self._skipped_attachment_extensions:
+                    attachment_id = f"attach_{attachment['id']}"
+                    base_document.metadata.setdefault(IndexerKeywords.DEPENDENT_DOCS.value, []).append(attachment_id)
+                    try:
+                        attachment_content = self._client.get_attachment_content(attachment['id'])
+                    except Exception as e:
+                        logger.error(f"Failed to download attachment {attachment['filename']} for issue {issue_key}: {str(e)}")
+                        attachment_content = self._client.get(path=f"secure/attachment/{attachment['id']}/{attachment['filename']}", not_json_response=True)
+                    content = load_content_from_bytes(attachment_content, ext, llm=self.llm) if ext not in '.pdf' \
+                        else parse_file_content(file_content=attachment_content, file_name=attachment['filename'], llm=self.llm, is_capture_image=True)
+                    if not content:
+                        continue
+                    yield Document(page_content=content,
+                                   metadata={
+                                       'id': attachment_id,
+                                       'issue_key': issue_key,
+                                       'source': f"{self.base_url}/browse/{issue_key}",
+                                       'filename': attachment['filename'],
+                                       'created': attachment['created'],
+                                       'mimeType': attachment['mimeType'],
+                                       'author': attachment.get('author', {}).get('name'),
+                                       IndexerKeywords.PARENT.value: base_document.metadata.get('id', None),
+                                       'type': 'attachment',
+                                   })
 
     def _jql_get_tickets(self, jql, fields="*all", start=0, limit=None, expand=None, validate_query=None):
         """
@@ -1430,9 +1430,9 @@ class JiraApiWrapper(BaseVectorStoreToolApiWrapper):
                                     Field(description="Whether to include attachment content in indexing",
                                           default=False)),
             'max_total_issues': (Optional[int], Field(description="Maximum number of issues to index", default=1000)),
-            'skip_attachment_extensions': (Optional[str], Field(
-                description="Comma-separated list of file extensions to skip when processing attachments",
-                default=None)),
+            'skip_attachment_extensions': (Optional[List[str]], Field(
+                description="List of file extensions to skip when processing attachments: i.e. ['.png', '.jpg']",
+                default=[])),
         }
 
     # def index_data(self,

alita_sdk/tools/qtest/__init__.py

@@ -35,7 +35,7 @@ class QtestToolkit(BaseToolkit):
             name,
             qtest_configuration=(Optional[QtestConfiguration], Field(description="QTest API token", json_schema_extra={
                 'configuration_types': ['qtest']})),
-            qtest_project_id=(int, Field(description="QTest project id", json_schema_extra={'toolkit_name': True,
+            qtest_project_id=(int, Field(default=None, description="QTest project id", json_schema_extra={'toolkit_name': True,
                                                                                             'max_toolkit_length': QtestToolkit.toolkit_max_length})),
             selected_tools=(List[Literal[tuple(selected_tools)]],
                             Field(default=[], json_schema_extra={'args_schemas': selected_tools})),

alita_sdk/tools/utils/available_tools_decorator.py

@@ -0,0 +1,6 @@
+def extend_with_parent_available_tools(method):
+    def wrapper(self, *args, **kwargs):
+        child_tools = method(self, *args, **kwargs)
+        parent_tools = super(self.__class__, self).get_available_tools()
+        return parent_tools + child_tools
+    return wrapper

alita_sdk/tools/xray/__init__.py

@@ -17,9 +17,7 @@ name = "xray_cloud"
 def get_tools(tool):
     return XrayToolkit().get_toolkit(
         selected_tools=tool['settings'].get('selected_tools', []),
-        base_url=tool['settings'].get('base_url', None),
-        client_id=tool['settings'].get('client_id', None),
-        client_secret=tool['settings'].get('client_secret', None),
+        xray_configuration=tool['settings'].get('xray_configuration', {}),
         limit=tool['settings'].get('limit', 20),
         verify_ssl=tool['settings'].get('verify_ssl', True),
         toolkit_name=tool.get('toolkit_name'),
@@ -29,7 +27,6 @@ def get_tools(tool):
         pgvector_configuration=tool['settings'].get('pgvector_configuration', {}),
         embedding_model=tool['settings'].get('embedding_model'),
         collection_name=str(tool['toolkit_name']),
-        vectorstore_type="PGVector"
     ).get_tools()
 
 
@@ -44,7 +41,7 @@ class XrayToolkit(BaseToolkit):
         return create_model(
             name,
             limit=(Optional[int], Field(description="Limit", default=100)),
-            xray_configuration=(Optional[XrayConfiguration], Field(description="Xray Configuration", json_schema_extra={'configuration_types': ['xray']})),
+            xray_configuration=(XrayConfiguration, Field(description="Xray Configuration", json_schema_extra={'configuration_types': ['xray']})),
             pgvector_configuration=(Optional[PgVectorConfiguration], Field(default=None,
                                                                            description="PgVector Configuration",
                                                                            json_schema_extra={
@@ -71,7 +68,7 @@ class XrayToolkit(BaseToolkit):
         wrapper_payload = {
             **kwargs,
             # Use xray_configuration fields
-            **kwargs.get('xray_configuration', {}),
+            **(kwargs.get('xray_configuration') or {}),
             **(kwargs.get('pgvector_configuration') or {}),
         }
         xray_api_wrapper = XrayApiWrapper(**wrapper_payload)

alita_sdk/tools/zephyr_enterprise/api_wrapper.py

@@ -1,14 +1,13 @@
 import logging
-from importlib.metadata import metadata
-from operator import ifloordiv
-from typing import Optional, List, Generator
+from typing import Optional, List, Generator, Literal
 
 from langchain_core.tools import ToolException
 from pydantic import create_model, model_validator, PrivateAttr, Field, SecretStr
 
 from langchain_core.documents import Document
 from .zephyr_enterprise import ZephyrClient
-from ..elitea_base import BaseToolApiWrapper, BaseVectorStoreToolApiWrapper, extend_with_vector_tools
+from ..non_code_indexer_toolkit import NonCodeIndexerToolkit
+from ..utils.available_tools_decorator import extend_with_parent_available_tools
 
 logger = logging.getLogger(__name__)
 
@@ -23,7 +22,7 @@ zql_description = """
                     "folder=\"TestToolkit\"", "name~\"TestToolkit5\"
                     """
 
-class ZephyrApiWrapper(BaseVectorStoreToolApiWrapper):
+class ZephyrApiWrapper(NonCodeIndexerToolkit):
     base_url: str
     token: SecretStr
     _client: Optional[ZephyrClient] = PrivateAttr()
@@ -34,7 +33,7 @@ class ZephyrApiWrapper(BaseVectorStoreToolApiWrapper):
         base_url = values.get('base_url')
         token = values.get('token')
         cls._client = ZephyrClient(base_url=base_url, token=token)
-        return values
+        return super().validate_toolkit(values)
 
     def get_test_case(self, testcase_id: str):
 
@@ -153,7 +152,8 @@ class ZephyrApiWrapper(BaseVectorStoreToolApiWrapper):
         Returns a list of fields for index_data args schema.
         """
         return {
-            "zql": (str, Field(description=zql_description, examples=["folder=\"TestToolkit\"", "name~\"TestToolkit5\""]))
+            "zql": (str, Field(description=zql_description, examples=["folder=\"TestToolkit\"", "name~\"TestToolkit5\""])),
+            'chunking_tool': (Literal['json'], Field(description="Name of chunking tool", default='json'))
         }
 
     def _base_loader(self, zql: str, **kwargs) -> Generator[Document, None, None]:
@@ -166,12 +166,17 @@ class ZephyrApiWrapper(BaseVectorStoreToolApiWrapper):
             }
             yield Document(page_content='', metadata=metadata)
 
-    def _process_document(self, document: Document) -> Generator[Document, None, None]:
-        id = document.metadata['id']
-        test_case_content = self.get_test_case_steps(id)
-        document.page_content = test_case_content
-
-    @extend_with_vector_tools
+    def _extend_data(self, documents: Generator[Document, None, None]) -> Generator[Document, None, None]:
+        for document in documents:
+            try:
+                id = document.metadata['id']
+                test_case_content = self.get_test_case_steps(id)
+                document.page_content = test_case_content
+            except Exception as e:
+                logging.error(f"Failed to process document: {e}")
+            yield document
+
+    @extend_with_parent_available_tools
     def get_available_tools(self):
         return [
             {

alita_sdk/tools/zephyr_essential/api_wrapper.py

@@ -1,13 +1,17 @@
 import json
-from typing import Optional, Generator
+import logging
+from typing import Optional, Generator, Literal
 from pydantic import model_validator, create_model, Field, SecretStr, PrivateAttr
 
 from .client import ZephyrEssentialAPI
-from ..elitea_base import extend_with_vector_tools, BaseVectorStoreToolApiWrapper
 from langchain_core.documents import Document
 from langchain_core.tools import ToolException
 
-class ZephyrEssentialApiWrapper(BaseVectorStoreToolApiWrapper):
+from ..non_code_indexer_toolkit import NonCodeIndexerToolkit
+from ..utils.available_tools_decorator import extend_with_parent_available_tools
+
+
+class ZephyrEssentialApiWrapper(NonCodeIndexerToolkit):
     token: SecretStr
     _client: ZephyrEssentialAPI = PrivateAttr()
 
@@ -22,7 +26,7 @@ class ZephyrEssentialApiWrapper(BaseVectorStoreToolApiWrapper):
             base_url=base_url,
             token=token
         )
-        return values
+        return super().validate_toolkit(values)
 
     def list_test_cases(self, project_key: Optional[str] = None, folder_id: Optional[str] = None, max_results: int = None, start_at: int = None):
         """List test cases with optional filters."""
@@ -229,6 +233,11 @@ class ZephyrEssentialApiWrapper(BaseVectorStoreToolApiWrapper):
         except json.JSONDecodeError as e:
             raise ValueError(f"Invalid JSON string: {str(e)}")
 
+    def _index_tool_params(self):
+        return {
+            'chunking_tool':(Literal['json'], Field(description="Name of chunking tool", default='json'))
+        }
+
     def _base_loader(self, **kwargs) -> Generator[Document, None, None]:
         try:
             test_cases = self.list_test_cases()
@@ -236,36 +245,37 @@ class ZephyrEssentialApiWrapper(BaseVectorStoreToolApiWrapper):
             raise ToolException(f"Unable to extract test cases: {e}")
 
         for case in test_cases:
-            case['type'] = "TEST_CASE"
             metadata = {
                 k: v for k, v in case.items()
                 if isinstance(v, (str, int, float, bool, list, dict))
             }
-
-            yield Document(page_content=json.dumps(case), metadata=metadata)
-
-    def _process_document(self, document: Document) -> Generator[Document, None, None]:
-        try:
-            base_data = json.loads(document.page_content)
-
-            if base_data['type'] and base_data['type'] == "TEST_CASE":
-                additional_content = self._process_test_case(base_data)
-                base_data['test_case_content'] = additional_content
-
-            document.page_content = json.dumps(base_data)
-        except json.JSONDecodeError as e:
-            raise ToolException(f"Failed to decode JSON from document: {e}")
-
-    def _process_test_case(self, case):
-        steps = self.get_test_case_test_steps(case['key'])
-        script = self.get_test_case_test_script(case['key'])
-        additional_content = {
-            "steps": "" if isinstance(steps, ToolException) else steps,
-            "script": "" if isinstance(script, ToolException) else script,
-        }
-        return additional_content
-
-    @extend_with_vector_tools
+            metadata['type'] = "TEST_CASE"
+
+            yield Document(page_content="", metadata=metadata)
+
+    def _extend_data(self, documents: Generator[Document, None, None]) -> Generator[Document, None, None]:
+        for document in documents:
+            try:
+                if 'type' in document.metadata and document.metadata['type'] == "TEST_CASE":
+                    additional_content = self._process_test_case(document.metadata['key'])
+                    for steps_type, content in additional_content.items():
+                        if content:
+                            document.page_content = json.dumps(content)
+                            document.metadata["steps_type"] = steps_type
+            except Exception as e:
+                logging.error(f"Failed to process document: {e}")
+            yield document
+
+    def _process_test_case(self, key) -> dict:
+        steps = self.get_test_case_test_steps(key)
+        if steps and not isinstance(steps, ToolException):
+            return {"steps": steps}
+        script = self.get_test_case_test_script(key)
+        if script and not isinstance(script, ToolException):
+            return {"script": script}
+        return {"empty": ""}
+
+    @extend_with_parent_available_tools
     def get_available_tools(self):
         return [
             {

alita_sdk/tools/zephyr_scale/api_wrapper.py

@@ -1,15 +1,18 @@
 import json
 import logging
 import re
-from typing import Any, Optional, List, Dict, Tuple, Union, Generator
+from typing import Any, Optional, List, Dict, Tuple, Union, Generator, Literal
 
 from pydantic import model_validator, BaseModel, SecretStr
 from langchain_core.tools import ToolException
 from pydantic import create_model, PrivateAttr
 from pydantic.fields import Field
 
-from ..elitea_base import BaseVectorStoreToolApiWrapper, BaseIndexParams, extend_with_vector_tools
 from langchain_core.documents import Document
+
+from ..non_code_indexer_toolkit import NonCodeIndexerToolkit
+from ..utils.available_tools_decorator import extend_with_parent_available_tools
+
 try:
     from alita_sdk.runtime.langchain.interfaces.llm_processor import get_embeddings
 except ImportError:
@@ -249,7 +252,7 @@ ZephyrUpdateTestSteps = create_model(
 )
 
 
-class ZephyrScaleApiWrapper(BaseVectorStoreToolApiWrapper):
+class ZephyrScaleApiWrapper(NonCodeIndexerToolkit):
     # url for a Zephyr server
     base_url: Optional[str] = ""
     # auth with Jira token (cloud & server)
@@ -296,7 +299,7 @@ class ZephyrScaleApiWrapper(BaseVectorStoreToolApiWrapper):
         # else:
         # Cloud version is enabled for now
         cls._api = ZephyrScale(token=values['token']).api
-        return values
+        return super().validate_toolkit(values)
 
     def get_tests(self, project_key: str = None, folder_id: str = None, maxResults: Optional[int] = 10, startAt: Optional[int] = 0):
         """Retrieves all test cases. Query parameters can be used to filter the results.
@@ -1210,7 +1213,8 @@ class ZephyrScaleApiWrapper(BaseVectorStoreToolApiWrapper):
         
             Example:
                 'folder = "Authentication" AND label in ("Smoke", "Critical") AND text ~ "login" AND orderBy = "name" AND orderDirection = "ASC"'
-            """))
+            """)),
+            'chunking_tool': (Literal['json'], Field(description="Name of chunking tool", default='json'))
         }
 
     def _base_loader(self, project_key: str, jql: str, **kwargs) -> Generator[Document, None, None]:
@@ -1250,8 +1254,8 @@ class ZephyrScaleApiWrapper(BaseVectorStoreToolApiWrapper):
             for key, value in folder.items():
                 if value is not None:
                     metadata[key] = value
-            page_content['type'] = "FOLDER"
-            yield Document(page_content=json.dumps(page_content), metadata=metadata)
+            metadata['type'] = "FOLDER"
+            yield Document(page_content="", metadata=metadata)
 
     def _get_test_cases_docs(self, project_key: str, jql: str) -> Generator[Document, None, None]:
         try:
@@ -1269,31 +1273,31 @@ class ZephyrScaleApiWrapper(BaseVectorStoreToolApiWrapper):
                 metadata['updated_on'] = last_version['createdOn']
             else:
                 metadata['updated_on'] = case['createdOn']
+            metadata['type'] = "TEST_CASE"
 
-            case['type'] = "TEST_CASE"
-
-            yield Document(page_content=json.dumps(case), metadata=metadata)
-
-    def _process_document(self, document: Document) -> Generator[Document, None, None]:
-        try:
-            base_data = json.loads(document.page_content)
-
-            if base_data['type'] and base_data['type'] == "TEST_CASE":
-                additional_content = self._process_test_case(base_data)
-                base_data['test_case_content'] = additional_content
+            yield Document(page_content="", metadata=metadata)
 
-            document.page_content = json.dumps(base_data)
-        except json.JSONDecodeError as e:
-            raise ToolException(f"Failed to decode JSON from document: {e}")
-
-    def _process_test_case(self, case):
-        steps = self.get_test_steps(case['key'], return_list=True)
-        script = self.get_test_script(case['key'], return_only_script=True)
-        additional_content = {
-            "steps": "" if isinstance(steps, ToolException) else steps,
-            "script": "" if isinstance(script, ToolException) else script,
-        }
-        return additional_content
+    def _extend_data(self, documents: Generator[Document, None, None]) -> Generator[Document, None, None]:
+        for document in documents:
+            try:
+                if 'type' in document.metadata and document.metadata['type'] == "TEST_CASE":
+                    additional_content = self._process_test_case(document.metadata['key'])
+                    for steps_type, content in additional_content.items():
+                        if content:
+                            document.page_content = json.dumps(content)
+                            document.metadata["steps_type"] = steps_type
+            except Exception as e:
+                logging.error(f"Failed to process document: {e}")
+            yield document
+
+    def _process_test_case(self, key):
+        steps = self.get_test_steps(key, return_list=True)
+        if steps and not isinstance(steps, ToolException):
+            return {"steps": steps}
+        script = self.get_test_script(key, return_only_script=True)
+        if script and not isinstance(script, ToolException):
+            return {"script": script}
+        return {"empty": ""}
 
     def get_tests_recursive(self, project_key: str = None, folder_id: str = None, maxResults: Optional[int] = 100, startAt: Optional[int] = 0):
         """Retrieves all test cases recursively from a folder and all its subfolders.
@@ -1552,7 +1556,7 @@ class ZephyrScaleApiWrapper(BaseVectorStoreToolApiWrapper):
         except Exception as e:
             return ToolException(f"Error updating test steps for test case {test_case_key}: {str(e)}")
 
-    @extend_with_vector_tools
+    @extend_with_parent_available_tools
     def get_available_tools(self):
         return [
             {

{alita_sdk-0.3.256.dist-info → alita_sdk-0.3.258.dist-info}/METADATA

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

@@ -10,7 +10,7 @@ alita_sdk/configurations/confluence.py,sha256=mAW2fgSEOg-BAV768Sc6b_EuRA3H5UL9xf
 alita_sdk/configurations/delta_lake.py,sha256=ADWcjabi7Krq2yxIpoc_tmhdncdgot2GBphE7ziDeTY,1133
 alita_sdk/configurations/embedding.py,sha256=8GSC8Feh8CH7bT_6cQhNqlS6raE91S2YRAtb2N9bUA8,552
 alita_sdk/configurations/figma.py,sha256=vecZ20IyZgnFO2GdphkovYHMISRPcUYh7fxkUQsPwX8,1306
-alita_sdk/configurations/github.py,sha256=GSj6sA4f6SfW0ZpoHXKi5FzbPDC6wE1AlscwWqIPj14,1832
+alita_sdk/configurations/github.py,sha256=NAiXotsl766IwjOOsVrs70PuoP4gvXla2Yycsv1qBpI,3569
 alita_sdk/configurations/gitlab.py,sha256=0W35igIlue6QxOnPgw65ToLf4HSdPVuRyObdwQuEld8,1053
 alita_sdk/configurations/jira.py,sha256=ASh8I2iVXzOOtwjRX7kYNllXpCXyAIxFMP_YD4Q0PTI,1379
 alita_sdk/configurations/pgvector.py,sha256=P-Q07ocIg4CXN_7hUBDM6r9gN62XS1N2jyP79tM9Tig,500
@@ -26,7 +26,7 @@ alita_sdk/configurations/zephyr_enterprise.py,sha256=5W1QEcv62Y5Rk_kApI2QmOwvWZe
 alita_sdk/runtime/__init__.py,sha256=4W0UF-nl3QF2bvET5lnah4o24CoTwSoKXhuN0YnwvEE,828
 alita_sdk/runtime/clients/__init__.py,sha256=BdehU5GBztN1Qi1Wul0cqlU46FxUfMnI6Vq2Zd_oq1M,296
 alita_sdk/runtime/clients/artifact.py,sha256=H3pJAh5G-zWVyJ6YbqHGk4jA8U6HfacQduiTivpJZ3Y,3210
-alita_sdk/runtime/clients/client.py,sha256=irj2uTGdIQj8Wd1ZGdi5yDCFm_n9TiRhEhODJz4yI84,43493
+alita_sdk/runtime/clients/client.py,sha256=ZkXP3-e785EyYoJLNowAGV3bRqv3fs1XqD_iuqqkg9Y,43513
 alita_sdk/runtime/clients/datasource.py,sha256=HAZovoQN9jBg0_-lIlGBQzb4FJdczPhkHehAiVG3Wx0,1020
 alita_sdk/runtime/clients/prompt.py,sha256=li1RG9eBwgNK_Qf0qUaZ8QNTmsncFrAL2pv3kbxZRZg,1447
 alita_sdk/runtime/langchain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -119,19 +119,19 @@ alita_sdk/runtime/utils/toolkit_runtime.py,sha256=MU63Fpxj0b5_r1IUUc0Q3-PN9VwL7r
 alita_sdk/runtime/utils/toolkit_utils.py,sha256=I9QFqnaqfVgN26LUr6s3XlBlG6y0CoHURnCzG7XcwVs,5311
 alita_sdk/runtime/utils/utils.py,sha256=CpEl3LCeLbhzQySz08lkKPm7Auac6IiLF7WB8wmArMI,589
 alita_sdk/tools/__init__.py,sha256=ko5TToGYZFmBrho26DRAVvrkHWxQ2sfs8gVAASinYp8,10611
-alita_sdk/tools/base_indexer_toolkit.py,sha256=gOjE1igKyjG1LohMj0XMlj1IGaFp7eEEDqyEG6-xLmc,18405
+alita_sdk/tools/base_indexer_toolkit.py,sha256=mfPo2iWhfHg-ihMC59nCf9XTomCPqPmKAudGkbpgBSE,18193
 alita_sdk/tools/elitea_base.py,sha256=PfelIUb5YFTjDN_1jNYT9tJbjfYr11PAUrPQHyW2d5I,32830
 alita_sdk/tools/non_code_indexer_toolkit.py,sha256=v9uq1POE1fQKCd152mbqDtF-HSe0qoDj83k4E5LAkMI,1080
-alita_sdk/tools/ado/__init__.py,sha256=bArTObt5cqG1SkijKevWGbsIILHBA3aCStg8Q1jd69k,1243
+alita_sdk/tools/ado/__init__.py,sha256=u2tdDgufGuDb-7lIgKKQlqgStL9Wd1gzNmRNYems2c0,1267
 alita_sdk/tools/ado/utils.py,sha256=PTCludvaQmPLakF2EbCGy66Mro4-rjDtavVP-xcB2Wc,1252
-alita_sdk/tools/ado/repos/__init__.py,sha256=zPLrWuAZamPrcUStOYHwWb-_Cvq6qm2JOwbn4Nnog2w,5374
-alita_sdk/tools/ado/repos/repos_wrapper.py,sha256=nPVsS10Se52yHmZ_YXVGywCSaYLlBEYBTBlhBcDJr80,50143
+alita_sdk/tools/ado/repos/__init__.py,sha256=n-IhKED05RwQGWT4LfCaxJ85uDyG4S9zTjSjK6A8N4o,5192
+alita_sdk/tools/ado/repos/repos_wrapper.py,sha256=e3bGsM03m0UggSQfoVh5Gg_M1MYt_BTKS-s9G2Unc1k,49739
 alita_sdk/tools/ado/test_plan/__init__.py,sha256=4fEw_3cm4shuZ868HhAU-uMH3xNXPyb3uRjyNWoBKls,5243
-alita_sdk/tools/ado/test_plan/test_plan_wrapper.py,sha256=jQt8kFmdAzsopjByLTMiSnWtoqz_IUOmYkhPTVGeMnU,20265
+alita_sdk/tools/ado/test_plan/test_plan_wrapper.py,sha256=57nYl-F4OPDwZu-amg9ptG0Z4iAYMBobcKlrbGgSAg8,20079
 alita_sdk/tools/ado/wiki/__init__.py,sha256=uBKo_Meu2ZxMxcxGsMmvCXyplRE2um1_PIRvdYd37rM,5171
-alita_sdk/tools/ado/wiki/ado_wrapper.py,sha256=zg6wMRar1DTp-ZRlYaQifBEnpYmTrHXskTNPdrLdy8s,14759
+alita_sdk/tools/ado/wiki/ado_wrapper.py,sha256=YfumbP0lI04hXWY01P9f-q9JBWuQRaeIJEvYDpcZnu0,14238
 alita_sdk/tools/ado/work_item/__init__.py,sha256=HNcdIMwTSNe-25_Pg-KmVVXTFci3vNa84tkTFkls36c,5373
-alita_sdk/tools/ado/work_item/ado_wrapper.py,sha256=gEywCL_kS0k1jWcDhsmYUybpIP08tH8go6CixLJGwT4,28409
+alita_sdk/tools/ado/work_item/ado_wrapper.py,sha256=uPhDp2zC8t42FQk7xc7gNyYs2o-hfOxsrw_rw31f7Sw,28223
 alita_sdk/tools/advanced_jira_mining/__init__.py,sha256=pUTzECqGvYaR5qWY3JPUhrImrZgc7pCXuqSe5eWIE80,4604
 alita_sdk/tools/advanced_jira_mining/data_mining_wrapper.py,sha256=nZPtuwVWp8VeHw1B8q9kdwf-6ZvHnlXTOGdcIMDkKpw,44211
 alita_sdk/tools/aws/__init__.py,sha256=tB6GCOg4XGSpR6qgbgAF4MUQ5-YmQCbWurWgrVKEKQ8,181
@@ -148,7 +148,7 @@ alita_sdk/tools/bitbucket/__init__.py,sha256=_ywYlgYoE6gtJlLR94MHcS4EPWIaCFU_Mxy
 alita_sdk/tools/bitbucket/api_wrapper.py,sha256=xKa2dQ-gw2YbLJx7P1xrc3JUfgBkXkMsEG-s0mzh3KI,11023
 alita_sdk/tools/bitbucket/bitbucket_constants.py,sha256=UsbhQ1iEvrKoxceTFPWTYhaXS1zSxbmjs1TwY0-P4gw,462
 alita_sdk/tools/bitbucket/cloud_api_wrapper.py,sha256=VELi65tLXvszwCGQSqVfyVal0ylx9DgAmAGpRQL_Zkg,15522
-alita_sdk/tools/bitbucket/tools.py,sha256=zKBUq7t9zLa1EvhlVZzyVcZSvwvdcbtz0oslgPFZeeo,15307
+alita_sdk/tools/bitbucket/tools.py,sha256=o1hwSEbSmjd5YgF-AsWls2ZyrewUuegBn9S6xTwsT74,15326
 alita_sdk/tools/browser/__init__.py,sha256=iByi9uMGjd6v44SagIPTm5fu1vWnxIkjn3xsx86uRwI,5249
 alita_sdk/tools/browser/crawler.py,sha256=jhE35dU94eQLURSM-D50tspOqEMsiGzMDbYNqNSR2mU,2279
 alita_sdk/tools/browser/duck_duck_go_search.py,sha256=iKws923v34o-ySXohJw-8xTDBWlj3fMsnzC_ZRuPugE,2002
@@ -171,7 +171,7 @@ alita_sdk/tools/carrier/tools.py,sha256=xBKXKNEdPQ_kWysoV7w6y4cDjtAMno8Qj2ubI4zr
 alita_sdk/tools/carrier/ui_reports_tool.py,sha256=Y6EstTRCa9d11ipFUFGOYlpiEhFx7aOQcgZ_M5Gd1lQ,13708
 alita_sdk/tools/carrier/update_ui_test_schedule_tool.py,sha256=jh9Q86cMCEqpsFopJPNIP0wlr7sYVa_3lhlq6lRmkGg,11850
 alita_sdk/tools/carrier/utils.py,sha256=rl7aq-F6ed_PapDM15w8EtS0BkgsjpDrNdKYuDCMOaI,4376
-alita_sdk/tools/chunkers/__init__.py,sha256=myaBVvPbUsz6PXtBDpA4EiPQgLvIv3q_WPh86kxlccI,774
+alita_sdk/tools/chunkers/__init__.py,sha256=5Nq--vAyECTh85HhDr0qEB-2UPXHXZWefP8hzX5zVO8,847
 alita_sdk/tools/chunkers/models.py,sha256=NNkLSljZboYDj6vbqeHmcjj9JrTHbkVWmoHGsL98q3k,3032
 alita_sdk/tools/chunkers/utils.py,sha256=gOyDHhXSH6Wlmxj_OsMOa2vydZuHD6HZql4PH-SYcTw,192
 alita_sdk/tools/chunkers/code/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -194,6 +194,7 @@ alita_sdk/tools/chunkers/code/treesitter/treesitter_rs.py,sha256=LgKyNffBy30gIr8
 alita_sdk/tools/chunkers/code/treesitter/treesitter_ts.py,sha256=Qs1a_BBN296iZc5hh8UNF9sc0G0-A_XZVhP3Na1ZNDg,387
 alita_sdk/tools/chunkers/sematic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/tools/chunkers/sematic/base.py,sha256=bRHpCFbOy-KPe4HBGpegrvIhvOsd7sDRfmb06T8tSuU,349
+alita_sdk/tools/chunkers/sematic/json_chunker.py,sha256=wzlGeoS4qZOQA6Alcc0sn6zAxZSGVdwCLm0wHqmMODA,1038
 alita_sdk/tools/chunkers/sematic/markdown_chunker.py,sha256=HmAGKuIodnMcHl-kBwAb1NY0GKKwAskRFvGaW3m4HAM,3859
 alita_sdk/tools/chunkers/sematic/proposal_chunker.py,sha256=t8JjX9TH6yHXXaemiDK1E6000tlES2Kl8XfyezmlIoo,5116
 alita_sdk/tools/chunkers/sematic/statistical_chunker.py,sha256=VDQcMC-ky72GqdWJiHMmcRmfJTTU5XglBF1IWg2Qews,13403
@@ -246,7 +247,7 @@ alita_sdk/tools/google/bigquery/tool.py,sha256=Esf9Hsp8I0e7-5EdkFqQ-bid0cfrg-bfS
 alita_sdk/tools/google_places/__init__.py,sha256=mHKc7u9P2gqGDzqqJNQC9qiZYEm5gncnM_1XjtrM17o,3152
 alita_sdk/tools/google_places/api_wrapper.py,sha256=7nZly6nk4f4Tm7s2MVdnnwlb-1_WHRrDhyjDiqoyPjA,4674
 alita_sdk/tools/jira/__init__.py,sha256=k9Alxe1tEHYYzkCZv9hd89JMzBgv7cZiVT6k_tsO7hg,6073
-alita_sdk/tools/jira/api_wrapper.py,sha256=i8x8CttVEW_FFEl6hBNdzCqc-aMyy1FhqkiEHGaDmpo,76178
+alita_sdk/tools/jira/api_wrapper.py,sha256=MRDBL_VES6Qs1mHXetwqfk-2BibsNsXQIiaz0bAocFY,76353
 alita_sdk/tools/keycloak/__init__.py,sha256=0WB9yXMUUAHQRni1ghDEmd7GYa7aJPsTVlZgMCM9cQ0,3050
 alita_sdk/tools/keycloak/api_wrapper.py,sha256=cOGr0f3S3-c6tRDBWI8wMnetjoNSxiV5rvC_0VHb8uw,3100
 alita_sdk/tools/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -284,7 +285,7 @@ alita_sdk/tools/postman/api_wrapper.py,sha256=bKgnEQVGv3QhqTevzBOwiXxAd0-Y5vUI1-
 alita_sdk/tools/postman/postman_analysis.py,sha256=2d-Oi2UORosIePIUyncSONw9hY7dw8Zc7BQvCd4aqpg,45115
 alita_sdk/tools/pptx/__init__.py,sha256=vVUrWnj7KWJgEk9oxGSsCAQ2SMSXrp_SFOdUHYQKcAo,3444
 alita_sdk/tools/pptx/pptx_wrapper.py,sha256=yyCYcTlIY976kJ4VfPo4dyxj4yeii9j9TWP6W8ZIpN8,29195
-alita_sdk/tools/qtest/__init__.py,sha256=4vXCB9GSKNFeRTimSB7wklAnO-4reZgrw0Nw1_QuRKE,4070
+alita_sdk/tools/qtest/__init__.py,sha256=3NUBDnwIZFFmdNNzCo4u7hBfhgU3AT5NkyMBtdxS9yw,4084
 alita_sdk/tools/qtest/api_wrapper.py,sha256=cWXpmjjel9CYIXXjetJkARLYZXqvHufSghctTHN0ggc,22296
 alita_sdk/tools/qtest/tool.py,sha256=kKzNPS4fUC76WQQttQ6kdVANViHEvKE8Kf174MQiNYU,562
 alita_sdk/tools/rally/__init__.py,sha256=JvLt_hW_hC1WiCcwBwi1TlOH7QudJpM2z7XXGWYVaqI,3423
@@ -312,10 +313,11 @@ alita_sdk/tools/testio/api_wrapper.py,sha256=BvmL5h634BzG6p7ajnQLmj-uoAw1gjWnd4F
 alita_sdk/tools/testrail/__init__.py,sha256=0kETjWKLU7R6mugBWsjwEUsh10pipbAeNSGJAO0FBh0,4634
 alita_sdk/tools/testrail/api_wrapper.py,sha256=5T-QyTzt-J0rI32xc_E684lCdgyWeHSyeTYiwQwtGyg,32275
 alita_sdk/tools/utils/__init__.py,sha256=155xepXPr4OEzs2Mz5YnjXcBpxSv1X2eznRUVoPtyK0,3268
+alita_sdk/tools/utils/available_tools_decorator.py,sha256=IbrdfeQkswxUFgvvN7-dyLMZMyXLiwvX7kgi3phciCk,273
 alita_sdk/tools/utils/content_parser.py,sha256=zqeyuxZqZqVFq5M5sZM-falMdlOw48FyZnp3Z0XUpCw,9868
 alita_sdk/tools/vector_adapters/VectorStoreAdapter.py,sha256=a6FAsiix_EvATIKUf5YT6vHh5LDyJ5uSP3LJqoxFo04,17367
 alita_sdk/tools/vector_adapters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-alita_sdk/tools/xray/__init__.py,sha256=GGpbiBdDQ9kMFqJEHYi7XwKpkuMMHi-ZF-IM8yFIgUM,4380
+alita_sdk/tools/xray/__init__.py,sha256=BnG2StSfX44CUMtrjHTcSCDWxxse5tCZqwyaZSkBKIc,4230
 alita_sdk/tools/xray/api_wrapper.py,sha256=A8PJmY2k7TowaD_vk6ZxkMnSUoZUt9A6g4TJrZfNTAw,32225
 alita_sdk/tools/yagmail/__init__.py,sha256=c4Qn3em0tLxzRmFKpzbBgY9W2EnOoKf0azoDJHng5CY,2208
 alita_sdk/tools/yagmail/yagmail_wrapper.py,sha256=SKoGVd1X4Ew3ad5tOdtPoY00M6jStNdT3q7GXEjQc5g,1952
@@ -324,18 +326,18 @@ alita_sdk/tools/zephyr/__init__.py,sha256=8B2Ibz5QTmB5WkV0q8Sq4kuj92FFaFWZLrT877
 alita_sdk/tools/zephyr/api_wrapper.py,sha256=lJCYPG03ej0qgdpLflnS7LFB4HSAfGzIvTjAJt07CQs,6244
 alita_sdk/tools/zephyr/rest_client.py,sha256=7vSD3oYIX-3KbAFed-mphSQif_VRuXrq5O07ryNQ7Pk,6208
 alita_sdk/tools/zephyr_enterprise/__init__.py,sha256=1E0xuyYx7QSuqIRKclEapI7MvxXjJ3Lwf4YpDXPzehw,4087
-alita_sdk/tools/zephyr_enterprise/api_wrapper.py,sha256=p9EpkO5tif3JJzprz2_VuLsQ1yET7TwwBfPOKJGwt9c,11215
+alita_sdk/tools/zephyr_enterprise/api_wrapper.py,sha256=km2TYNu5ppRkspN1PyYetu6iBGj-xKVIwGHty1r_wAw,11552
 alita_sdk/tools/zephyr_enterprise/zephyr_enterprise.py,sha256=hV9LIrYfJT6oYp-ZfQR0YHflqBFPsUw2Oc55HwK0H48,6809
 alita_sdk/tools/zephyr_essential/__init__.py,sha256=BpRicA38JI9YEDuim1acZRGcDw-ZYTdP4Ewbiju37h8,3761
-alita_sdk/tools/zephyr_essential/api_wrapper.py,sha256=TpNov35XPgjM9eymCEFqv22mbpdVvLMBTb9WVqUcvNA,36795
+alita_sdk/tools/zephyr_essential/api_wrapper.py,sha256=WZvtC_CLVZxwsLWITOGZpFqPWp3JgzCVtYmmq__vU5c,37237
 alita_sdk/tools/zephyr_essential/client.py,sha256=bfNcUKNqj9MFWTludGbbqD4qZlxrBaC2JtWsCfZMqSY,9722
 alita_sdk/tools/zephyr_scale/__init__.py,sha256=imuHOqdyOqtxQObeBZfyFvKPXfKVNuYfwKn2c9jJyeo,4299
-alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=JAeWf-RXohsxheUpT0iMDClc_izj-zxMwafXCW4jtC0,78015
+alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=HOt9ShtJI_1tVPcwd3Rwk-VS0SMLqcPNYbN1wqfeuhc,78330
 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.256.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.256.dist-info/METADATA,sha256=Pz8NwBvPngJixME90E16UprodmhH7mPX8aR1oTzAHno,18897
-alita_sdk-0.3.256.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.256.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.256.dist-info/RECORD,,
+alita_sdk-0.3.258.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.258.dist-info/METADATA,sha256=2j5Uqki3lIbS41bXwlA4RiIVvtsBROzSHIoisJhN1gc,18897
+alita_sdk-0.3.258.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.258.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.258.dist-info/RECORD,,