alita-sdk 0.3.348rc1__py3-none-any.whl → 0.3.348rc3__py3-none-any.whl

alita_sdk/tools/figma/api_wrapper.py

@@ -345,7 +345,8 @@ class FigmaApiWrapper(NonCodeIndexerToolkit):
                             text_nodes[node['id']] = self.get_texts_recursive(node)
         # process image nodes
         if image_nodes:
-            images = self._client.get_file_images(file_key, image_nodes).images or {}
+            file_images = self._client.get_file_images(file_key, image_nodes)
+            images = self._client.get_file_images(file_key, image_nodes).images or {} if file_images else {}
             total_images = len(images)
             if total_images == 0:
                 logging.info(f"No images found for file {file_key}.")

alita_sdk/tools/qtest/__init__.py

@@ -17,8 +17,10 @@ def get_tools(tool):
     toolkit = QtestToolkit.get_toolkit(
         selected_tools=tool['settings'].get('selected_tools', []),
         qtest_project_id=tool['settings'].get('qtest_project_id', tool['settings'].get('project_id', None)),
+        no_of_tests_shown_in_dql_search=tool['settings'].get('no_of_tests_shown_in_dql_search'),
         qtest_configuration=tool['settings']['qtest_configuration'],
-        toolkit_name=tool.get('toolkit_name')
+        toolkit_name=tool.get('toolkit_name'),
+        llm=tool['settings'].get('llm', None)
     )
     return toolkit.tools
 
@@ -37,7 +39,10 @@ class QtestToolkit(BaseToolkit):
                 'configuration_types': ['qtest']})),
             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)]],
+            no_of_tests_shown_in_dql_search=(Optional[int], Field(description="Max number of items returned by dql search",
+                                                                  default=10)),
+
+        selected_tools=(List[Literal[tuple(selected_tools)]],
                             Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
             __config__=ConfigDict(json_schema_extra={'metadata': {"label": "QTest", "icon_url": "qtest.svg",
                                                                   "categories": ["test management"],

alita_sdk/tools/qtest/api_wrapper.py

@@ -1,16 +1,19 @@
+import base64
 import json
 import logging
+import re
 from traceback import format_exc
-from typing import Any
+from typing import Any, Optional
 
 import swagger_client
 from langchain_core.tools import ToolException
 from pydantic import Field, PrivateAttr, model_validator, create_model, SecretStr
 from sklearn.feature_extraction.text import strip_tags
-from swagger_client import TestCaseApi, SearchApi, PropertyResource
+from swagger_client import TestCaseApi, SearchApi, PropertyResource, ModuleApi
 from swagger_client.rest import ApiException
 
 from ..elitea_base import BaseToolApiWrapper
+from ..utils.content_parser import parse_file_content
 
 QTEST_ID = "QTest Id"
 
@@ -64,7 +67,10 @@ logger = logging.getLogger(__name__)
 
 QtestDataQuerySearch = create_model(
     "QtestDataQuerySearch",
-    dql=(str, Field(description="Qtest Data Query Language (DQL) query string")))
+    dql=(str, Field(description="Qtest Data Query Language (DQL) query string")),
+    extract_images=(Optional[bool], Field(description="Should images be processed by llm", default=False)),
+    prompt=(Optional[str], Field(description="Prompt for image processing", default=None))
+)
 
 QtestCreateTestCase = create_model(
     "QtestCreateTestCase",
@@ -92,6 +98,8 @@ UpdateTestCase = create_model(
 FindTestCaseById = create_model(
     "FindTestCaseById",
     test_id=(str, Field(description="Test case ID e.g. TC-1234")),
+    extract_images=(Optional[bool], Field(description="Should images be processed by llm", default=False)),
+    prompt=(Optional[str], Field(description="Prompt for image processing", default=None))
 )
 
 DeleteTestCase = create_model(
@@ -99,6 +107,17 @@ DeleteTestCase = create_model(
     qtest_id=(int, Field(description="Qtest id e.g. 3253490123")),
 )
 
+GetModules = create_model(
+    "GetModules",
+    parent_id=(Optional[int],
+               Field(description="ID of the parent Module. Leave it blank to retrieve Modules under root",
+                                    default=None)),
+    search=(Optional[str],
+               Field(description="The free-text to search for Modules by names. You can utilize this parameter to search for Modules. Leave it blank to retrieve all Modules under root or the parent Module",
+                     default=None)),
+
+)
+
 class QtestApiWrapper(BaseToolApiWrapper):
     base_url: str
     qtest_project_id: int
@@ -107,6 +126,7 @@ class QtestApiWrapper(BaseToolApiWrapper):
     page: int = 1
     no_of_tests_shown_in_dql_search: int = 10
     _client: Any = PrivateAttr()
+    llm: Any
 
     @model_validator(mode='before')
     @classmethod
@@ -140,6 +160,9 @@ class QtestApiWrapper(BaseToolApiWrapper):
         # Instantiate the TestCaseApi instance according to the qtest api documentation and swagger client
         return swagger_client.TestCaseApi(self._client)
 
+    def __instantiate_module_api_instance(self) -> ModuleApi:
+        return swagger_client.ModuleApi(self._client)
+
     def __build_body_for_create_test_case(self, test_cases_data: list[dict],
                                           folder_to_place_test_cases_to: str = '') -> list:
         initial_project_properties = self.__get_properties_form_project()
@@ -220,7 +243,7 @@ class QtestApiWrapper(BaseToolApiWrapper):
             raise ToolException(
                 f"Unable to create test case in project - {self.qtest_project_id} with the following content:\n{test_case_content}.\n\n Stacktrace was {stacktrace}") from e
 
-    def __parse_data(self, response_to_parse: dict, parsed_data: list):
+    def __parse_data(self, response_to_parse: dict, parsed_data: list, extract_images: bool=False, prompt: str=None):
         import html
         for item in response_to_parse['items']:
             parsed_data_row = {
@@ -231,8 +254,8 @@ class QtestApiWrapper(BaseToolApiWrapper):
                 QTEST_ID: item['id'],
                 'Steps': list(map(lambda step: {
                     'Test Step Number': step[0] + 1,
-                    'Test Step Description': step[1]['description'],
-                    'Test Step Expected Result': step[1]['expected']
+                    'Test Step Description': self._process_image(step[1]['description'], extract_images, prompt),
+                    'Test Step Expected Result':  self._process_image(step[1]['expected'], extract_images, prompt)
                 }, enumerate(item['test_steps']))),
                 'Status': ''.join([properties['field_value_name'] for properties in item['properties']
                                    if properties['field_name'] == 'Status']),
@@ -245,7 +268,27 @@ class QtestApiWrapper(BaseToolApiWrapper):
             }
             parsed_data.append(parsed_data_row)
 
-    def __perform_search_by_dql(self, dql: str) -> list:
+    def _process_image(self, content: str, extract: bool=False, prompt: str=None):
+        #extract image by regex
+        img_regex = r'<img\s+src="data:image\/[^;]+;base64,([^"]+)"\s+[^>]*data-filename="([^"]+)"[^>]*>'
+
+        def replace_image(match):
+            base64_content = match.group(1)
+            file_name = match.group(2)
+
+            file_content = base64.b64decode(base64_content)
+
+            if extract:
+                description = f"<img description=\"{parse_file_content(file_content=file_content, file_name=file_name, prompt=prompt, llm=self.llm)}\">"
+            else:
+                description = ""
+
+            return description
+        #replace image tag by description
+        content = re.sub(img_regex, replace_image, content)
+        return content
+
+    def __perform_search_by_dql(self, dql: str, extract_images:bool=False, prompt: str=None) -> list:
         search_instance: SearchApi = swagger_client.SearchApi(self._client)
         body = swagger_client.ArtifactSearchParams(object_type='test-cases', fields=['*'],
                                                    query=dql)
@@ -256,7 +299,7 @@ class QtestApiWrapper(BaseToolApiWrapper):
             api_response = search_instance.search_artifact(self.qtest_project_id, body, append_test_steps=append_test_steps,
                                                            include_external_properties=include_external_properties,
                                                            page_size=self.no_of_items_per_page, page=self.page)
-            self.__parse_data(api_response, parsed_data)
+            self.__parse_data(api_response, parsed_data, extract_images, prompt)
 
             if api_response['links']:
                 while api_response['links'][0]['rel'] == 'next':
@@ -265,7 +308,7 @@ class QtestApiWrapper(BaseToolApiWrapper):
                                                                    append_test_steps=append_test_steps,
                                                                    include_external_properties=include_external_properties,
                                                                    page_size=self.no_of_items_per_page, page=next_page)
-                    self.__parse_data(api_response, parsed_data)
+                    self.__parse_data(api_response, parsed_data, extract_images, prompt)
         except ApiException as e:
             stacktrace = format_exc()
             logger.error(f"Exception when calling SearchApi->search_artifact: \n {stacktrace}")
@@ -333,9 +376,9 @@ class QtestApiWrapper(BaseToolApiWrapper):
             logger.error(f"Error: {format_exc()}")
             raise e
 
-    def search_by_dql(self, dql: str):
+    def search_by_dql(self, dql: str, extract_images:bool=False, prompt: str=None):
         """Search for the test cases in qTest using Data Query Language """
-        parsed_data = self.__perform_search_by_dql(dql)
+        parsed_data = self.__perform_search_by_dql(dql, extract_images, prompt)
         return "Found " + str(
             len(parsed_data)) + f" Qtest test cases:\n" + str(parsed_data[:self.no_of_tests_shown_in_dql_search])
 
@@ -384,10 +427,10 @@ class QtestApiWrapper(BaseToolApiWrapper):
             raise ToolException(
                 f"""Unable to update test case in project with id - {self.qtest_project_id} and test id - {test_id}.\n Exception: \n {stacktrace}""") from e
 
-    def find_test_case_by_id(self, test_id: str) -> str:
+    def find_test_case_by_id(self, test_id: str, extract_images=False, prompt=None) -> str:
         """ Find the test case by its id. Id should be in format TC-123. """
         dql: str = f"Id = '{test_id}'"
-        return f"{self.search_by_dql(dql=dql)}"
+        return f"{self.search_by_dql(dql=dql, extract_images=extract_images, prompt=prompt)}"
 
     def delete_test_case(self, qtest_id: int) -> str:
         """ Delete the test case by its id. Id should be in format 3534653120. """
@@ -401,6 +444,20 @@ class QtestApiWrapper(BaseToolApiWrapper):
             raise ToolException(
                 f"""Unable to delete test case in project with id - {self.qtest_project_id} and qtest_id - {qtest_id}. \n Exception: \n {stacktrace}""") from e
 
+    def get_modules(self, parent_id: int = None, search: str = None):
+        """
+        :param int project_id: ID of the project (required)
+        :param int parent_id: ID of the parent Module. Leave it blank to retrieve Modules under root
+        :param str search: The free-text to search for Modules by names. You can utilize this parameter to search for Modules. Leave it blank to retrieve all Modules under root or the parent Module
+        """
+        module_api = self.__instantiate_module_api_instance()
+        kwargs = {}
+        if parent_id:
+            kwargs["parent_id"] = parent_id
+        if search:
+            kwargs["search"] = search
+        return module_api.get_sub_modules_of(project_id=self.qtest_project_id, **kwargs)
+
     def get_available_tools(self):
         return [
             {
@@ -444,5 +501,12 @@ class QtestApiWrapper(BaseToolApiWrapper):
                 "description": """Link tests to Jira requirements. The input is jira issue id and th list of test ids in format '["TC-123", "TC-234", "TC-345"]'""",
                 "args_schema": QtestLinkTestCaseToJiraRequirement,
                 "ref": self.link_tests_to_jira_requirement,
+            },
+            {
+                "name": "get_modules",
+                "mode": "get_modules",
+                "description": self.get_modules.__doc__,
+                "args_schema": GetModules,
+                "ref": self.get_modules,
             }
         ]

{alita_sdk-0.3.348rc1.dist-info → alita_sdk-0.3.348rc3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.348rc1
+Version: 0.3.348rc3
 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.348rc1.dist-info → alita_sdk-0.3.348rc3.dist-info}/RECORD

@@ -236,7 +236,7 @@ alita_sdk/tools/custom_open_api/api_wrapper.py,sha256=sDSFpvEqpSvXHGiBISdQQcUecf
 alita_sdk/tools/elastic/__init__.py,sha256=iwnSRppRpzvJ1da2K3Glu8Uu41MhBDCYbguboLkEbW0,2818
 alita_sdk/tools/elastic/api_wrapper.py,sha256=pl8CqQxteJAGwyOhMcld-ZgtOTFwwbv42OITQVe8rM0,1948
 alita_sdk/tools/figma/__init__.py,sha256=W6vIMMkZI2Lmpg6_CRRV3oadaIbVI-qTLmKUh6enqWs,4509
-alita_sdk/tools/figma/api_wrapper.py,sha256=yK45guP6oMStTpfNLXRYgIZtNWkuWzgjFm_Vzu-ivNg,33687
+alita_sdk/tools/figma/api_wrapper.py,sha256=k5V2ufv9k3LL5dwjSVi2qIZdx6H0613mlXId3gQudMw,33788
 alita_sdk/tools/github/__init__.py,sha256=2rHu0zZyZGnLC5CkHgDIhe14N9yCyaEfrrt7ydH8478,5191
 alita_sdk/tools/github/api_wrapper.py,sha256=uDwYckdnpYRJtb0uZnDkaz2udvdDLVxuCh1tSwspsiU,8411
 alita_sdk/tools/github/github_client.py,sha256=0YkpD6Zm4X46jMNN57ZIypo2YObtgxCGQokJAF-laFs,86597
@@ -299,8 +299,8 @@ 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=bnD6rDM9dS0vpgaUmza4z67WLcZVhJk7S9ZG8OPij0Q,4116
-alita_sdk/tools/qtest/api_wrapper.py,sha256=cWXpmjjel9CYIXXjetJkARLYZXqvHufSghctTHN0ggc,22296
+alita_sdk/tools/qtest/__init__.py,sha256=Jf0xo5S_4clXR2TfCbJbB1sFgCbcFQRM-YYX2ltWBzo,4461
+alita_sdk/tools/qtest/api_wrapper.py,sha256=jQSnmYUXyqZyRImzOaxJ7TjrvPwxUw-AWJk3UErwHcw,25597
 alita_sdk/tools/qtest/tool.py,sha256=kKzNPS4fUC76WQQttQ6kdVANViHEvKE8Kf174MQiNYU,562
 alita_sdk/tools/rally/__init__.py,sha256=2BPPXJxAOKgfmaxVFVvxndfK0JxOXDLkoRmzu2dUwOE,3512
 alita_sdk/tools/rally/api_wrapper.py,sha256=mouzU6g0KML4UNapdk0k6Q0pU3MpJuWnNo71n9PSEHM,11752
@@ -350,8 +350,8 @@ alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=kT0TbmMvuKhDUZc0i7KO18O38JM9S
 alita_sdk/tools/zephyr_squad/__init__.py,sha256=0ne8XLJEQSLOWfzd2HdnqOYmQlUliKHbBED5kW_Vias,2895
 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.348rc1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.348rc1.dist-info/METADATA,sha256=EoQDDMvqLF-l2fQt_maEM9xNEg8YSyOTM0zW6fXDT6s,19074
-alita_sdk-0.3.348rc1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.348rc1.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.348rc1.dist-info/RECORD,,
+alita_sdk-0.3.348rc3.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.348rc3.dist-info/METADATA,sha256=5aAIwbbXXLrqYverX8ZY63HRvrWbXeK7tPm7BImltU0,19074
+alita_sdk-0.3.348rc3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.348rc3.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.348rc3.dist-info/RECORD,,