alita-sdk 0.3.247__tar.gz → 0.3.249__tar.gz

{alita_sdk-0.3.247 → alita_sdk-0.3.249}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.247
+Version: 0.3.249
 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.247 → alita_sdk-0.3.249}/alita_sdk/configurations/__init__.py

@@ -40,6 +40,7 @@ _safe_import_configuration('delta_lake', 'delta_lake', 'DeltaLakeConfiguration')
 _safe_import_configuration('bigquery', 'bigquery', 'BigQueryConfiguration')
 _safe_import_configuration('xray', 'xray', 'XrayConfiguration')
 _safe_import_configuration('zephyr', 'zephyr', 'ZephyrConfiguration')
+_safe_import_configuration('zephyr_enterprise', 'zephyr_enterprise', 'ZephyrEnterpriseConfiguration')
 _safe_import_configuration('figma', 'figma', 'FigmaConfiguration')
 _safe_import_configuration('rally', 'rally', 'RallyConfiguration')
 

{alita_sdk-0.3.247 → alita_sdk-0.3.249}/alita_sdk/configurations/zephyr.py

@@ -7,7 +7,7 @@ class ZephyrConfiguration(BaseModel):
     model_config = ConfigDict(
         json_schema_extra={
             "metadata": {
-                "label": "Zephyr",
+                "label": "Zephyr Scale",
                 "icon_url": "zephyr.svg",
                 "sections": {
                     "auth": {

{alita_sdk-0.3.247 → alita_sdk-0.3.249}/alita_sdk/configurations/zephyr_enterprise.py

@@ -7,7 +7,7 @@ class ZephyrEnterpriseConfiguration(BaseModel):
     model_config = ConfigDict(
         json_schema_extra={
             "metadata": {
-                "label": "Zephyr",
+                "label": "Zephyr Enterprise",
                 "icon_url": "zephyr.svg",
                 "section": "credentials",
                 "type": "zephyr-enterprise",

{alita_sdk-0.3.247 → alita_sdk-0.3.249}/alita_sdk/runtime/toolkits/artifact.py

@@ -28,7 +28,7 @@ class ArtifactToolkit(BaseToolkit):
 
             # embedding model settings
             embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.",
-                                                  json_schema_extra={'configuration_types': ['embedding_model']})),
+                                                  json_schema_extra={'configuration_model': 'embedding'})),
 
             __config__=ConfigDict(json_schema_extra={'metadata': {"label": "Artifact", "icon_url": None}})
         )

{alita_sdk-0.3.247 → alita_sdk-0.3.249}/alita_sdk/runtime/tools/vectorstore.py

@@ -585,7 +585,8 @@ class VectorStoreWrapper(BaseToolApiWrapper):
         
         # Apply cutoff filter
         if cut_off:
-            combined_items = [item for item in combined_items if abs(item[1]) >= cut_off]
+            # Filter out items above the cutoff score (since the lower the score, the better)
+            combined_items = [item for item in combined_items if abs(item[1]) <= cut_off]
         
         # Sort by score and limit results
         # DISABLED: for chroma we want ascending order (lower score is better), for others descending

{alita_sdk-0.3.247 → alita_sdk-0.3.249}/alita_sdk/tools/ado/repos/__init__.py

@@ -53,7 +53,7 @@ class AzureDevOpsReposToolkit(BaseToolkit):
             # indexer settings
             pgvector_configuration=(Optional[PgVectorConfiguration], Field(default=None, description="PgVector Configuration", json_schema_extra={'configuration_types': ['pgvector']})),
             # embedder settings
-            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_types': ['embedding_model']})),
+            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_model': 'embedding'})),
 
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
             __config__={'json_schema_extra': {'metadata':

{alita_sdk-0.3.247 → alita_sdk-0.3.249}/alita_sdk/tools/ado/test_plan/__init__.py

@@ -32,7 +32,7 @@ class AzureDevOpsPlansToolkit(BaseToolkit):
             pgvector_configuration=(Optional[PgVectorConfiguration], Field(default=None,
                                                                            description="PgVector Configuration", json_schema_extra={'configuration_types': ['pgvector']})),
             # embedder settings
-            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_types': ['embedding_model']})),
+            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_model': 'embedding'})),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
             __config__={'json_schema_extra': {'metadata':
                 {

{alita_sdk-0.3.247 → alita_sdk-0.3.249}/alita_sdk/tools/ado/wiki/__init__.py

@@ -32,7 +32,7 @@ class AzureDevOpsWikiToolkit(BaseToolkit):
             pgvector_configuration=(Optional[PgVectorConfiguration], Field(default=None,
                                                                            description="PgVector Configuration", json_schema_extra={'configuration_types': ['pgvector']})),
             # embedder settings
-            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_types': ['embedding_model']})),
+            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_model': 'embedding'})),
             selected_tools=(List[Literal[tuple(selected_tools)]],
                             Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
             __config__={

{alita_sdk-0.3.247 → alita_sdk-0.3.249}/alita_sdk/tools/ado/work_item/__init__.py

@@ -34,7 +34,7 @@ class AzureDevOpsWorkItemsToolkit(BaseToolkit):
                                                                            description="PgVector Configuration",
                                                                            json_schema_extra={'configuration_types': ['pgvector']})),
             # embedder settings
-            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_types': ['embedding_model']})),
+            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_model': 'embedding'})),
             __config__={
                 'json_schema_extra': {
                     'metadata': {

{alita_sdk-0.3.247 → alita_sdk-0.3.249}/alita_sdk/tools/bitbucket/__init__.py

@@ -55,7 +55,7 @@ class AlitaBitbucketToolkit(BaseToolkit):
             bitbucket_configuration=(Optional[BitbucketConfiguration], Field(description="Bitbucket Configuration", json_schema_extra={'configuration_types': ['bitbucket']})),
             pgvector_configuration=(Optional[PgVectorConfiguration], Field(default=None, description="PgVector Configuration", json_schema_extra={'configuration_types': ['pgvector']})),
             # embedder settings
-            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_types': ['embedding_model']})),
+            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_model': 'embedding'})),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
             __config__=ConfigDict(json_schema_extra=
             {

{alita_sdk-0.3.247 → alita_sdk-0.3.249}/alita_sdk/tools/confluence/__init__.py

@@ -83,7 +83,7 @@ class ConfluenceToolkit(BaseToolkit):
                                                                            description="PgVector Configuration",
                                                                            json_schema_extra={'configuration_types': ['pgvector']})),
             # embedder settings
-            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_types': ['embedding_model']})),
+            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_model': 'embedding'})),
 
             selected_tools=(List[Literal[tuple(selected_tools)]],
                             Field(default=[], json_schema_extra={'args_schemas': selected_tools})),

{alita_sdk-0.3.247 → alita_sdk-0.3.249}/alita_sdk/tools/elitea_base.py

@@ -369,6 +369,16 @@ class BaseVectorStoreToolApiWrapper(BaseToolApiWrapper):
         vectorstore_wrapper = self._init_vector_store()
         return vectorstore_wrapper.list_collections()
 
+    def _build_collection_filter(self, filter: dict | str, collection_suffix: str = "") -> dict:
+        """Builds a filter for the collection based on the provided suffix."""
+
+        filter = filter if isinstance(filter, dict) else json.loads(filter)
+        if collection_suffix:
+            filter.update({"collection": {
+                "$eq": collection_suffix.strip()
+            }})
+        return filter
+
     def search_index(self,
                      query: str,
                      collection_suffix: str = "",
@@ -380,13 +390,7 @@ class BaseVectorStoreToolApiWrapper(BaseToolApiWrapper):
                      **kwargs):
         """ Searches indexed documents in the vector store."""
         vectorstore = self._init_vector_store()
-        # build filter on top of collection_suffix
-        filter = filter if isinstance(filter, dict) else json.loads(filter)
-        if collection_suffix:
-            filter.update({"collection": {
-                "$eq": collection_suffix.strip()
-            }})
-
+        filter = self._build_collection_filter(filter, collection_suffix)
         found_docs = vectorstore.search_documents(
             query,
             doctype=self.doctype,
@@ -411,6 +415,8 @@ class BaseVectorStoreToolApiWrapper(BaseToolApiWrapper):
                      extended_search: Optional[List[str]] = None,
                      **kwargs):
         """ Searches indexed documents in the vector store."""
+
+        filter = self._build_collection_filter(filter, collection_suffix)
         vectorstore = self._init_vector_store()
         found_docs = vectorstore.stepback_search(
             query,
@@ -423,7 +429,7 @@ class BaseVectorStoreToolApiWrapper(BaseToolApiWrapper):
             reranking_config=reranking_config,
             extended_search=extended_search
         )
-        return f"Found {len(found_docs)} documents matching the query\n{json.dumps(found_docs, indent=4)}" if found_docs else "No documents found matching the query."
+        return found_docs if found_docs else f"No documents found by query '{query}' and filter '{filter}'"
 
     def stepback_summary_index(self,
                      query: str,
@@ -437,17 +443,20 @@ class BaseVectorStoreToolApiWrapper(BaseToolApiWrapper):
                      **kwargs):
         """ Generates a summary of indexed documents using stepback technique."""
         vectorstore = self._init_vector_store()
-        return vectorstore.stepback_summary(
-            query, 
-            messages, 
-            self.doctype, 
-            filter=filter, 
-            cut_off=cut_off, 
+        filter = self._build_collection_filter(filter, collection_suffix)
+
+        found_docs = vectorstore.stepback_summary(
+            query,
+            messages,
+            self.doctype,
+            filter=filter,
+            cut_off=cut_off,
             search_top=search_top,
-            full_text_search=full_text_search, 
-            reranking_config=reranking_config, 
+            full_text_search=full_text_search,
+            reranking_config=reranking_config,
             extended_search=extended_search
         )
+        return found_docs if found_docs else f"No documents found by query '{query}' and filter '{filter}'"
 
     def _get_vector_search_tools(self):
         """

{alita_sdk-0.3.247 → alita_sdk-0.3.249}/alita_sdk/tools/figma/__init__.py

@@ -59,7 +59,7 @@ class FigmaToolkit(BaseToolkit):
             pgvector_configuration=(Optional[PgVectorConfiguration], Field(description="PgVector Configuration", json_schema_extra={'configuration_types': ['pgvector']})),
 
             # embedder settings
-            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_types': ['embedding_model']})),
+            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_model': 'embedding'})),
             __config__=ConfigDict(
                 json_schema_extra={
                      "metadata": {

{alita_sdk-0.3.247 → alita_sdk-0.3.249}/alita_sdk/tools/github/__init__.py

@@ -66,7 +66,7 @@ class AlitaGitHubToolkit(BaseToolkit):
             active_branch=(Optional[str], Field(description="Active branch", default="main")),
             base_branch=(Optional[str], Field(description="Github Base branch", default="main")),
             # embedder settings
-            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_types': ['embedding_model']})),
+            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_model': 'embedding'})),
             selected_tools=(List[Literal[tuple(selected_tools)]],
                             Field(default=[], json_schema_extra={'args_schemas': selected_tools}))
         )

{alita_sdk-0.3.247 → alita_sdk-0.3.249}/alita_sdk/tools/gitlab/__init__.py

@@ -51,7 +51,7 @@ class AlitaGitlabToolkit(BaseToolkit):
                                                                            description="PgVector Configuration", json_schema_extra={'configuration_types': ['pgvector']})),
             # embedder settings
             embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.",
-                                                  json_schema_extra={'configuration_types': ['embedding_model']})),
+                                                  json_schema_extra={'configuration_model': 'embedding'})),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
             __config__=ConfigDict(json_schema_extra={
                 'metadata': {

{alita_sdk-0.3.247 → alita_sdk-0.3.249}/alita_sdk/tools/jira/__init__.py

@@ -79,7 +79,7 @@ class JiraToolkit(BaseToolkit):
             pgvector_configuration=(Optional[PgVectorConfiguration], Field(default=None,
                                                                            description="PgVector Configuration", json_schema_extra={'configuration_types': ['pgvector']})),
             # embedder settings
-            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_types': ['embedding_model']})),
+            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_model': 'embedding'})),
 
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
             __config__=ConfigDict(json_schema_extra={

{alita_sdk-0.3.247 → alita_sdk-0.3.249}/alita_sdk/tools/sharepoint/__init__.py

@@ -46,7 +46,7 @@ class SharepointToolkit(BaseToolkit):
                                                                            description="PgVector Configuration",
                                                                            json_schema_extra={'configuration_types': ['pgvector']})),
             # embedder settings
-            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_types': ['embedding_model']})),
+            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_model': 'embedding'})),
             __config__=ConfigDict(json_schema_extra={
                 'metadata': {
                     "label": "Sharepoint", "icon_url": "sharepoint.svg",

{alita_sdk-0.3.247 → alita_sdk-0.3.249}/alita_sdk/tools/testrail/__init__.py

@@ -46,7 +46,7 @@ class TestrailToolkit(BaseToolkit):
             pgvector_configuration=(Optional[PgVectorConfiguration], Field(default = None,
                                                                            description="PgVector Configuration", json_schema_extra={'configuration_types': ['pgvector']})),
             # embedder settings
-            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_types': ['embedding_model']})),
+            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_model': 'embedding'})),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
             __config__=ConfigDict(json_schema_extra={'metadata':
                                                          {"label": "Testrail", "icon_url": "testrail-icon.svg",

{alita_sdk-0.3.247 → alita_sdk-0.3.249}/alita_sdk/tools/xray/__init__.py

@@ -50,7 +50,7 @@ class XrayToolkit(BaseToolkit):
                                                                            json_schema_extra={
                                                                                'configuration_types': ['pgvector']})),
             # embedder settings
-            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_types': ['embedding_model']})),
+            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_model': 'embedding'})),
 
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
             __config__={'json_schema_extra':

{alita_sdk-0.3.247 → alita_sdk-0.3.249}/alita_sdk/tools/zephyr_enterprise/__init__.py

@@ -42,7 +42,7 @@ class ZephyrEnterpriseToolkit(BaseToolkit):
                                                                                'configuration_types': ['pgvector']},
                                                                            default=None)),
             # embedder settings
-            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_types': ['embedding_model']})),
+            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_model': 'embedding'})),
             selected_tools=(List[Literal[tuple(selected_tools)]], []),
             __config__=ConfigDict(json_schema_extra={
                 'metadata': {

{alita_sdk-0.3.247 → alita_sdk-0.3.249}/alita_sdk/tools/zephyr_essential/__init__.py

@@ -43,7 +43,7 @@ class ZephyrEssentialToolkit(BaseToolkit):
                                                                            json_schema_extra={
                                                                                'configuration_types': ['pgvector']})),
             # embedder settings
-            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_types': ['embedding_model']})),
+            embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.", json_schema_extra={'configuration_model': 'embedding'})),
             __config__={'json_schema_extra': {'metadata': {"label": "Zephyr Essential", "icon_url": "zephyr.svg",
                             "categories": ["test management"],
                             "extra_categories": ["test automation", "test case management", "test planning"]

{alita_sdk-0.3.247 → alita_sdk-0.3.249}/alita_sdk/tools/zephyr_scale/__init__.py

@@ -47,7 +47,7 @@ class ZephyrScaleToolkit(BaseToolkit):
                                                                                'configuration_types': ['pgvector']})),
             # embedder settings
             embedding_model=(Optional[str], Field(default=None, description="Embedding configuration.",
-                                                     json_schema_extra={'configuration_types': ['embedding_model']})),
+                                                     json_schema_extra={'configuration_model': 'embedding'})),
             selected_tools=(List[Literal[tuple(selected_tools)]],
                             Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
             __config__={

{alita_sdk-0.3.247 → alita_sdk-0.3.249}/alita_sdk.egg-info/PKG-INFO

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

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