alita-sdk 0.3.290__py3-none-any.whl → 0.3.292__py3-none-any.whl

alita_sdk/configurations/github.py

@@ -84,4 +84,68 @@ class GithubConfiguration(BaseModel):
             "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."
-        )
+        )
+
+    @staticmethod
+    def check_connection(settings: dict) -> str | None:
+        """
+        Check GitHub connection using provided settings.
+        Returns None if connection is successful, error message otherwise.
+        """
+        import requests
+        from requests.auth import HTTPBasicAuth
+        import jwt
+        import time
+
+        base_url = settings.get('base_url', 'https://api.github.com')
+        access_token = settings.get('access_token')
+        username = settings.get('username')
+        password = settings.get('password')
+        app_id = settings.get('app_id')
+        app_private_key = settings.get('app_private_key')
+
+        # if all auth methods are None or empty, allow anonymous access
+        if not any([access_token, (username and password), (app_id and app_private_key)]):
+            return None
+
+        headers = {'Accept': 'application/vnd.github.v3+json'}
+        auth = None
+
+        try:
+            # Determine authentication method
+            if access_token:
+                headers['Authorization'] = f'token {access_token}'
+            elif username and password:
+                auth = HTTPBasicAuth(username, password)
+            elif app_id and app_private_key:
+                # Generate JWT for GitHub App authentication
+                payload = {
+                    'iat': int(time.time()),
+                    'exp': int(time.time()) + 600,  # 10 minutes
+                    'iss': app_id
+                }
+                jwt_token = jwt.encode(payload, app_private_key, algorithm='RS256')
+                headers['Authorization'] = f'Bearer {jwt_token}'
+
+            # Test connection with user endpoint
+            response = requests.get(f'{base_url}/user', headers=headers, auth=auth, timeout=10)
+
+            if response.status_code == 200:
+                return None
+            elif response.status_code == 401:
+                return "Authentication failed: Invalid credentials"
+            elif response.status_code == 403:
+                return "Access forbidden: Check your permissions"
+            elif response.status_code == 404:
+                return "GitHub API endpoint not found"
+            else:
+                return f"Connection failed with status {response.status_code}: {response.text}"
+
+        except requests.exceptions.ConnectionError:
+            return "Connection error: Unable to reach GitHub API"
+        except requests.exceptions.Timeout:
+            return "Connection timeout: GitHub API did not respond in time"
+        except jwt.InvalidKeyError:
+            return "Invalid private key format for GitHub App authentication"
+        except Exception as e:
+            return f"Unexpected error: {str(e)}"

alita_sdk/configurations/zephyr_essential.py

@@ -1,3 +1,5 @@
+from typing import Optional
+
 from pydantic import BaseModel, ConfigDict, Field, SecretStr
 
 
@@ -14,5 +16,5 @@ class ZephyrEssentialConfiguration(BaseModel):
             }
         }
     )
-    base_url: str = Field(description="Zephyr Essential Base URL")
+    base_url: Optional[str] = Field(description="Zephyr Essential API Base URL", default=None)
     token: SecretStr = Field(description="Zephyr Essential API Token")

alita_sdk/runtime/langchain/document_loaders/AlitaMarkdownLoader.py

@@ -0,0 +1,66 @@
+from pathlib import Path
+from typing import Any, List, Union, Generator, Iterator
+from langchain_core.documents import Document
+
+from langchain_community.document_loaders.unstructured import (
+    UnstructuredFileLoader,
+    validate_unstructured_version,
+)
+
+class AlitaMarkdownLoader(UnstructuredFileLoader):
+
+    def __init__(
+        self,
+        file_path: Union[str, Path],
+        mode: str = "elements",
+        chunker_config: dict = None,
+        **unstructured_kwargs: Any,
+    ):
+        """
+        Args:
+            file_path: The path to the Markdown file to load.
+            mode: The mode to use when loading the file. Can be one of "single",
+                "multi", or "all". Default is "single".
+            chunker_config: Configuration dictionary for the markdown chunker.
+            **unstructured_kwargs: Any kwargs to pass to the unstructured.
+        """
+        file_path = str(file_path)
+        validate_unstructured_version("0.4.16")
+        self.chunker_config = chunker_config or {
+            "strip_header": False,
+            "return_each_line": False,
+            "headers_to_split_on": [],
+            "max_tokens": 512,
+            "token_overlap": 10,
+        }
+        super().__init__(file_path=file_path, mode=mode, **unstructured_kwargs)
+
+    def _file_content_generator(self) -> Generator[Document, None, None]:
+        """
+        Creates a generator that yields a single Document object
+        representing the entire content of the Markdown file.
+        """
+        with open(self.file_path, "r", encoding="utf-8") as file:
+            content = file.read()
+        yield Document(page_content=content, metadata={"source": self.file_path})
+
+    def _get_elements(self) -> List[Document]:
+        """
+        Processes the Markdown file using the markdown_chunker and returns the chunks.
+        """
+        from alita_sdk.tools.chunkers.sematic.markdown_chunker import markdown_chunker
+
+        # Create a generator for the file content
+        file_content_generator = self._file_content_generator()
+
+        # Use the markdown_chunker to process the content
+        chunks = markdown_chunker(file_content_generator, config=self.chunker_config)
+
+        # Convert the generator to a list of Document objects
+        return list(chunks)
+
+    def lazy_load(self) -> Iterator[Document]:
+        """Load file."""
+        elements = self._get_elements()
+        self._post_process_elements(elements)
+        yield from elements

alita_sdk/runtime/langchain/document_loaders/AlitaPDFLoader.py

@@ -67,10 +67,13 @@ class AlitaPDFLoader:
             return self._load_docs()
 
     def _load_docs(self):
-        return PyPDFLoader(file_path=self.file_path,
+        docs = PyPDFLoader(file_path=self.file_path,
                         password=self.password,
                         headers=self.headers,
                         extract_images=self.extract_images,
                         extraction_mode=self.extraction_mode,
                         images_parser=ImageParser(llm=self.llm, prompt=self.prompt),
-                        extraction_kwargs=self.extraction_kwargs).load()
+                        extraction_kwargs=self.extraction_kwargs).load()
+        for doc in docs:
+            doc.metadata['chunk_id'] = doc.metadata['page']
+        return docs

alita_sdk/runtime/langchain/document_loaders/constants.py

@@ -25,6 +25,7 @@ from .AlitaJSONLoader import AlitaJSONLoader
 from .AlitaPDFLoader import AlitaPDFLoader
 from .AlitaPowerPointLoader import AlitaPowerPointLoader
 from .AlitaTextLoader import AlitaTextLoader
+from .AlitaMarkdownLoader import AlitaMarkdownLoader
 
 loaders_map = {
     '.png': {
@@ -96,7 +97,7 @@ loaders_map = {
         'allowed_to_override': ['max_tokens']
     },
     '.md': {
-        'class': UnstructuredMarkdownLoader,
+        'class': AlitaMarkdownLoader,
         'is_multimodal_processing': False,
         'kwargs': {},
         'allowed_to_override': ['max_tokens']

alita_sdk/tools/ado/repos/repos_wrapper.py

@@ -159,6 +159,7 @@ class ArgsSchema(Enum):
             Field(
                 description="""List of comments, where each comment is a dictionary specifying details about the comment,
                 e.g. [{'file_path': 'src/main.py', 'comment_text': 'Logic needs improvement', 'right_line': 20}]""",
+                default=None,
                 examples=[
                     {
                         "file_path": "src/main.py",

alita_sdk/tools/jira/api_wrapper.py

@@ -18,7 +18,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 load_file_docs
+from ..utils.content_parser import file_extension_by_chunker
 from ...runtime.utils.utils import IndexerKeywords
 
 logger = logging.getLogger(__name__)
@@ -1262,6 +1262,7 @@ class JiraApiWrapper(NonCodeIndexerToolkit):
         self._include_attachments = kwargs.get('include_attachments', False)
         self._included_fields = fields_to_extract.copy() if fields_to_extract else []
         self._include_comments = kwargs.get('include_comments', True)
+        self._chunking_tool = kwargs.get('chunking_tool', None)
 
         try:
             # Prepare fields to extract
@@ -1313,7 +1314,8 @@ class JiraApiWrapper(NonCodeIndexerToolkit):
                                                                            doc.page_content,
                                                                            attachment_resolver),
                                     doc.page_content)
-            doc.page_content = processed_content
+            doc.metadata[IndexerKeywords.CONTENT_IN_BYTES.value] = processed_content.encode('utf-8')
+            doc.metadata[IndexerKeywords.CONTENT_FILE_NAME.value] = f"base_doc{file_extension_by_chunker(self._chunking_tool)}"
             yield doc
 
     def _process_document(self, base_document: Document) -> Generator[Document, None, None]:
@@ -1337,13 +1339,11 @@ class JiraApiWrapper(NonCodeIndexerToolkit):
                     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_docs = load_file_docs(file_content=attachment_content, file_name=attachment['filename'], llm=self.llm, is_capture_image=True, excel_by_sheets=True)
-                    if not content_docs or isinstance(content_docs, ToolException):
-                        continue
-                    for doc in content_docs:
-                        yield Document(page_content=doc.page_content,
-                                       metadata={
-                                           **doc.metadata,
+
+                    yield Document(page_content='',
+                                   metadata={
+                                           IndexerKeywords.CONTENT_IN_BYTES.value: attachment_content,
+                                           IndexerKeywords.CONTENT_FILE_NAME.value: attachment['filename'],
                                            'id': attachment_id,
                                            'issue_key': issue_key,
                                            'source': f"{self.base_url}/browse/{issue_key}",
@@ -1358,8 +1358,10 @@ class JiraApiWrapper(NonCodeIndexerToolkit):
             comments = self.get_processed_comments_list_with_image_description(issue_key)
             if comments:
                 for comment in comments:
-                    yield Document(page_content=comment.get('processed_content'),
+                    yield Document(page_content='',
                                    metadata={
+                                       IndexerKeywords.CONTENT_IN_BYTES.value: comment.get('processed_content').encode('utf-8'),
+                                       IndexerKeywords.CONTENT_FILE_NAME.value: "comment.md",
                                        'id': comment.get('id'),
                                        'issue_key': issue_key,
                                        'source': f"{self.base_url}/browse/{issue_key}",

alita_sdk/tools/utils/content_parser.py

@@ -234,12 +234,29 @@ def process_content_by_type(document: Document, content, extension_source: str,
             if chunking_config and (users_config_for_extension := chunking_config.get(extension, {})):
                 for key in set(users_config_for_extension.keys()) & set(allowed_to_override):
                     loader_kwargs[key] = users_config_for_extension[key]
-
+            loader_kwargs['llm'] = llm
+            loader_kwargs['prompt'] = image_processing_prompt
             loader = loader_cls(file_path=temp_file_path, **loader_kwargs)
-            for chunk in loader.load():
+            counter = 1
+            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:
+                if 'chunk_id' not in chunk.metadata:
+                    chunk.metadata['chunk_id'] = counter
+                document_metadata = document.metadata.copy()
+                document_metadata['id'] = f"{document.metadata['id']}_{chunk.metadata['chunk_id']}"
+                counter+=1
                 yield Document(
                     page_content=sanitize_for_postgres(chunk.page_content),
-                    metadata={**document.metadata, **chunk.metadata}
+                    metadata={**document_metadata, **chunk.metadata}
                 )
     finally:
         if temp_file_path and os.path.exists(temp_file_path):

alita_sdk/tools/vector_adapters/VectorStoreAdapter.py

@@ -53,10 +53,16 @@ class PGVectorAdapter(VectorStoreAdapter):
     """Adapter for PGVector database operations."""
 
     def get_vectorstore_params(self, collection_name: str, connection_string: Optional[str] = None) -> Dict[str, Any]:
+        try:
+            from tools import this  # pylint: disable=E0401,C0415
+            worker_config = this.for_module("indexer_worker").descriptor.config
+        except:  # pylint: disable=W0702
+            worker_config = {}
+        #
         return {
             "use_jsonb": True,
             "collection_name": collection_name,
-            "create_extension": True,
+            "create_extension": worker_config.get("pgvector_create_extension", True),
             "alita_sdk_options": {
                 "target_schema": collection_name,
             },

alita_sdk/tools/xray/api_wrapper.py

@@ -16,7 +16,7 @@ from ..elitea_base import (
 from ..non_code_indexer_toolkit import NonCodeIndexerToolkit
 from ..utils.available_tools_decorator import extend_with_parent_available_tools
 from ...runtime.utils.utils import IndexerKeywords
-from ..utils.content_parser import load_file_docs
+from ..utils.content_parser import file_extension_by_chunker
 
 try:
     from alita_sdk.runtime.langchain.interfaces.llm_processor import get_embeddings
@@ -109,7 +109,7 @@ XrayCreateTest = create_model(
 
 XrayCreateTests = create_model(
     "XrayCreateTests",
-    graphql_mutations=(list[str], Field(description="list of GraphQL mutations:\n" + _graphql_mutation_description))
+    graphql_mutations=(List[str], Field(description="list of GraphQL mutations:\n" + _graphql_mutation_description))
 )
 
 def _parse_tests(test_results) -> List[Any]:
@@ -417,7 +417,9 @@ class XrayApiWrapper(NonCodeIndexerToolkit):
                     if attachments_data:
                         metadata["_attachments_data"] = attachments_data
 
-                yield Document(page_content=page_content, metadata=metadata)
+                metadata[IndexerKeywords.CONTENT_IN_BYTES.value] = page_content.encode('utf-8')
+                metadata[IndexerKeywords.CONTENT_FILE_NAME.value] = f"base_doc{file_extension_by_chunker(kwargs['chunking_tool'])}"
+                yield Document(page_content='', metadata=metadata)
                 
         except Exception as e:
             logger.error(f"Error processing test data: {e}")
@@ -559,17 +561,9 @@ class XrayApiWrapper(NonCodeIndexerToolkit):
             logger.error(f"Error processing attachment: {e}")
 
     def _load_attachment(self, content, file_name, attachment_metadata) -> Generator[Document, None, None]:
-        content_docs = load_file_docs(file_content=content, file_name=file_name,
-                                      llm=self.llm, is_capture_image=True, excel_by_sheets=True)
-
-        if not content_docs or isinstance(content_docs, ToolException):
-            return
-        for doc in content_docs:
-            yield Document(page_content=doc.page_content,
-                           metadata={
-                               **doc.metadata,
-                               **attachment_metadata
-                           })
+        attachment_metadata[IndexerKeywords.CONTENT_IN_BYTES.value] = content
+        attachment_metadata[IndexerKeywords.CONTENT_FILE_NAME.value] = file_name
+        yield Document(page_content='', metadata=attachment_metadata)
 
     def _index_tool_params(self, **kwargs) -> dict[str, tuple[type, Field]]:
         return {

alita_sdk/tools/zephyr_enterprise/api_wrapper.py

@@ -1,3 +1,4 @@
+import json
 import logging
 from typing import Optional, List, Generator, Literal
 
@@ -8,6 +9,8 @@ from langchain_core.documents import Document
 from .zephyr_enterprise import ZephyrClient
 from ..non_code_indexer_toolkit import NonCodeIndexerToolkit
 from ..utils.available_tools_decorator import extend_with_parent_available_tools
+from ..utils.content_parser import file_extension_by_chunker
+from ...runtime.utils.utils import IndexerKeywords
 
 logger = logging.getLogger(__name__)
 
@@ -157,6 +160,7 @@ class ZephyrApiWrapper(NonCodeIndexerToolkit):
         }
 
     def _base_loader(self, zql: str, **kwargs) -> Generator[Document, None, None]:
+        self._chunking_tool = kwargs.get('chunking_tool', None)
         test_cases = self.get_testcases_by_zql(zql=zql, return_as_list=True)
         for test_case in test_cases:
             metadata = {
@@ -175,7 +179,10 @@ class ZephyrApiWrapper(NonCodeIndexerToolkit):
             try:
                 id = document.metadata['id']
                 test_case_content = self.get_test_case_steps(id)
-                document.page_content = f"Test case: {document.metadata['name']}\nTest_case_content: {test_case_content}"
+                page_content = json.dumps(test_case_content)
+                document.metadata[IndexerKeywords.CONTENT_IN_BYTES.value] = page_content.encode('utf-8')
+                document.metadata[
+                    IndexerKeywords.CONTENT_FILE_NAME.value] = f"base_doc{file_extension_by_chunker(self._chunking_tool)}"
             except Exception as e:
                 logging.error(f"Failed to process document: {e}")
             yield document

alita_sdk/tools/zephyr_essential/api_wrapper.py

@@ -9,6 +9,8 @@ from langchain_core.tools import ToolException
 
 from ..non_code_indexer_toolkit import NonCodeIndexerToolkit
 from ..utils.available_tools_decorator import extend_with_parent_available_tools
+from ..utils.content_parser import file_extension_by_chunker
+from ...runtime.utils.utils import IndexerKeywords
 
 
 class ZephyrEssentialApiWrapper(NonCodeIndexerToolkit):
@@ -270,6 +272,7 @@ class ZephyrEssentialApiWrapper(NonCodeIndexerToolkit):
         }
 
     def _base_loader(self, **kwargs) -> Generator[Document, None, None]:
+        self._chunking_tool = kwargs.get('chunking_tool', None)
         try:
             test_cases = self.list_test_cases()
         except Exception as e:
@@ -291,7 +294,10 @@ class ZephyrEssentialApiWrapper(NonCodeIndexerToolkit):
                     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)
+                            page_content = json.dumps(content)
+                            document.metadata[IndexerKeywords.CONTENT_IN_BYTES.value] = page_content.encode('utf-8')
+                            document.metadata[
+                                IndexerKeywords.CONTENT_FILE_NAME.value] = f"base_doc{file_extension_by_chunker(self._chunking_tool)}"
                             document.metadata["steps_type"] = steps_type
             except Exception as e:
                 logging.error(f"Failed to process document: {e}")
@@ -614,8 +620,74 @@ UpdateTestCase = create_model(
     json=(str, Field(description=("""
         JSON body to update a test case. Example:
         {
-          "name": "Updated Test Case Name",
-          "description": "Updated Test Case Description"
+          "id": 1,
+          "key": "SA-T10",
+          "name": "Check axial pump",
+          "project": {
+            "id": 10005,
+            "self": "https://<api-base-url>/projects/10005"
+          },
+          "createdOn": "2018-05-15T13:15:13Z",
+          "objective": "To ensure the axial pump can be enabled",
+          "precondition": "Latest version of the axial pump available",
+          "estimatedTime": 138000,
+          "labels": [
+            "Regression",
+            "Performance",
+            "Automated"
+          ],
+          "component": {
+            "id": 10001,
+            "self": "https://<jira-instance>.atlassian.net/rest/api/2/component/10001"
+          },
+          "priority": {
+            "id": 10002,
+            "self": "https://<api-base-url>/priorities/10002"
+          },
+          "status": {
+            "id": 10000,
+            "self": "https://<api-base-url>/statuses/10000"
+          },
+          "folder": {
+            "id": 100006,
+            "self": "https://<api-base-url>/folders/10006"
+          },
+          "owner": {
+            "self": "https://<jira-instance>.atlassian.net/rest/api/2/user?accountId=5b10a2844c20165700ede21g",
+            "accountId": "5b10a2844c20165700ede21g"
+          },
+          "testScript": {
+            "self": "https://<api-base-url>/testCases/PROJ-T1/testscript"
+          },
+          "customFields": {
+            "Build Number": 20,
+            "Release Date": "2020-01-01",
+            "Pre-Condition(s)": "User should have logged in. <br> User should have navigated to the administration panel.",
+            "Implemented": false,
+            "Category": [
+              "Performance",
+              "Regression"
+            ],
+            "Tester": "fa2e582e-5e15-521e-92e3-47e6ca2e7256"
+          },
+          "links": {
+            "self": "string",
+            "issues": [
+              {
+                "self": "string",
+                "issueId": 10100,
+                "id": 1,
+                "target": "https://<jira-instance>.atlassian.net/rest/api/2/issue/10000",
+                "type": "COVERAGE"
+              }
+            ],
+            "webLinks": [
+              {
+                "self": "string",
+                "description": "A link to atlassian.com",
+                "url": "https://atlassian.com",
+                "id": 1,
+                "type": "COVERAGE"
         }
         """
     )))
@@ -632,9 +704,9 @@ CreateTestCaseIssueLink = create_model(
     json=(str, Field(description=("""
         JSON body to create an issue link. Example:
         {
-          "issueKey": "ISSUE_KEY",
-          "description": "Link Description"
+          "issueId": 10100
         }
+        where issueId - Jira issue id
         """
     )))
 )
@@ -676,8 +748,10 @@ CreateTestCaseTestScript = create_model(
     json=(str, Field(description=("""
         JSON body to create a test script. Example:
         {
-          "script": "Test Script Content"
+          "type": "bdd",
+          "text": "Attempt to login to the application"
         }
+        Where type - Test script type. Allowed: plain, bdd
         """
     )))
 )
@@ -768,9 +842,71 @@ UpdateTestCycle = create_model(
     json=(str, Field(description=("""
         JSON body to update a test cycle. Example:
         {
-          "name": "Updated Test Cycle Name",
-          "description": "Updated Test Cycle Description"
-        }
+          "id": 1,
+          "key": "SA-R40",
+          "name": "Sprint 1 Regression Test Cycle",
+          "project": {
+            "id": 10005,
+            "self": "https://<api-base-url>/projects/10005"
+          },
+          "jiraProjectVersion": {
+            "id": 10000,
+            "self": "https://<jira-instance>.atlassian.net/rest/api/2/version/10000"
+          },
+          "status": {
+            "id": 10000,
+            "self": "https://<api-base-url>/statuses/10000"
+          },
+          "folder": {
+            "id": 100006,
+            "self": "https://<api-base-url>/folders/10006"
+          },
+          "description": "Regression test cycle 1 to ensure no breaking changes",
+          "plannedStartDate": "2018-05-19T13:15:13Z",
+          "plannedEndDate": "2018-05-20T13:15:13Z",
+          "owner": {
+            "self": "https://<jira-instance>.atlassian.net/rest/api/2/user?accountId=5b10a2844c20165700ede21g",
+            "accountId": "5b10a2844c20165700ede21g"
+          },
+          "customFields": {
+            "Build Number": 20,
+            "Release Date": "2020-01-01",
+            "Pre-Condition(s)": "User should have logged in. <br> User should have navigated to the administration panel.",
+            "Implemented": false,
+            "Category": [
+              "Performance",
+              "Regression"
+            ],
+            "Tester": "fa2e582e-5e15-521e-92e3-47e6ca2e7256"
+          },
+          "links": {
+            "self": "string",
+            "issues": [
+              {
+                "self": "string",
+                "issueId": 10100,
+                "id": 1,
+                "target": "https://<jira-instance>.atlassian.net/rest/api/2/issue/10000",
+                "type": "COVERAGE"
+              }
+            ],
+            "webLinks": [
+              {
+                "self": "string",
+                "description": "A link to atlassian.com",
+                "url": "https://atlassian.com",
+                "id": 1,
+                "type": "COVERAGE"
+              }
+            ],
+            "testPlans": [
+              {
+                "id": 1,
+                "self": "https://<api-base-url>/links/1",
+                "type": "RELATED",
+                "testPlanId": 2,
+                "target": "https://<jira-instance>.atlassian.net/rest/api/2/testplan/123"
+              }
         """
     )))
 )
@@ -786,9 +922,9 @@ CreateTestCycleIssueLink = create_model(
     json=(str, Field(description=("""
         JSON body to create an issue link. Example:
         {
-          "issueKey": "ISSUE_KEY",
-          "description": "Link Description"
+          "issueId": 10100
         }
+        where issueId - Jira issue id
         """
     )))
 )
@@ -820,10 +956,34 @@ CreateTestExecution = create_model(
     json=(str, Field(description=("""
         JSON body to create a test execution. Example:
         {
-          "testCaseKey": "TEST_CASE_KEY",
-          "testCycleKey": "TEST_CYCLE_KEY",
-          "status": "PASS"
-        }
+          "projectKey": "TIS",
+          "testCaseKey": "SA-T10",
+          "testCycleKey": "SA-R10",
+          "statusName": "In Progress",
+          "testScriptResults": [
+            {
+              "statusName": "In Progress",
+              "actualEndDate": "2018-05-20T13:15:13Z",
+              "actualResult": "User logged in successfully"
+            }
+          ],
+          "environmentName": "Chrome Latest Version",
+          "actualEndDate": "2018-05-20T13:15:13Z",
+          "executionTime": 120000,
+          "executedById": "5b10a2844c20165700ede21g",
+          "assignedToId": "5b10a2844c20165700ede21g",
+          "comment": "Test failed user could not login",
+          "customFields": {
+            "Build Number": 20,
+            "Release Date": "2020-01-01",
+            "Pre-Condition(s)": "User should have logged in. <br> User should have navigated to the administration panel.",
+            "Implemented": false,
+            "Category": [
+              "Performance",
+              "Regression"
+            ],
+            "Tester": "fa2e582e-5e15-521e-92e3-47e6ca2e7256"
+          }
         """
     )))
 )
@@ -839,8 +999,13 @@ UpdateTestExecution = create_model(
     json=(str, Field(description=("""
         JSON body to update a test execution. Example:
         {
-          "status": "FAIL",
-          "comment": "Updated comment"
+          "statusName": "In Progress",
+          "environmentName": "Chrome Latest Version",
+          "actualEndDate": "2018-05-20T13:15:13Z",
+          "executionTime": 120000,
+          "executedById": "5b10a2844c20165700ede21g",
+          "assignedToId": "5b10a2844c20165700ede21g",
+          "comment": "Test failed user could not login"
         }
         """
     )))
@@ -858,12 +1023,14 @@ UpdateTestExecutionTestSteps = create_model(
     test_execution_id_or_key=(str, Field(description="ID or key of the test execution to update test steps for.")),
     json=(str, Field(description=("""
         "JSON body to update test steps. Example:
-        ["
-          {"
-            "step": "Step 1",
-            "status": "PASS"
-          }
-        ]
+        {
+          "steps": [
+            {
+              "actualResult": "User logged in successfully",
+              "statusName": "In Progress"
+            }
+          ]
+        }
         """
     )))
 )
@@ -884,9 +1051,9 @@ CreateTestExecutionIssueLink = create_model(
     json=(str, Field(description=("""
         JSON body to create an issue link. Example:
         {
-          "issueKey": "ISSUE_KEY",
-          "description": "Link Description"
+          "issueId": 10100
         }
+        where issueId - Jira issue id
         """
     )))
 )

alita_sdk/tools/zephyr_essential/client.py

@@ -1,4 +1,5 @@
 import requests
+from langchain_core.tools import ToolException
 
 class ZephyrEssentialAPI:
     def __init__(self, base_url: str, token: str):
@@ -14,10 +15,11 @@ class ZephyrEssentialAPI:
         })
         try:
             resp = requests.request(method=method, url=url, headers=headers, json=json, params=params, files=files)
-            resp.raise_for_status()
-            if resp.headers.get("Content-Type", "").startswith("application/json"):
-                return resp.json()
-            return resp.text
+            if resp.status_code < 300:
+                if resp.headers.get("Content-Type", "").startswith("application/json"):
+                    return resp.json()
+                return resp.text
+            return ToolException(f"Error performing request {method} {api_path}: {resp.content}")
         except requests.RequestException as e:
             raise Exception(f"Error performing request {method} {api_path}: {str(e)}")
 

alita_sdk/tools/zephyr_scale/api_wrapper.py

@@ -12,6 +12,8 @@ from langchain_core.documents import Document
 
 from ..non_code_indexer_toolkit import NonCodeIndexerToolkit
 from ..utils.available_tools_decorator import extend_with_parent_available_tools
+from ..utils.content_parser import file_extension_by_chunker
+from ...runtime.utils.utils import IndexerKeywords
 
 try:
     from alita_sdk.runtime.langchain.interfaces.llm_processor import get_embeddings
@@ -1218,6 +1220,7 @@ class ZephyrScaleApiWrapper(NonCodeIndexerToolkit):
         }
 
     def _base_loader(self, project_key: str, jql: str, **kwargs) -> Generator[Document, None, None]:
+        self._chunking_tool = kwargs.get('chunking_tool', None)
         for test_case_doc in self._get_test_cases_docs(project_key, jql):
             yield test_case_doc
         for folder_doc in self._get_folders_docs(project_key):
@@ -1284,7 +1287,10 @@ class ZephyrScaleApiWrapper(NonCodeIndexerToolkit):
                     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)
+                            page_content = json.dumps(content)
+                            document.metadata[IndexerKeywords.CONTENT_IN_BYTES.value] = page_content.encode('utf-8')
+                            document.metadata[
+                                IndexerKeywords.CONTENT_FILE_NAME.value] = f"base_doc{file_extension_by_chunker(self._chunking_tool)}"
                             document.metadata["steps_type"] = steps_type
             except Exception as e:
                 logging.error(f"Failed to process document: {e}")

{alita_sdk-0.3.290.dist-info → alita_sdk-0.3.292.dist-info}/METADATA

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

@@ -12,7 +12,7 @@ alita_sdk/configurations/confluence.py,sha256=mAW2fgSEOg-BAV768Sc6b_EuRA3H5UL9xf
 alita_sdk/configurations/delta_lake.py,sha256=sCNDClaO6-b1Qv5WDxOMHjQUrHnr6S6EPQuS9o6q8Z8,1130
 alita_sdk/configurations/embedding.py,sha256=8GSC8Feh8CH7bT_6cQhNqlS6raE91S2YRAtb2N9bUA8,552
 alita_sdk/configurations/figma.py,sha256=vecZ20IyZgnFO2GdphkovYHMISRPcUYh7fxkUQsPwX8,1306
-alita_sdk/configurations/github.py,sha256=NAiXotsl766IwjOOsVrs70PuoP4gvXla2Yycsv1qBpI,3569
+alita_sdk/configurations/github.py,sha256=6F7P9BPu9zE54q8m8IPjIct8cEX3WODYzZvn9WtxOXI,6283
 alita_sdk/configurations/gitlab.py,sha256=0W35igIlue6QxOnPgw65ToLf4HSdPVuRyObdwQuEld8,1053
 alita_sdk/configurations/google_places.py,sha256=jXhlPXywkDhHNrgB4KoVWogf3CVi1bY3wFLRzDqMr-E,589
 alita_sdk/configurations/jira.py,sha256=ASh8I2iVXzOOtwjRX7kYNllXpCXyAIxFMP_YD4Q0PTI,1379
@@ -32,7 +32,7 @@ alita_sdk/configurations/testrail.py,sha256=k0fPmHBIrWAfEKhrDdB9Rdirw-UFHFoXkReP
 alita_sdk/configurations/xray.py,sha256=_XCW5eGEAyhDuRvXCtTAZ24BGsIwzvIaeAZT10xoW6M,1147
 alita_sdk/configurations/zephyr.py,sha256=ndqGYFy5OFxjoXB7DzC71rd5W6qGBGAlKMWoqT8TuNk,1653
 alita_sdk/configurations/zephyr_enterprise.py,sha256=UaBk3qWcT2-bCzko5HEPvgxArw1ZpvOvwXwFYrIHZjM,710
-alita_sdk/configurations/zephyr_essential.py,sha256=dXZHw4Yn_Mg-VhUn7qeCbd4hVpt9FzYbjiZ9ckAbFck,695
+alita_sdk/configurations/zephyr_essential.py,sha256=tUIrh-PRNvdrLBj6rJXqlF-h6oaMXUQI1wgit07kFBw,752
 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=TPvROw1qu4IyUEGuf7x40IKRpb5eFZpYGN3-8LfQE0M,3461
@@ -61,14 +61,15 @@ alita_sdk/runtime/langchain/document_loaders/AlitaGitRepoLoader.py,sha256=5WXGcy
 alita_sdk/runtime/langchain/document_loaders/AlitaImageLoader.py,sha256=QwgBJE-BvOasjgT1hYHZc0MP0F_elirUjSzKixoM6fY,6610
 alita_sdk/runtime/langchain/document_loaders/AlitaJSONLoader.py,sha256=1mGZjltnqsSXkp1Jw-lQroyNFiCPpjb9ZbdoqOlqPeU,3354
 alita_sdk/runtime/langchain/document_loaders/AlitaJiraLoader.py,sha256=M2q3YThkps0yAZOjfoLcyE7qycVTYKcXEGtpmp0N6C8,10950
-alita_sdk/runtime/langchain/document_loaders/AlitaPDFLoader.py,sha256=PrYQYPPzo1bZSqMFmoFube5c7XMBUQTySM-tbtbOcWE,3281
+alita_sdk/runtime/langchain/document_loaders/AlitaMarkdownLoader.py,sha256=RGHDfleYTn7AAc3H-yFZrjm06L0Ux14ZtEJpFlVBNCA,2474
+alita_sdk/runtime/langchain/document_loaders/AlitaPDFLoader.py,sha256=toXdQbT9TuBCdB4t62t2cPalBY_2RZy2lqKSMU7YVhw,3386
 alita_sdk/runtime/langchain/document_loaders/AlitaPowerPointLoader.py,sha256=SKAAPo3DfMtRPxICKrPzlXXkC5RfaeiRj7lejLXTi7o,2337
 alita_sdk/runtime/langchain/document_loaders/AlitaQtestLoader.py,sha256=CUVVnisxm7b5yZWV6rn0Q3MEEaO1GWNcfnz5yWz8T0k,13283
 alita_sdk/runtime/langchain/document_loaders/AlitaTableLoader.py,sha256=o0SRFPZ-VskltgThVRX80rT19qtB4gPzxED9SENTNWo,4145
 alita_sdk/runtime/langchain/document_loaders/AlitaTextLoader.py,sha256=uNcV0En49_0u0RYB1sP1XfNspT2Xc5CacuJr9Jqv79Q,2972
 alita_sdk/runtime/langchain/document_loaders/ImageParser.py,sha256=gao5yCCKdDai_Gx7YdEx5U6oMyJYzn69eYmEvWLh-fc,656
 alita_sdk/runtime/langchain/document_loaders/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-alita_sdk/runtime/langchain/document_loaders/constants.py,sha256=9v4ZRbFy9SqyF5uaQwIAJ3IgGanb8ISoZMKgEYEhOLU,5594
+alita_sdk/runtime/langchain/document_loaders/constants.py,sha256=xdZlVZhqWFrxYXTk5E7IOKLT7MowmMQSPoRjk0StQEw,5640
 alita_sdk/runtime/langchain/document_loaders/utils.py,sha256=9xghESf3axBbwxATyVuS0Yu-TWe8zWZnXgCD1ZVyNW0,2414
 alita_sdk/runtime/langchain/interfaces/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/runtime/langchain/interfaces/kwextractor.py,sha256=kSJA9L8g8UArmHu7Bd9dIO0Rrq86JPUb8RYNlnN68FQ,3072
@@ -137,7 +138,7 @@ alita_sdk/tools/non_code_indexer_toolkit.py,sha256=v9uq1POE1fQKCd152mbqDtF-HSe0q
 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=n-IhKED05RwQGWT4LfCaxJ85uDyG4S9zTjSjK6A8N4o,5192
-alita_sdk/tools/ado/repos/repos_wrapper.py,sha256=e3bGsM03m0UggSQfoVh5Gg_M1MYt_BTKS-s9G2Unc1k,49739
+alita_sdk/tools/ado/repos/repos_wrapper.py,sha256=y1wJZNhuoWaSkFsGU5Pct3LClc1xfgmgBy2u_dgBF-4,49769
 alita_sdk/tools/ado/test_plan/__init__.py,sha256=4fEw_3cm4shuZ868HhAU-uMH3xNXPyb3uRjyNWoBKls,5243
 alita_sdk/tools/ado/test_plan/test_plan_wrapper.py,sha256=p3mbBiu0jthZG-Bj2-duO7DDZQ4LHYZ6Q1nOVd-KTrc,21944
 alita_sdk/tools/ado/wiki/__init__.py,sha256=uBKo_Meu2ZxMxcxGsMmvCXyplRE2um1_PIRvdYd37rM,5171
@@ -258,7 +259,7 @@ alita_sdk/tools/google/bigquery/tool.py,sha256=Esf9Hsp8I0e7-5EdkFqQ-bid0cfrg-bfS
 alita_sdk/tools/google_places/__init__.py,sha256=Tg_dfKTc0qxcG-1HVuQQB11PYph2RDWSUVhrlgxqk64,3491
 alita_sdk/tools/google_places/api_wrapper.py,sha256=7nZly6nk4f4Tm7s2MVdnnwlb-1_WHRrDhyjDiqoyPjA,4674
 alita_sdk/tools/jira/__init__.py,sha256=0NJikFWEqK8DSohXPUYD4iDLJFS8btxhqQ60acWwC3k,6063
-alita_sdk/tools/jira/api_wrapper.py,sha256=qflTFkkP1CDYKWPvHyiFiIaqTZm_Bom4ZnhdcWnFRe4,79019
+alita_sdk/tools/jira/api_wrapper.py,sha256=Cm9sULiw4Til6wmhsVM3-IIUAMkRVbWdRsScEUYHss4,79260
 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
@@ -325,11 +326,11 @@ alita_sdk/tools/testrail/__init__.py,sha256=0kETjWKLU7R6mugBWsjwEUsh10pipbAeNSGJ
 alita_sdk/tools/testrail/api_wrapper.py,sha256=xKQbjwL602J55KZiAdMcMtsuzK2jky0DUcrrdsazj0A,32981
 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=VKJAYIRGZqal2OV9fdaKGz3pJIaPcFO6HRXNUQwG6eI,13056
-alita_sdk/tools/vector_adapters/VectorStoreAdapter.py,sha256=a6FAsiix_EvATIKUf5YT6vHh5LDyJ5uSP3LJqoxFo04,17367
+alita_sdk/tools/utils/content_parser.py,sha256=tppnQfgoER24acx3CJW1AUvkay_C7lvz_qVB-r-RQOo,13852
+alita_sdk/tools/vector_adapters/VectorStoreAdapter.py,sha256=ypBEAkFRGHv5edW0N9rdo1yKurNGQ4pRVEWtrN_7SeA,17656
 alita_sdk/tools/vector_adapters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/tools/xray/__init__.py,sha256=AsHDvRgyD-6vvGyAyQDPWHbOD2WoMJ5Llt029bLuu6U,4277
-alita_sdk/tools/xray/api_wrapper.py,sha256=z-eIS4ZYd42AxfUMrJv_nT4JWBZ_dOvlrToSpWjZ5zs,30512
+alita_sdk/tools/xray/api_wrapper.py,sha256=DLQ5dT9XUiTJWc7OT6L9mU6CIZ8Sf6hGSBunYA6_k5k,30441
 alita_sdk/tools/yagmail/__init__.py,sha256=c4Qn3em0tLxzRmFKpzbBgY9W2EnOoKf0azoDJHng5CY,2208
 alita_sdk/tools/yagmail/yagmail_wrapper.py,sha256=SKoGVd1X4Ew3ad5tOdtPoY00M6jStNdT3q7GXEjQc5g,1952
 alita_sdk/tools/zephyr/Zephyr.py,sha256=ODZbg9Aw0H0Rbv-HcDXLI4KHbPiLDHoteDofshw9A_k,1508
@@ -337,18 +338,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=IoWQPH2lf2Yuj2ejZ74818JTXdrMoh_ZxAJORukYODU,4172
-alita_sdk/tools/zephyr_enterprise/api_wrapper.py,sha256=JJ3sOf_qEVogqYNdPno0WkJE9MTf0cP29h9oiuWNXDs,11880
+alita_sdk/tools/zephyr_enterprise/api_wrapper.py,sha256=OQJbdcpeVm_5WiTfg-w0wHXFnrKQgHhkufz_nr2wSn4,12270
 alita_sdk/tools/zephyr_enterprise/zephyr_enterprise.py,sha256=hV9LIrYfJT6oYp-ZfQR0YHflqBFPsUw2Oc55HwK0H48,6809
 alita_sdk/tools/zephyr_essential/__init__.py,sha256=2SymL6TIF1ad52w5DTtWaYvNygEvE1ssNNeKx3Y-_Zg,3980
-alita_sdk/tools/zephyr_essential/api_wrapper.py,sha256=PN8KbNBzbsjgkdpyogav1cmwIwu_6dGh0EA2mAm2U68,40685
-alita_sdk/tools/zephyr_essential/client.py,sha256=bfNcUKNqj9MFWTludGbbqD4qZlxrBaC2JtWsCfZMqSY,9722
+alita_sdk/tools/zephyr_essential/api_wrapper.py,sha256=VgihduP4N7x1qlpwYIkIX0ctEqutuqVqAYgrxDFfBu0,47054
+alita_sdk/tools/zephyr_essential/client.py,sha256=fX_n2pACNzDiHxry3F4Xc6baUdw7d9U2m4srbfBv5Eg,9882
 alita_sdk/tools/zephyr_scale/__init__.py,sha256=eetAVRclO1j_N0T3mRnWeLfi3BS98i5FwhNReXO0PlE,4289
-alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=HOt9ShtJI_1tVPcwd3Rwk-VS0SMLqcPNYbN1wqfeuhc,78330
+alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=2M6yFoOmfH5BwxMwTLQO76yD0YdfvccOYHlARJFiqGo,78796
 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.290.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.290.dist-info/METADATA,sha256=_q_iaBu4iOesU-YkYvYP82lYMbQ44yIpHwlDZSEiafw,18897
-alita_sdk-0.3.290.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.290.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.290.dist-info/RECORD,,
+alita_sdk-0.3.292.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.292.dist-info/METADATA,sha256=MDskVNgQodg7ergC1j4qH_nuw0W7bfnV1-sVIxkL8SA,18897
+alita_sdk-0.3.292.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.292.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.292.dist-info/RECORD,,