alita-sdk 0.3.317__tar.gz → 0.3.319__tar.gz

{alita_sdk-0.3.317 → alita_sdk-0.3.319}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.317
+Version: 0.3.319
 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 <lifedj27@gmail.com>, Artem Dubrovskiy <ad13box@gmail.com>
 License-Expression: Apache-2.0

{alita_sdk-0.3.317 → alita_sdk-0.3.319}/alita_sdk/runtime/langchain/document_loaders/AlitaExcelLoader.py

@@ -17,6 +17,7 @@ import pandas as pd
 from json import loads
 
 from langchain_core.tools import ToolException
+from langchain_core.documents import Document
 from .AlitaTableLoader import AlitaTableLoader
     
 
@@ -65,6 +66,20 @@ class AlitaExcelLoader(AlitaTableLoader):
         else:
             return df.to_string(index=False)
 
+    def load(self) -> list:
+        docs = []
+        content_per_sheet = self.get_content()
+        for sheet_name, content in content_per_sheet.items():
+            metadata = {
+                "source": f'{self.file_path}:{sheet_name}',
+                "sheet_name": sheet_name,
+                "file_type": "excel",
+                "excel_by_sheets": self.excel_by_sheets,
+                "return_type": self.return_type,
+            }
+            docs.append(Document(page_content=f"Sheet: {sheet_name}\n {str(content)}", metadata=metadata))
+        return docs
+
     def read(self, lazy: bool = False):
         df = pd.read_excel(self.file_path, sheet_name=None)
         docs = []

{alita_sdk-0.3.317 → alita_sdk-0.3.319}/alita_sdk/runtime/langchain/document_loaders/AlitaTableLoader.py

@@ -17,8 +17,6 @@ from langchain_core.documents import Document
 from typing import List, Optional, Iterator
 from json import dumps
 from .utils import cleanse_data
-from ..tools.log import print_log
-
 
 class AlitaTableLoader(BaseLoader):
     def __init__(self,
@@ -74,7 +72,6 @@ class AlitaTableLoader(BaseLoader):
                 docs.append(Document(page_content=row, metadata=metadata))
                 continue
             if self.json_documents:
-                # print_log(row)
                 metadata['columns'] = list(row.keys())
                 metadata['og_data'] = dumps(row)
                 docs.append(Document(page_content=self.row_processor(row), metadata=metadata))

{alita_sdk-0.3.317 → alita_sdk-0.3.319}/alita_sdk/runtime/langchain/document_loaders/constants.py

@@ -122,6 +122,7 @@ loaders_map = {
         'class': AlitaExcelLoader,
         'is_multimodal_processing': False,
         'kwargs': {
+            'excel_by_sheets': True,
             'raw_content': True,
             'cleanse': False
         },
@@ -131,6 +132,7 @@ loaders_map = {
         'class': AlitaExcelLoader,
         'is_multimodal_processing': False,
         'kwargs': {
+            'excel_by_sheets': True,
             'raw_content': True,
             'cleanse': False
         },

{alita_sdk-0.3.317 → alita_sdk-0.3.319}/alita_sdk/runtime/tools/vectorstore_base.py

@@ -10,7 +10,6 @@ from pydantic import BaseModel, model_validator, Field
 
 from alita_sdk.tools.elitea_base import BaseToolApiWrapper
 from alita_sdk.tools.vector_adapters.VectorStoreAdapter import VectorStoreAdapterFactory
-from ..langchain.tools.vector import VectorAdapter
 from ..utils.logging import dispatch_custom_event
 
 logger = getLogger(__name__)
@@ -138,10 +137,7 @@ class VectorStoreWrapperBase(BaseToolApiWrapper):
     vectorstore_params: Optional[dict]  = None
     max_docs_per_add: int = 100
     dataset: str = None
-    embedding: Any = None
     vectorstore: Any = None
-    # Review usage of old adapter
-    vectoradapter: Any = None
     pg_helper: Any = None
     embeddings: Any = None
     # New adapter for vector database operations
@@ -152,17 +148,13 @@ class VectorStoreWrapperBase(BaseToolApiWrapper):
     def validate_toolkit(cls, values):
         from ..langchain.interfaces.llm_processor import get_vectorstore
         logger.debug(f"Validating toolkit: {values}")
-        if 'vectorstore_params' in values:
-            values["dataset"] = values.get('vectorstore_params').get('collection_name')
-        if values.get('embedding_model'):
-            values['embeddings'] = values['alita'].get_embeddings(values['embedding_model'])
+        values["dataset"] = values.get('collection_name')
+
+        if values.get('alita') and values.get('embedding_model'):
+            values['embeddings'] = values.get('alita').get_embeddings(values.get('embedding_model'))
+
         if values.get('vectorstore_type') and values.get('vectorstore_params') and values.get('embedding_model'):
             values['vectorstore'] = get_vectorstore(values['vectorstore_type'], values['vectorstore_params'], embedding_func=values['embeddings'])
-            values['vectoradapter'] = VectorAdapter(
-                vectorstore=values['vectorstore'],
-                embeddings=values['embeddings'],
-                quota_params=None,
-            )
             # Initialize the new vector adapter
             values['vector_adapter'] = VectorStoreAdapterFactory.create_adapter(values['vectorstore_type'])
             logger.debug(f"Vectorstore wrapper initialized: {values}")
@@ -223,8 +215,6 @@ class VectorStoreWrapperBase(BaseToolApiWrapper):
         self._log_data("Cleaning index before re-indexing all documents. Previous index will be removed", tool_name="index_documents")
         try:
             self._clean_collection(collection_suffix)
-            self.vectoradapter.persist()
-            self.vectoradapter.vacuum()
             self._log_data("Previous index has been removed",
                            tool_name="index_documents")
         except Exception as e:
@@ -238,7 +228,6 @@ class VectorStoreWrapperBase(BaseToolApiWrapper):
                 logger.warning(f"Document is missing required metadata field 'id' or 'updated_on': {doc.metadata}")
 
         logger.debug(f"Indexing documents: {documents}")
-        logger.debug(self.vectoradapter)
 
         # if collection_suffix is provided, add it to metadata of each document
         if collection_suffix:

{alita_sdk-0.3.317 → alita_sdk-0.3.319}/alita_sdk/tools/ado/test_plan/test_plan_wrapper.py

@@ -411,11 +411,11 @@ class TestPlanApiWrapper(NonCodeIndexerToolkit):
     def _index_tool_params(self):
         """Return the parameters for indexing data."""
         return {
+            'chunking_tool': (Literal['html', ''], Field(description="Name of chunking tool", default='html')),
             "plan_id": (int, Field(description="ID of the test plan for which test cases are requested")),
             "suite_ids": (Optional[List[int]], Field(description='List of test suite IDs for which test cases are requested '
                                                                  '(can be empty for all suites indexing from the plan). '
                                                                  'Example: [2, 23]', default=[])),
-            'chunking_tool':(Literal['html', ''], Field(description="Name of chunking tool", default='html'))
         }
 
     @extend_with_parent_available_tools

{alita_sdk-0.3.317 → alita_sdk-0.3.319}/alita_sdk/tools/ado/wiki/ado_wrapper.py

@@ -248,9 +248,9 @@ class AzureDevOpsApiWrapper(NonCodeIndexerToolkit):
     def _index_tool_params(self):
         """Return the parameters for indexing data."""
         return {
+            'chunking_tool': (Literal['markdown', ''], Field(description="Name of chunking tool", default='markdown')),
             "wiki_identifier": (str, Field(description="Wiki identifier to index, e.g., 'ABCProject.wiki'")),
             'title_contains': (Optional[str], Field(default=None, description="Optional filter to include only pages with titles containing exact this string")),
-            'chunking_tool':(Literal['markdown', ''], Field(description="Name of chunking tool", default='markdown'))
         }
 
     @extend_with_parent_available_tools

{alita_sdk-0.3.317 → alita_sdk-0.3.319}/alita_sdk/tools/base_indexer_toolkit.py

@@ -5,7 +5,7 @@ from typing import Any, Optional, List, Literal, Dict, Generator
 from langchain_core.documents import Document
 from pydantic import create_model, Field, SecretStr
 
-from .utils.content_parser import file_extension_by_chunker, process_content_by_type
+from .utils.content_parser import file_extension_by_chunker, process_document_by_type
 from .vector_adapters.VectorStoreAdapter import VectorStoreAdapterFactory
 from ..runtime.tools.vectorstore_base import VectorStoreWrapperBase
 from ..runtime.utils.utils import IndexerKeywords
@@ -87,10 +87,10 @@ BaseStepbackSearchParams = create_model(
 BaseIndexDataParams = create_model(
     "indexData",
     __base__=BaseIndexParams,
-    progress_step=(Optional[int], Field(default=10, ge=0, le=100,
-                         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")),
+    progress_step=(Optional[int], Field(default=10, ge=0, le=100,
+                         description="Optional step size for progress reporting during indexing")),
     chunking_config=(Optional[dict], Field(description="Chunking tool configuration", default_factory=dict)),
 )
 
@@ -102,7 +102,6 @@ class BaseIndexerToolkit(VectorStoreWrapperBase):
 
     connection_string: Optional[SecretStr] = None
     collection_name: Optional[str] = None
-    _embedding: Optional[Any] = None
     alita: Any = None # Elitea client, if available
 
     def __init__(self, **kwargs):
@@ -116,7 +115,6 @@ class BaseIndexerToolkit(VectorStoreWrapperBase):
         if connection_string:
             # Initialize vectorstore params only if connection string is provided
             kwargs['vectorstore_params'] = VectorStoreAdapterFactory.create_adapter(vectorstore_type).get_vectorstore_params(collection_name, connection_string)
-            kwargs['_embedding'] = kwargs.get('alita').get_embeddings(kwargs.get('embedding_model'))
         super().__init__(**kwargs)
 
     def _index_tool_params(self, **kwargs) -> dict[str, tuple[type, Field]]:
@@ -181,14 +179,14 @@ class BaseIndexerToolkit(VectorStoreWrapperBase):
 
         if chunking_config is None:
             chunking_config = {}
-        chunking_config['embedding'] = self._embedding
+        chunking_config['embedding'] = self.embeddings
         chunking_config['llm'] = self.llm
 
         for document in documents:
             if content_type := document.metadata.get(IndexerKeywords.CONTENT_FILE_NAME.value, None):
                 # apply parsing based on content type and chunk if chunker was applied to parent doc
                 content = document.metadata.pop(IndexerKeywords.CONTENT_IN_BYTES.value, None)
-                yield from process_content_by_type(
+                yield from process_document_by_type(
                     document=document,
                     content=content,
                     extension_source=content_type, llm=self.llm, chunking_config=chunking_config)
@@ -199,7 +197,7 @@ class BaseIndexerToolkit(VectorStoreWrapperBase):
                     continue
                 # apply parsing based on content type resolved from chunking_tool
                 content_type = file_extension_by_chunker(chunking_tool)
-                yield from process_content_by_type(
+                yield from process_document_by_type(
                     document=document,
                     content=content_in_bytes,
                     extension_source=content_type, llm=self.llm, chunking_config=chunking_config)

{alita_sdk-0.3.317 → alita_sdk-0.3.319}/alita_sdk/tools/carrier/backend_reports_tool.py

@@ -154,6 +154,7 @@ class CreateExcelReportTool(BaseTool):
             "tp_threshold": (int, Field(default=None, description="Throughput threshold")),
             "rt_threshold": (int, Field(default=None, description="Response time threshold")),
             "er_threshold": (int, Field(default=None, description="Error rate threshold")),
+            "include_group_pauses": (bool, Field(default=False, description="Include group pauses in Gatling Excel report")),
         }
     )
 
@@ -200,6 +201,7 @@ class CreateExcelReportTool(BaseTool):
             "tp_threshold": 10,
             "rt_threshold": 500,
             "er_threshold": 5,
+            "include_group_pauses": False,
         }
 
     def _request_parameter_confirmation(self, default_parameters):
@@ -217,7 +219,8 @@ class CreateExcelReportTool(BaseTool):
         excel_report_file_name = f'/tmp/reports_test_results_{report["build_id"]}_excel_report.xlsx'
         bucket_name = report["name"].replace("_", "").replace(" ", "").lower()
 
-        result_stats_j = self._parse_report(test_log_file_path, lg_type, parameters["think_time"], is_absolute_file_path=True)
+        result_stats_j = self._parse_report(test_log_file_path, lg_type, parameters["think_time"],
+                                            parameters["include_group_pauses"], is_absolute_file_path=True)
         calc_thr_j = self._calculate_thresholds(result_stats_j, parameters)
 
         return self._generate_and_upload_report(
@@ -233,21 +236,22 @@ class CreateExcelReportTool(BaseTool):
         excel_report_file_name = f'{file_path}_{current_date}.xlsx'
         bucket_name = bucket
 
-        result_stats_j = self._parse_report(file_path, lg_type, parameters["think_time"], is_absolute_file_path=True)
+        result_stats_j = self._parse_report(file_path, lg_type, parameters["think_time"],
+                                            parameters["include_group_pauses"], is_absolute_file_path=True)
         calc_thr_j = self._calculate_thresholds(result_stats_j, parameters)
 
         return self._generate_and_upload_report(
             result_stats_j, carrier_report, calc_thr_j, parameters, excel_report_file_name, bucket_name, file_path
         )
 
-    def _parse_report(self, file_path, lg_type, think_time, is_absolute_file_path=False):
+    def _parse_report(self, file_path, lg_type, think_time, include_group_pauses, is_absolute_file_path=False):
         """Parse the report based on its type."""
         if lg_type == "gatling":
             if is_absolute_file_path:
                 report_file = file_path
             else:
                 report_file = get_latest_log_file(file_path, "simulation.log")
-            parser = GatlingReportParser(report_file, think_time)
+            parser = GatlingReportParser(report_file, include_group_pauses, think_time)
             result_stats_j = parser.parse()
             result_stats_j["requests"].update(result_stats_j["groups"])
         elif lg_type == "jmeter":

{alita_sdk-0.3.317 → alita_sdk-0.3.319}/alita_sdk/tools/carrier/excel_reporter.py

@@ -118,9 +118,10 @@ class JMeterReportParser(PerformanceReportParser):
 
 class GatlingReportParser(PerformanceReportParser):
 
-    def __init__(self, log_file: str, think_times="5,0-10,0"):
+    def __init__(self, log_file: str, include_group_pauses, think_times="5,0-10,0"):
         self.calculated_think_time = think_times
         self.log_file = log_file
+        self.include_group_pauses = include_group_pauses
 
     @staticmethod
     def convert_timestamp_to_datetime(timestamp: int) -> datetime:
@@ -210,7 +211,7 @@ class GatlingReportParser(PerformanceReportParser):
                                 ramp_end = self.convert_timestamp_to_datetime(int(line.split('\t')[3]))
 
                     elif line.startswith('GROUP'):
-                        self.parse_group_line(groups, line)
+                        self.parse_group_line(groups, line, self.include_group_pauses)
         except FileNotFoundError as e:
             print(f"File not found: {e}")
             raise
@@ -242,11 +243,14 @@ class GatlingReportParser(PerformanceReportParser):
             requests[request_name].append((response_time, status))
 
     @staticmethod
-    def parse_group_line(groups, line):
+    def parse_group_line(groups, line, include_group_pauses):
         parts = line.split('\t')
         if len(parts) >= 6:
             group_name = parts[1]
-            response_time = int(parts[4])
+            if include_group_pauses:
+                response_time = int(parts[3]) - int(parts[2])
+            else:
+                response_time = int(parts[4])
             status = parts[5].strip()
             groups[group_name].append((response_time, status))
 

{alita_sdk-0.3.317 → alita_sdk-0.3.319}/alita_sdk/tools/confluence/api_wrapper.py

@@ -1656,8 +1656,7 @@ class ConfluenceAPIWrapper(NonCodeIndexerToolkit):
 
     @extend_with_parent_available_tools
     def get_available_tools(self):
-        # Confluence-specific tools
-        confluence_tools = [
+        return [
             {
                 "name": "create_page",
                 "ref": self.create_page,
@@ -1772,9 +1771,4 @@ class ConfluenceAPIWrapper(NonCodeIndexerToolkit):
                 "description": self.get_page_attachments.__doc__,
                 "args_schema": GetPageAttachmentsInput,
             }
-        ]
-
-        # Add standardized vector search tools from base class
-        vector_search_tools = self._get_vector_search_tools()
-
-        return confluence_tools + vector_search_tools
+        ]

{alita_sdk-0.3.317 → alita_sdk-0.3.319}/alita_sdk/tools/elitea_base.py

@@ -39,10 +39,13 @@ BaseIndexParams = create_model(
 BaseCodeIndexParams = create_model(
     "BaseCodeIndexParams",
     collection_suffix=(str, Field(description="Suffix for collection name (max 7 characters) used to separate datasets", min_length=1, max_length=7)),
+    clean_index=(Optional[bool], Field(default=False, description="Optional flag to enforce clean existing index before indexing new data")),
+    progress_step=(Optional[int], Field(default=5, ge=0, le=100,
+                         description="Optional step size for progress reporting during indexing")),
     branch=(Optional[str], Field(description="Branch to index files from. Defaults to active branch if None.", default=None)),
     whitelist=(Optional[List[str]], Field(description='File extensions or paths to include. Defaults to all files if None. Example: ["*.md", "*.java"]', default=None)),
     blacklist=(Optional[List[str]], Field(description='File extensions or paths to exclude. Defaults to no exclusions if None. Example: ["*.md", "*.java"]', default=None)),
-    clean_index=(Optional[bool], Field(default=False, description="Optional flag to enforce clean existing index before indexing new data")),
+
 )
 
 RemoveIndexParams = create_model(
@@ -115,10 +118,10 @@ BaseStepbackSearchParams = create_model(
 BaseIndexDataParams = create_model(
     "indexData",
     __base__=BaseIndexParams,
-    progress_step=(Optional[int], Field(default=5, ge=0, le=100,
-                         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")),
+    progress_step=(Optional[int], Field(default=5, ge=0, le=100,
+                         description="Optional step size for progress reporting during indexing")),
     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)),
 )

{alita_sdk-0.3.317 → alita_sdk-0.3.319}/alita_sdk/tools/jira/api_wrapper.py

@@ -17,7 +17,7 @@ from ..llm.img_utils import ImageDescriptionCache
 from ..non_code_indexer_toolkit import NonCodeIndexerToolkit
 from ..utils import is_cookie_token, parse_cookie_string
 from ..utils.available_tools_decorator import extend_with_parent_available_tools
-from ..utils.content_parser import file_extension_by_chunker
+from ..utils.content_parser import file_extension_by_chunker, process_content_by_type
 from ...runtime.utils.utils import IndexerKeywords
 
 logger = logging.getLogger(__name__)
@@ -132,6 +132,13 @@ GetRemoteLinks = create_model(
     jira_issue_key=(str, Field(description="Jira issue key from which remote links will be extracted, e.g. TEST-1234"))
 )
 
+GetIssueAttachments = create_model(
+    "GetIssueAttachments",
+    jira_issue_key=(str, Field(description="Jira issue key from which remote links will be extracted, e.g. TEST-1234")),
+    attachment_pattern=(Optional[str], Field(description="Regex pattern to filter attachment filenames. If not provided,"
+                                                         " all attachments will be processed", default=None))
+)
+
 ListCommentsInput = create_model(
     "ListCommentsInputModel",
     issue_key=(str, Field(description="The issue key of the Jira issue from which comments will be extracted, e.g. 'TEST-123'."))
@@ -725,19 +732,37 @@ class JiraApiWrapper(NonCodeIndexerToolkit):
             logger.error(f"Error listing Jira projects: {stacktrace}")
             return ToolException(f"Error listing Jira projects: {stacktrace}")
 
-    def get_attachments_content(self, jira_issue_key: str):
-        """ Extract content of all attachments related to specified Jira issue key.
-         NOTE: only parsable attachments will be considered """
+    def get_attachments_content(self, jira_issue_key: str, attachment_pattern: Optional[str] = None):
+        """ Extract the content of all attachments related to a specified Jira issue key.
+         NOTE: only parsable attachments will be considered
+         Args:
+            jira_issue_key: The key of the Jira issue, e.g. "TEST-123
+            attachment_pattern: Optional regex pattern to filter attachments by filename.
+            If provided, only attachments with filenames matching this pattern will be processed.
+         Returns:
+            A string containing the content of all relevant attachments, separated by double newlines.
+         """
 
         attachment_data = []
         attachments = self._client.get_attachments_ids_from_issue(issue=jira_issue_key)
         for attachment in attachments:
+            if attachment_pattern and not re.search(attachment_pattern, attachment['filename']):
+                logger.info(f"Skipping attachment {attachment['filename']} as it does not match pattern {attachment_pattern}")
+                continue
+            logger.info(f"Processing attachment {attachment['filename']} with ID {attachment['attachment_id']}")
             if self.api_version == "3":
                 attachment_data.append(self._client.get_attachment_content(attachment['attachment_id']))
             else:
-                extracted_attachment = self._client.get_attachment(attachment_id=attachment['attachment_id'])
-                if extracted_attachment['mimeType'] in SUPPORTED_ATTACHMENT_MIME_TYPES:
-                    attachment_data.append(self._extract_attachment_content(extracted_attachment))
+                try:
+                    attachment_content = self._client.get_attachment_content(attachment['attachment_id'])
+                except Exception as e:
+                    logger.error(
+                        f"Failed to download attachment {attachment['filename']} for issue {jira_issue_key}: {str(e)}")
+                    attachment_content = self._client.get(
+                        path=f"secure/attachment/{attachment['attachment_id']}/{attachment['filename']}", not_json_response=True)
+                content_docs = process_content_by_type(attachment_content, attachment['filename'], llm=self.llm)
+                attachment_data.append("filename: " + attachment['filename'] + "\ncontent: " + str([doc.page_content for doc in content_docs]))
+
         return "\n\n".join(attachment_data)
 
     def execute_generic_rq(self, method: str, relative_url: str, params: Optional[str] = "", *args):
@@ -1632,7 +1657,7 @@ class JiraApiWrapper(NonCodeIndexerToolkit):
             {
                 "name": "get_attachments_content",
                 "description": self.get_attachments_content.__doc__,
-                "args_schema": GetRemoteLinks,
+                "args_schema": GetIssueAttachments,
                 "ref": self.get_attachments_content,
             },
             {

{alita_sdk-0.3.317 → alita_sdk-0.3.319}/alita_sdk/tools/utils/content_parser.py

@@ -201,17 +201,40 @@ def load_content_from_bytes(file_content: bytes, extension: str = None, loader_e
         if temp_file_path and os.path.exists(temp_file_path):
             os.remove(temp_file_path)
 
-def process_content_by_type(document: Document, content, extension_source: str, llm = None, chunking_config=None) -> Generator[Document, None, None]:
+def process_document_by_type(content, extension_source: str, document: Document = None, llm = None, chunking_config=None) \
+        -> Generator[Document, None, None]:
+    """Process the content of a file based on its type using a configured loader cosidering the origin document."""
+    try:
+        chunks = process_content_by_type(content, extension_source, llm, chunking_config)
+    except Exception as e:
+        msg = f"Error during content for file {extension_source}:\n{e}"
+        logger.warning(msg)
+        yield Document(
+            page_content=msg,
+            metadata={**document.metadata, 'chunk_id': 1}
+        )
+        return
+    for chunk in chunks:
+        yield Document(
+            page_content=sanitize_for_postgres(chunk.page_content),
+            metadata={**document.metadata, **chunk.metadata}
+                )
+
+
+def process_content_by_type(content, filename: str, llm=None, chunking_config=None) -> \
+        Generator[Document, None, None]:
+    """Process the content of a file based on its type using a configured loader."""
     temp_file_path = None
     try:
-        extension = "." + extension_source.split('.')[-1].lower()
+        extension = "." + filename.split('.')[-1].lower()
 
         with tempfile.NamedTemporaryFile(mode='w+b', suffix=extension, delete=False) as temp_file:
             temp_file_path = temp_file.name
             if content is None:
-                logger.warning(f"'{IndexerKeywords.CONTENT_IN_BYTES.value}' ie expected but not found in document metadata.")
+                logger.warning(
+                    f"'{IndexerKeywords.CONTENT_IN_BYTES.value}' ie expected but not found in document metadata.")
                 return
-            
+
             temp_file.write(content)
             temp_file.flush()
 
@@ -238,21 +261,7 @@ def process_content_by_type(document: Document, content, extension_source: str,
                 loader_kwargs.pop(LoaderProperties.PROMPT_DEFAULT.value)
                 loader_kwargs[LoaderProperties.PROMPT.value] = image_processing_prompt
             loader = loader_cls(file_path=temp_file_path, **loader_kwargs)
-            try:
-                chunks = loader.load()
-            except Exception as e:
-                msg = f"Error during content for file {temp_file_path}:\n{e}"
-                logger.warning(msg)
-                yield Document(
-                    page_content=msg,
-                    metadata={**document.metadata, 'chunk_id':1}
-                )
-                return
-            for chunk in chunks:
-                yield Document(
-                    page_content=sanitize_for_postgres(chunk.page_content),
-                    metadata={**document.metadata, **chunk.metadata}
-                )
+            return loader.load()
     finally:
         if temp_file_path and os.path.exists(temp_file_path):
             os.remove(temp_file_path)

{alita_sdk-0.3.317 → alita_sdk-0.3.319}/alita_sdk/tools/xray/api_wrapper.py

@@ -561,6 +561,8 @@ class XrayApiWrapper(NonCodeIndexerToolkit):
 
     def _index_tool_params(self, **kwargs) -> dict[str, tuple[type, Field]]:
         return {
+            'chunking_tool': (Literal['json', ''],
+                              Field(description="Name of chunking tool for base document", default='json')),
             'jql': (Optional[str], Field(description="""JQL query for searching test cases in Xray.
 
             Standard JQL query syntax for filtering Xray test cases. Examples:
@@ -595,8 +597,6 @@ class XrayApiWrapper(NonCodeIndexerToolkit):
             'skip_attachment_extensions': (Optional[List[str]], Field(
                 description="List of file extensions to skip when processing attachments (e.g., ['.exe', '.zip', '.bin'])",
                 default=None)),
-            'chunking_tool': (Literal['json', ''],
-                              Field(description="Name of chunking tool for base document", default='json')),
         }
 
     def _get_tests_direct(self, jql: str) -> List[Dict]:

{alita_sdk-0.3.317 → alita_sdk-0.3.319}/alita_sdk/tools/zephyr_enterprise/api_wrapper.py

@@ -155,8 +155,9 @@ class ZephyrApiWrapper(NonCodeIndexerToolkit):
         Returns a list of fields for index_data args schema.
         """
         return {
-            "zql": (str, Field(description=zql_description, examples=["folder=\"TestToolkit\"", "name~\"TestToolkit5\""])),
-            'chunking_tool': (Literal['json', ''], Field(description="Name of chunking tool", default='json'))
+            'chunking_tool': (Literal['json', ''], Field(description="Name of chunking tool", default='json')),
+            "zql": (str, Field(description=zql_description, examples=["folder=\"TestToolkit\"", "name~\"TestToolkit5\""]))
+
         }
 
     def _base_loader(self, zql: str, **kwargs) -> Generator[Document, None, None]:

{alita_sdk-0.3.317 → alita_sdk-0.3.319}/alita_sdk/tools/zephyr_scale/api_wrapper.py

@@ -1198,6 +1198,7 @@ class ZephyrScaleApiWrapper(NonCodeIndexerToolkit):
         Returns a list of fields for index_data args schema.
         """
         return {
+            "chunking_tool": (Literal['json', ''], Field(description="Name of chunking tool", default='json')),
             "project_key": (str, Field(description="Jira project key filter")),
             "jql":(str, Field(description="""JQL-like query for searching test cases.
             Supported fields:
@@ -1216,7 +1217,6 @@ class ZephyrScaleApiWrapper(NonCodeIndexerToolkit):
             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]:

{alita_sdk-0.3.317 → alita_sdk-0.3.319}/alita_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.317
+Version: 0.3.319
 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 <lifedj27@gmail.com>, Artem Dubrovskiy <ad13box@gmail.com>
 License-Expression: Apache-2.0

{alita_sdk-0.3.317 → alita_sdk-0.3.319}/pyproject.toml

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