alita-sdk 0.3.279__tar.gz → 0.3.281__tar.gz

{alita_sdk-0.3.279 → alita_sdk-0.3.281}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.279
+Version: 0.3.281
 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.279 → alita_sdk-0.3.281}/alita_sdk/configurations/browser.py

@@ -15,4 +15,4 @@ class BrowserConfiguration(BaseModel):
         }
     )
     google_cse_id: str = Field(description="Google CSE id", default=None)
-    google_api_key: SecretStr = Field(description="Google API key", default=None, json_schema_extra={'secret': True})
+    google_api_key: SecretStr = Field(description="Google API key", json_schema_extra={'secret': True})

{alita_sdk-0.3.279 → alita_sdk-0.3.281}/alita_sdk/configurations/qtest.py

@@ -15,5 +15,5 @@ class QtestConfiguration(BaseModel):
         }
     )
     base_url: str = Field(description="QTest base url")
-    qtest_api_token: SecretStr = Field(description="QTest API token", default=None)
+    qtest_api_token: SecretStr = Field(description="QTest API token")
 

{alita_sdk-0.3.279 → alita_sdk-0.3.281}/alita_sdk/configurations/zephyr_enterprise.py

@@ -10,7 +10,7 @@ class ZephyrEnterpriseConfiguration(BaseModel):
                 "label": "Zephyr Enterprise",
                 "icon_url": "zephyr.svg",
                 "section": "credentials",
-                "type": "zephyr-enterprise",
+                "type": "zephyr_enterprise",
                 "categories": ["test management"],
                 "extra_categories": ["zephyr", "test automation", "test case management", "test planning"],
             }

{alita_sdk-0.3.279 → alita_sdk-0.3.281}/alita_sdk/configurations/zephyr_essential.py

@@ -8,11 +8,11 @@ class ZephyrEssentialConfiguration(BaseModel):
                 "label": "Zephyr Essential",
                 "icon_url": "zephyr.svg",
                 "section": "credentials",
-                "type": "zephyr-essential",
+                "type": "zephyr_essential",
                 "categories": ["test management"],
                 "extra_categories": ["zephyr", "test automation", "test case management", "test planning"],
             }
         }
     )
-    base_url: str = Field(description="Zephyr Essential Base URL", default=None)
-    token: SecretStr = Field(description="Zephyr Essential API Token", default=None)
+    base_url: str = Field(description="Zephyr Essential Base URL")
+    token: SecretStr = Field(description="Zephyr Essential API Token")

{alita_sdk-0.3.279 → alita_sdk-0.3.281}/alita_sdk/runtime/tools/vectorstore.py

@@ -283,6 +283,7 @@ class VectorStoreWrapper(BaseToolApiWrapper):
 
         for document in documents:
             key = key_fn(document)
+            key = key if isinstance(key, str) else str(key)
             if key in indexed_keys and collection_suffix == indexed_data[key]['metadata'].get('collection'):
                 if compare_fn(document, indexed_data[key]):
                     # Disabled addition of new collection to already indexed documents

{alita_sdk-0.3.279 → alita_sdk-0.3.281}/alita_sdk/tools/__init__.py

@@ -67,7 +67,7 @@ _safe_import_tool('aws', 'cloud.aws', None, 'AWSToolkit')
 _safe_import_tool('azure', 'cloud.azure', None, 'AzureToolkit')
 _safe_import_tool('gcp', 'cloud.gcp', None, 'GCPToolkit')
 _safe_import_tool('k8s', 'cloud.k8s', None, 'KubernetesToolkit')
-_safe_import_tool('custom_open_api', 'custom_open_api', None, 'OpenApiToolkit')
+# _safe_import_tool('custom_open_api', 'custom_open_api', None, 'OpenApiToolkit')
 _safe_import_tool('elastic', 'elastic', None, 'ElasticToolkit')
 _safe_import_tool('keycloak', 'keycloak', None, 'KeycloakToolkit')
 _safe_import_tool('localgit', 'localgit', None, 'AlitaLocalGitToolkit')

{alita_sdk-0.3.279 → alita_sdk-0.3.281}/alita_sdk/tools/ado/test_plan/test_plan_wrapper.py

@@ -363,8 +363,20 @@ class TestPlanApiWrapper(NonCodeIndexerToolkit):
             logger.error(f"Error getting test cases: {e}")
             return ToolException(f"Error getting test cases: {e}")
 
-    def _base_loader(self, plan_id: str, suite_ids: Optional[list[str]] = [], chunking_tool: str = None, **kwargs) -> Generator[Document, None, None]:
+    def get_suites_in_plan(self, plan_id: int) -> List[dict]:
+        """Get all test suites in a test plan."""
+        try:
+            test_suites = self._client.get_test_suites_for_plan(self.project, plan_id)
+            return [suite.as_dict() for suite in test_suites]
+        except Exception as e:
+            logger.error(f"Error getting test suites: {e}")
+            return ToolException(f"Error getting test suites: {e}")
+
+    def _base_loader(self, plan_id: int, suite_ids: Optional[List[int]] = [], chunking_tool: str = None, **kwargs) -> Generator[Document, None, None]:
         cases = []
+        if not suite_ids:
+            suites = self.get_suites_in_plan(plan_id)
+            suite_ids = [suite['id'] for suite in suites if 'id' in suite]
         for sid in suite_ids:
             cases.extend(self.get_test_cases(plan_id, sid))
         #
@@ -381,6 +393,7 @@ class TestPlanApiWrapper(NonCodeIndexerToolkit):
                         'suite_id': case.get('test_suite', {}).get('id', ''),
                         'description': data.get('System.Description', ''),
                         'updated_on': data.get('System.Rev', ''),
+                        # content is in metadata for chunking tool post-processing
                         IndexerKeywords.CONTENT_IN_BYTES.value: data.get('Microsoft.VSTS.TCM.Steps', '').encode("utf-8")
                     })
             else:
@@ -398,8 +411,10 @@ class TestPlanApiWrapper(NonCodeIndexerToolkit):
     def _index_tool_params(self):
         """Return the parameters for indexing data."""
         return {
-            "plan_id": (str, Field(description="ID of the test plan for which test cases are requested")),
-            "suite_ids": (Optional[List[str]], Field(description='List of test suite IDs for which test cases are requested (can be empty). Example: ["2", "23"]', default=[])),
+            "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'))
         }
 

{alita_sdk-0.3.279 → alita_sdk-0.3.281}/alita_sdk/tools/base_indexer_toolkit.py

@@ -240,6 +240,7 @@ class BaseIndexerToolkit(VectorStoreWrapperBase):
 
         for document in documents:
             key = self.key_fn(document)
+            key = key if isinstance(key, str) else str(key)
             if key in indexed_keys and collection_suffix == indexed_data[key]['metadata'].get('collection'):
                 if self.compare_fn(document, indexed_data[key]):
                     continue

{alita_sdk-0.3.279 → alita_sdk-0.3.281}/alita_sdk/tools/xray/__init__.py

@@ -21,6 +21,7 @@ def get_tools(tool):
         limit=tool['settings'].get('limit', 20),
         verify_ssl=tool['settings'].get('verify_ssl', True),
         toolkit_name=tool.get('toolkit_name'),
+        llm=tool['settings'].get('llm', None),
         alita=tool['settings'].get('alita', None),
 
         # indexer settings

{alita_sdk-0.3.279 → alita_sdk-0.3.281}/alita_sdk/tools/zephyr_enterprise/__init__.py

@@ -36,7 +36,7 @@ class ZephyrEnterpriseToolkit(BaseToolkit):
         ZephyrEnterpriseToolkit.toolkit_max_length = get_max_toolkit_length(selected_tools)
         return create_model(
             name,
-            zephyr_configuration=(ZephyrEnterpriseConfiguration, Field(description="Zephyr Configuration", json_schema_extra={'configuration_types': ['zephyr-enterprise']})),
+            zephyr_configuration=(ZephyrEnterpriseConfiguration, Field(description="Zephyr Configuration", json_schema_extra={'configuration_types': ['zephyr_enterprise']})),
             pgvector_configuration=(Optional[PgVectorConfiguration], Field(description="PgVector Configuration",
                                                                            json_schema_extra={
                                                                                'configuration_types': ['pgvector']},

{alita_sdk-0.3.279 → alita_sdk-0.3.281}/alita_sdk/tools/zephyr_essential/__init__.py

@@ -36,7 +36,7 @@ class ZephyrEssentialToolkit(BaseToolkit):
         ZephyrEssentialToolkit.toolkit_max_length = get_max_toolkit_length(selected_tools)
         return create_model(
             name,
-            zephyr_essential_configuration=(ZephyrEssentialConfiguration, Field(description="Zephyr Essential Configuration", json_schema_extra={'configuration_types': ['zephyr-essential']})),
+            zephyr_essential_configuration=(ZephyrEssentialConfiguration, Field(description="Zephyr Essential Configuration", json_schema_extra={'configuration_types': ['zephyr_essential']})),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
             pgvector_configuration=(Optional[PgVectorConfiguration], Field(default=None,
                                                                            description="PgVector Configuration",

{alita_sdk-0.3.279 → alita_sdk-0.3.281}/alita_sdk.egg-info/PKG-INFO

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

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