alita-sdk 0.3.242__tar.gz → 0.3.244__tar.gz

{alita_sdk-0.3.242 → alita_sdk-0.3.244}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.242
+Version: 0.3.244
 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.242 → alita_sdk-0.3.244}/alita_sdk/tools/ado/__init__.py

@@ -18,8 +18,7 @@ def get_tools(tool_type, tool):
         "pgvector_configuration": tool['settings'].get('pgvector_configuration', {}),
         "collection_name": tool['toolkit_name'],
         "doctype": 'doc',
-        "embedding_model": "HuggingFaceEmbeddings",
-        "embedding_model_params": {"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
+        "embedding_configuration": tool['settings'].get('embedding_configuration', {}),
         "vectorstore_type": "PGVector"
     }
     if tool_type == 'ado_plans':

{alita_sdk-0.3.242 → alita_sdk-0.3.244}/alita_sdk/tools/ado/repos/__init__.py

@@ -24,10 +24,9 @@ def _get_toolkit(tool) -> BaseToolkit:
         active_branch=tool['settings'].get('active_branch', ""),
         toolkit_name=tool['settings'].get('toolkit_name', ""),
         pgvector_configuration=tool['settings'].get('pgvector_configuration', {}),
+        embedding_configuration=tool['settings'].get('embedding_configuration', {}),
         collection_name=tool['toolkit_name'],
         doctype='code',
-        embedding_model="HuggingFaceEmbeddings",
-        embedding_model_params={"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
         vectorstore_type="PGVector",
     )
 
@@ -110,6 +109,7 @@ class AzureDevOpsReposToolkit(BaseToolkit):
             # TODO use ado_repos_configuration fields
             **kwargs['ado_repos_configuration'],
             **(kwargs.get('pgvector_configuration') or {}),
+            **(kwargs.get('embedding_configuration') or {}),
         }
         azure_devops_repos_wrapper = ReposApiWrapper(**wrapper_payload)
         available_tools = azure_devops_repos_wrapper.get_available_tools()

{alita_sdk-0.3.242 → alita_sdk-0.3.244}/alita_sdk/tools/ado/test_plan/__init__.py

@@ -92,6 +92,7 @@ class AzureDevOpsPlansToolkit(BaseToolkit):
             # TODO use ado_configuration fields in TestPlanApiWrapper
             **kwargs['ado_configuration'],
             **(kwargs.get('pgvector_configuration') or {}),
+            **(kwargs.get('embedding_configuration') or {}),
         }
         azure_devops_api_wrapper = TestPlanApiWrapper(**wrapper_payload)
         available_tools = azure_devops_api_wrapper.get_available_tools()

{alita_sdk-0.3.242 → alita_sdk-0.3.244}/alita_sdk/tools/ado/wiki/__init__.py

@@ -90,6 +90,7 @@ class AzureDevOpsWikiToolkit(BaseToolkit):
             # TODO use ado_configuration fields in AzureDevOpsApiWrapper
             **kwargs['ado_configuration'],
             **(kwargs.get('pgvector_configuration') or {}),
+            **(kwargs.get('embedding_configuration') or {}),
         }
         azure_devops_api_wrapper = AzureDevOpsApiWrapper(**wrapper_payload)
         available_tools = azure_devops_api_wrapper.get_available_tools()

{alita_sdk-0.3.242 → alita_sdk-0.3.244}/alita_sdk/tools/ado/work_item/__init__.py

@@ -90,6 +90,7 @@ class AzureDevOpsWorkItemsToolkit(BaseToolkit):
             # TODO use ado_configuration fields in AzureDevOpsApiWrapper
             **kwargs['ado_configuration'],
             **(kwargs.get('pgvector_configuration') or {}),
+            **(kwargs.get('embedding_configuration') or {}),
         }
         azure_devops_api_wrapper = AzureDevOpsApiWrapper(**wrapper_payload)
         available_tools = azure_devops_api_wrapper.get_available_tools()

{alita_sdk-0.3.242 → alita_sdk-0.3.244}/alita_sdk/tools/bitbucket/__init__.py

@@ -31,8 +31,7 @@ def get_tools(tool):
         pgvector_configuration=tool['settings'].get('pgvector_configuration', {}),
         collection_name=str(tool['toolkit_name']),
         doctype='code',
-        embedding_model="HuggingFaceEmbeddings",
-        embedding_model_params={"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
+        embedding_configuration=tool['settings'].get('embedding_configuration', {}),
         vectorstore_type="PGVector",
         toolkit_name=tool.get('toolkit_name')
     ).get_tools()
@@ -101,6 +100,7 @@ class AlitaBitbucketToolkit(BaseToolkit):
             # TODO use bitbucket_configuration fields
             **kwargs['bitbucket_configuration'],
             **(kwargs.get('pgvector_configuration') or {}),
+            **(kwargs.get('embedding_configuration') or {}),
         }
         bitbucket_api_wrapper = BitbucketAPIWrapper(**wrapper_payload)
         available_tools: List[Dict] = __all__

{alita_sdk-0.3.242 → alita_sdk-0.3.244}/alita_sdk/tools/confluence/__init__.py

@@ -30,8 +30,7 @@ def get_tools(tool):
         pgvector_configuration=tool['settings'].get('pgvector_configuration', {}),
         collection_name=str(tool['toolkit_name']),
         doctype='doc',
-        embedding_model="HuggingFaceEmbeddings",
-        embedding_model_params={"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
+        embedding_configuration=tool['settings'].get('embedding_configuration', {}),
         vectorstore_type="PGVector"
     ).get_tools()
 
@@ -125,6 +124,7 @@ class ConfluenceToolkit(BaseToolkit):
             # TODO use confluence_configuration fields
             **kwargs['confluence_configuration'],
             **(kwargs.get('pgvector_configuration') or {}),
+            **(kwargs.get('embedding_configuration') or {}),
         }
         confluence_api_wrapper = ConfluenceAPIWrapper(**wrapper_payload)
         prefix = clean_string(toolkit_name, ConfluenceToolkit.toolkit_max_length) + TOOLKIT_SPLITTER if toolkit_name else ''

{alita_sdk-0.3.242 → alita_sdk-0.3.244}/alita_sdk/tools/elitea_base.py

@@ -301,7 +301,7 @@ class BaseVectorStoreToolApiWrapper(BaseToolApiWrapper):
         collection_suffix = kwargs.get("collection_suffix")
         progress_step = kwargs.get("progress_step")
         clean_index = kwargs.get("clean_index")
-        vs = self._init_vector_store(embeddings=self._embedding)
+        vs = self._init_vector_store()
         #
         return vs.index_documents(docs, collection_suffix=collection_suffix, progress_step=progress_step, clean_index=clean_index)
 
@@ -337,7 +337,7 @@ class BaseVectorStoreToolApiWrapper(BaseToolApiWrapper):
 
 
     # TODO: init store once and re-use the instance
-    def _init_vector_store(self, embeddings: Optional[Any] = None):
+    def _init_vector_store(self):
         """Initializes the vector store wrapper with the provided parameters."""
         try:
             from alita_sdk.runtime.tools.vectorstore import VectorStoreWrapper
@@ -353,7 +353,7 @@ class BaseVectorStoreToolApiWrapper(BaseToolApiWrapper):
                 embedding_model=self.embedding_model,
                 embedding_model_params=self.embedding_model_params,
                 vectorstore_params=vectorstore_params,
-                embeddings=embeddings,
+                embeddings=self._embedding,
                 process_document_func=self._process_documents,
             )
         return self._vector_store
@@ -411,7 +411,7 @@ class BaseVectorStoreToolApiWrapper(BaseToolApiWrapper):
                      extended_search: Optional[List[str]] = None,
                      **kwargs):
         """ Searches indexed documents in the vector store."""
-        vectorstore = self._init_vector_store(collection_suffix)
+        vectorstore = self._init_vector_store()
         found_docs = vectorstore.stepback_search(
             query,
             messages,
@@ -436,7 +436,7 @@ class BaseVectorStoreToolApiWrapper(BaseToolApiWrapper):
                      extended_search: Optional[List[str]] = None,
                      **kwargs):
         """ Generates a summary of indexed documents using stepback technique."""
-        vectorstore = self._init_vector_store(collection_suffix)
+        vectorstore = self._init_vector_store()
         return vectorstore.stepback_summary(
             query, 
             messages, 

{alita_sdk-0.3.242 → alita_sdk-0.3.244}/alita_sdk/tools/figma/__init__.py

@@ -27,8 +27,7 @@ def get_tools(tool):
             pgvector_configuration=tool['settings'].get('pgvector_configuration', {}),
             collection_name=str(tool['toolkit_name']),
             doctype='doc',
-            embedding_model="HuggingFaceEmbeddings",
-            embedding_model_params={"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
+            embedding_configuration=tool['settings'].get('embedding_configuration', {}),
             vectorstore_type="PGVector"
         )
         .get_tools()
@@ -98,6 +97,7 @@ class FigmaToolkit(BaseToolkit):
         wrapper_payload = {
             **kwargs,
             **(kwargs.get('pgvector_configuration') or {}),
+            **(kwargs.get('embedding_configuration') or {}),
         }
         figma_api_wrapper = FigmaApiWrapper(**wrapper_payload)
         prefix = clean_string(toolkit_name, cls.toolkit_max_length) + TOOLKIT_SPLITTER if toolkit_name else ''

{alita_sdk-0.3.242 → alita_sdk-0.3.244}/alita_sdk/tools/github/__init__.py

@@ -26,8 +26,7 @@ def _get_toolkit(tool) -> BaseToolkit:
         pgvector_configuration=tool['settings'].get('pgvector_configuration', {}),
         collection_name=str(tool['toolkit_name']),
         doctype='code',
-        embedding_model="HuggingFaceEmbeddings",
-        embedding_model_params={"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
+        embedding_configuration=tool['settings'].get('embedding_configuration', {}),
         vectorstore_type="PGVector",
         toolkit_name=tool.get('toolkit_name')
     )
@@ -84,6 +83,7 @@ class AlitaGitHubToolkit(BaseToolkit):
             # TODO use github_configuration fields
             **kwargs['github_configuration'],
             **(kwargs.get('pgvector_configuration') or {}),
+            **(kwargs.get('embedding_configuration') or {}),
         }
         github_api_wrapper = AlitaGitHubAPIWrapper(**wrapper_payload)
         available_tools: List[Dict] = github_api_wrapper.get_available_tools()

{alita_sdk-0.3.242 → alita_sdk-0.3.244}/alita_sdk/tools/gitlab/__init__.py

@@ -28,8 +28,7 @@ def get_tools(tool):
         pgvector_configuration=tool['settings'].get('pgvector_configuration', {}),
         collection_name=str(tool['toolkit_name']),
         doctype='code',
-        embedding_model="HuggingFaceEmbeddings",
-        embedding_model_params={"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
+        embedding_configuration=tool['settings'].get('embedding_configuration', {}),
         vectorstore_type="PGVector",
         toolkit_name=tool.get('toolkit_name')
     ).get_tools()
@@ -85,6 +84,7 @@ class AlitaGitlabToolkit(BaseToolkit):
             # TODO use gitlab_configuration fields
             **kwargs['gitlab_configuration'],
             **(kwargs.get('pgvector_configuration') or {}),
+            **(kwargs.get('embedding_configuration') or {}),
         }
         gitlab_api_wrapper = GitLabAPIWrapper(**wrapper_payload)
         prefix = clean_string(toolkit_name, cls.toolkit_max_length) + TOOLKIT_SPLITTER if toolkit_name else ''

{alita_sdk-0.3.242 → alita_sdk-0.3.244}/alita_sdk/tools/jira/__init__.py

@@ -27,9 +27,8 @@ def get_tools(tool):
         llm=tool['settings'].get('llm', None),
         alita=tool['settings'].get('alita', None),
         pgvector_configuration=tool['settings'].get('pgvector_configuration', {}),
+        embedding_configuration=tool['settings'].get('embedding_configuration', {}),
         collection_name=str(tool['toolkit_name']),
-        embedding_model="HuggingFaceEmbeddings",
-        embedding_model_params={"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
         vectorstore_type="PGVector",
         toolkit_name=tool.get('toolkit_name')
     ).get_tools()
@@ -106,6 +105,7 @@ class JiraToolkit(BaseToolkit):
             # TODO use jira_configuration fields
             **kwargs['jira_configuration'],
             **(kwargs.get('pgvector_configuration') or {}),
+            **(kwargs.get('embedding_configuration') or {}),
         }
         jira_api_wrapper = JiraApiWrapper(**wrapper_payload)
         prefix = clean_string(toolkit_name, cls.toolkit_max_length) + TOOLKIT_SPLITTER if toolkit_name else ''

{alita_sdk-0.3.242 → alita_sdk-0.3.244}/alita_sdk/tools/sharepoint/__init__.py

@@ -5,6 +5,7 @@ from pydantic import create_model, BaseModel, ConfigDict, Field, SecretStr
 from .api_wrapper import SharepointApiWrapper
 from ..base.tool import BaseAction
 from ..utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
+from ...configurations.embedding import EmbeddingConfiguration
 from ...configurations.pgvector import PgVectorConfiguration
 
 name = "sharepoint"
@@ -21,9 +22,8 @@ def get_tools(tool):
         alita=tool['settings'].get('alita', None),
         # indexer settings
         pgvector_configuration=tool['settings'].get('pgvector_configuration', {}),
+        embedding_configuration=tool['settings'].get('embedding_configuration', {}),
         collection_name=str(tool['toolkit_name']),
-        embedding_model="HuggingFaceEmbeddings",
-        embedding_model_params={"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
         vectorstore_type="PGVector")
             .get_tools())
 
@@ -44,10 +44,9 @@ class SharepointToolkit(BaseToolkit):
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
             # indexer settings
             pgvector_configuration=(Optional[PgVectorConfiguration], Field(description="PgVector Configuration", json_schema_extra={'configuration_types': ['pgvector']})),
-
             # embedder settings
-            embedding_model=(str, Field(description="Embedding model: i.e. 'HuggingFaceEmbeddings', etc.", default="HuggingFaceEmbeddings")),
-            embedding_model_params=(dict, Field(description="Embedding model parameters: i.e. `{'model_name': 'sentence-transformers/all-MiniLM-L6-v2'}", default={"model_name": "sentence-transformers/all-MiniLM-L6-v2"})),
+            embedding_configuration=(Optional[EmbeddingConfiguration], Field(description="Embedding configuration.",
+                                                                              json_schema_extra={'configuration_types': ['embedding']})),
             __config__=ConfigDict(json_schema_extra={
                 'metadata': {
                     "label": "Sharepoint", "icon_url": "sharepoint.svg",
@@ -63,6 +62,7 @@ class SharepointToolkit(BaseToolkit):
         wrapper_payload = {
             **kwargs,
             **(kwargs.get('pgvector_configuration') or {}),
+            **(kwargs.get('embedding_configuration') or {}),
         }
         sharepoint_api_wrapper = SharepointApiWrapper(**wrapper_payload)
         prefix = clean_string(toolkit_name, cls.toolkit_max_length) + TOOLKIT_SPLITTER if toolkit_name else ''

{alita_sdk-0.3.242 → alita_sdk-0.3.244}/alita_sdk/tools/testrail/__init__.py

@@ -24,9 +24,8 @@ def get_tools(tool):
 
         # indexer settings
         pgvector_configuration=tool['settings'].get('pgvector_configuration', {}),
+        embedding_configuration=tool['settings'].get('embedding_configuration', {}),
         collection_name=f"{tool.get('toolkit_name')}",
-        embedding_model="HuggingFaceEmbeddings",
-        embedding_model_params={"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
         vectorstore_type="PGVector"
     ).get_tools()
 
@@ -77,6 +76,7 @@ class TestrailToolkit(BaseToolkit):
             # TODO use testrail_configuration fields
             **kwargs['testrail_configuration'],
             **(kwargs.get('pgvector_configuration') or {}),
+            **(kwargs.get('embedding_configuration') or {}),
         }
         testrail_api_wrapper = TestrailAPIWrapper(**wrapper_payload)
         prefix = clean_string(toolkit_name, cls.toolkit_max_length) + TOOLKIT_SPLITTER if toolkit_name else ''

{alita_sdk-0.3.242 → alita_sdk-0.3.244}/alita_sdk/tools/xray/__init__.py

@@ -27,10 +27,9 @@ def get_tools(tool):
         alita=tool['settings'].get('alita', None),
 
         # indexer settings
-        connection_string=tool['settings'].get('connection_string', None),
+        pgvector_configuration=tool['settings'].get('pgvector_configuration', {}),
+        embedding_configuration=tool['settings'].get('embedding_configuration', {}),
         collection_name=str(tool['toolkit_name']),
-        embedding_model="HuggingFaceEmbeddings",
-        embedding_model_params={"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
         vectorstore_type="PGVector"
     ).get_tools()
 
@@ -76,6 +75,7 @@ class XrayToolkit(BaseToolkit):
             # Use xray_configuration fields
             **kwargs.get('xray_configuration', {}),
             **(kwargs.get('pgvector_configuration') or {}),
+            **(kwargs.get('embedding_configuration') or {}),
         }
         xray_api_wrapper = XrayApiWrapper(**wrapper_payload)
         prefix = clean_string(toolkit_name, cls.toolkit_max_length) + TOOLKIT_SPLITTER if toolkit_name else ''

{alita_sdk-0.3.242 → alita_sdk-0.3.244}/alita_sdk/tools/zephyr_enterprise/__init__.py

@@ -21,10 +21,9 @@ def get_tools(tool):
         alita=tool['settings'].get('alita', None),
 
         # indexer settings
-        connection_string=tool['settings'].get('connection_string', None),
+        pgvector_configuration=tool['settings'].get('pgvector_configuration', {}),
+        embedding_configuration=tool['settings'].get('embedding_configuration', {}),
         collection_name=str(tool['toolkit_name']),
-        embedding_model="HuggingFaceEmbeddings",
-        embedding_model_params={"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
         vectorstore_type="PGVector"
     ).get_tools()
 
@@ -65,6 +64,7 @@ class ZephyrEnterpriseToolkit(BaseToolkit):
             # Use zephyr_configuration fields
             **kwargs.get('zephyr_configuration', {}),
             **(kwargs.get('pgvector_configuration') or {}),
+            **(kwargs.get('embedding_configuration') or {}),
         }
         zephyr_api_wrapper = ZephyrApiWrapper(**wrapper_payload)
         prefix = clean_string(toolkit_name, cls.toolkit_max_length) + TOOLKIT_SPLITTER if toolkit_name else ''

{alita_sdk-0.3.242 → alita_sdk-0.3.244}/alita_sdk/tools/zephyr_essential/__init__.py

@@ -20,10 +20,9 @@ def get_tools(tool):
         alita=tool['settings'].get('alita', None),
 
         # indexer settings
-        connection_string = tool['settings'].get('connection_string', None),
         collection_name=str(tool['toolkit_name']),
-        embedding_model = "HuggingFaceEmbeddings",
-        embedding_model_params = {"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
+        pgvector_configuration=tool['settings'].get('pgvector_configuration', {}),
+        embedding_configuration=tool['settings'].get('embedding_configuration', {}),
         vectorstore_type = "PGVector"
     ).get_tools()
 
@@ -55,7 +54,14 @@ class ZephyrEssentialToolkit(BaseToolkit):
 
     @classmethod
     def get_toolkit(cls, selected_tools: list[str] | None = None, toolkit_name: Optional[str] = None, **kwargs):
-        zephyr_api_wrapper = ZephyrEssentialApiWrapper(**kwargs)
+        if selected_tools is None:
+            selected_tools = []
+        wrapper_payload = {
+            **kwargs,
+            **(kwargs.get('pgvector_configuration') or {}),
+            **(kwargs.get('embedding_configuration') or {}),
+        }
+        zephyr_api_wrapper = ZephyrEssentialApiWrapper(**wrapper_payload)
         prefix = clean_string(toolkit_name, cls.toolkit_max_length) + TOOLKIT_SPLITTER if toolkit_name else ''
         available_tools = zephyr_api_wrapper.get_available_tools()
         tools = []

{alita_sdk-0.3.242 → alita_sdk-0.3.244}/alita_sdk.egg-info/PKG-INFO

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

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