alita-sdk 0.3.221__py3-none-any.whl → 0.3.223__py3-none-any.whl

alita_sdk/runtime/tools/artifact.py

@@ -90,7 +90,6 @@ class ArtifactWrapper(BaseVectorStoreToolApiWrapper):
                 yield Document(page_content=str(value), metadata=metadata)
         else:
             document.page_content = json.dumps(str(page_content))
-            yield document
 
     @extend_with_vector_tools
     def get_available_tools(self):

alita_sdk/runtime/tools/vectorstore.py

@@ -532,10 +532,10 @@ class VectorStoreWrapper(BaseToolApiWrapper):
 
     def search_documents(self, query:str, doctype: str = 'code', 
                          filter:dict|str={}, cut_off: float=0.5,
-                         search_top:int=10, reranker:dict = {}, 
-                         full_text_search: Optional[Dict[str, Any]] = None,
-                         reranking_config: Optional[Dict[str, Dict[str, Any]]] = None,
-                         extended_search: Optional[List[str]] = None):
+                         search_top:int=10, full_text_search: Optional[Dict[str, Any]] = None,
+                         extended_search: Optional[List[str]] = None,
+                         reranker: dict = {}, reranking_config: Optional[Dict[str, Dict[str, Any]]] = None
+                         ):
         """Enhanced search documents method using JSON configurations for full-text search and reranking"""
         from alita_sdk.tools.code.loaders.codesearcher import search_format as code_format
         

alita_sdk/tools/ado/repos/__init__.py

@@ -65,6 +65,13 @@ class AzureDevOpsReposToolkit(BaseToolkit):
             connection_string = (Optional[SecretStr], Field(description="Connection string for vectorstore",
                                                             default=None,
                                                             json_schema_extra={'secret': True})),
+
+            # 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"})),
             
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
             __config__={'json_schema_extra': {'metadata':

alita_sdk/tools/ado/test_plan/__init__.py

@@ -33,6 +33,16 @@ class AzureDevOpsPlansToolkit(BaseToolkit):
             project=(str, Field(title="Project", description="ADO project", json_schema_extra={'configuration': True})),
             limit=(Optional[int], Field(description="ADO plans limit used for limitation of the list with results", default=5)),
             token=(SecretStr, Field(description="ADO token", json_schema_extra={'secret': True, 'configuration': True})),
+            # indexer settings
+            connection_string=(Optional[SecretStr], Field(description="Connection string for vectorstore",
+                                                          default=None,
+                                                          json_schema_extra={'secret': True})),
+            # 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"})),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
             __config__={'json_schema_extra': {'metadata':
                 {

alita_sdk/tools/ado/wiki/__init__.py

@@ -37,6 +37,16 @@ class AzureDevOpsWikiToolkit(BaseToolkit):
                    Field(description="ADO token",
                          json_schema_extra={'secret': True, 'configuration': True }
                          )),
+            # indexer settings
+            connection_string=(Optional[SecretStr], Field(description="Connection string for vectorstore",
+                                                          default=None,
+                                                          json_schema_extra={'secret': True})),
+            # 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"})),
             selected_tools=(List[Literal[tuple(selected_tools)]],
                             Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
             __config__={

alita_sdk/tools/ado/work_item/__init__.py

@@ -41,6 +41,12 @@ class AzureDevOpsWorkItemsToolkit(BaseToolkit):
             connection_string = (Optional[SecretStr], Field(description="Connection string for vectorstore",
                                                             default=None,
                                                             json_schema_extra={'secret': True})),
+            # 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"})),
             __config__={
                 'json_schema_extra': {
                     'metadata': {

alita_sdk/tools/bitbucket/__init__.py

@@ -27,7 +27,7 @@ def get_tools(tool):
         llm=tool['settings'].get('llm', None),
         alita=tool['settings'].get('alita', None),
         connection_string=tool['settings'].get('connection_string', None),
-        collection_name=str(tool['id']),
+        collection_name=str(tool['toolkit_name']),
         doctype='code',
         embedding_model="HuggingFaceEmbeddings",
         embedding_model_params={"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
@@ -60,6 +60,10 @@ class AlitaBitbucketToolkit(BaseToolkit):
             connection_string=(Optional[SecretStr], Field(description="Connection string for vectorstore",
                                                           default=None,
                                                           json_schema_extra={'secret': True})),
+
+            # 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"})),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
             __config__=ConfigDict(json_schema_extra=
             {

alita_sdk/tools/confluence/__init__.py

@@ -27,7 +27,7 @@ def get_tools(tool):
         toolkit_name=tool.get('toolkit_name'),
         # indexer settings
         connection_string = tool['settings'].get('connection_string', None),
-        collection_name=str(tool['id']),
+        collection_name=str(tool['toolkit_name']),
         doctype='doc',
         embedding_model="HuggingFaceEmbeddings",
         embedding_model_params={"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
@@ -84,6 +84,11 @@ class ConfluenceToolkit(BaseToolkit):
             connection_string = (Optional[SecretStr], Field(description="Connection string for vectorstore",
                                                             default=None,
                                                             json_schema_extra={'secret': True})),
+
+            # 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"})),
+
             __config__=ConfigDict(json_schema_extra={
                 'metadata': {
                     "label": "Confluence",

alita_sdk/tools/elitea_base.py

@@ -12,6 +12,7 @@ from pydantic import BaseModel, create_model, Field, SecretStr
 from alita_sdk.runtime.langchain.interfaces.llm_processor import get_embeddings
 from .chunkers import markdown_chunker
 from .utils import TOOLKIT_SPLITTER
+from .vector_adapters.VectorStoreAdapter import VectorStoreAdapterFactory
 from ..runtime.utils.utils import IndexerKeywords
 
 logger = logging.getLogger(__name__)
@@ -59,22 +60,22 @@ BaseSearchParams = create_model(
     )),
     cut_off=(Optional[float], Field(description="Cut-off score for search results", default=0.5)),
     search_top=(Optional[int], Field(description="Number of top results to return", default=10)),
-    reranker=(Optional[dict], Field(
-        description="Reranker configuration. Can be a dictionary with reranking parameters.",
-        default={}
-    )),
     full_text_search=(Optional[Dict[str, Any]], Field(
         description="Full text search parameters. Can be a dictionary with search options.",
         default=None
     )),
-    reranking_config=(Optional[Dict[str, Dict[str, Any]]], Field(
-        description="Reranking configuration. Can be a dictionary with reranking settings.",
-        default=None
-    )),
     extended_search=(Optional[List[str]], Field(
         description="List of additional fields to include in the search results.",
         default=None
     )),
+    reranker=(Optional[dict], Field(
+        description="Reranker configuration. Can be a dictionary with reranking parameters.",
+        default={}
+    )),
+    reranking_config=(Optional[Dict[str, Dict[str, Any]]], Field(
+        description="Reranking configuration. Can be a dictionary with reranking settings.",
+        default=None
+    )),
 )
 
 BaseStepbackSearchParams = create_model(
@@ -111,7 +112,7 @@ BaseStepbackSearchParams = create_model(
 BaseIndexDataParams = create_model(
     "indexData",
     __base__=BaseIndexParams,
-    progress_step=(Optional[int], Field(default=None, ge=0, le=100,
+    progress_step=(Optional[int], Field(default=10, ge=0, le=100,
                          description="Optional step size for progress reporting during indexing")),
     clean_index=(Optional[bool], Field(default=False,
                        description="Optional flag to enforce clean existing index before indexing new data")),
@@ -219,6 +220,10 @@ class BaseVectorStoreToolApiWrapper(BaseToolApiWrapper):
     embedding_model_params: Optional[Dict[str, Any]] = {"model_name": "sentence-transformers/all-MiniLM-L6-v2"}
     vectorstore_type: Optional[str] = "PGVector"
 
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+        self._adapter = VectorStoreAdapterFactory.create_adapter(self.vectorstore_type)
+
     def _index_tool_params(self, **kwargs) -> dict[str, tuple[type, Field]]:
         """
         Returns a list of fields for index_data args schema.
@@ -329,36 +334,24 @@ class BaseVectorStoreToolApiWrapper(BaseToolApiWrapper):
 
     # TODO: init store once and re-use the instance
     def _init_vector_store(self, collection_suffix: str = "", embeddings: Optional[Any] = None):
-        """ Initializes the vector store wrapper with the provided parameters."""
+        """Initializes the vector store wrapper with the provided parameters."""
         try:
             from alita_sdk.runtime.tools.vectorstore import VectorStoreWrapper
         except ImportError:
             from alita_sdk.runtime.tools.vectorstore import VectorStoreWrapper
-        
+
         # Validate collection_suffix length
         if collection_suffix and len(collection_suffix.strip()) > 7:
             raise ToolException("collection_suffix must be 7 characters or less")
-        
+
         # Create collection name with suffix if provided
         collection_name = str(self.collection_name)
         if collection_suffix and collection_suffix.strip():
             collection_name = f"{self.collection_name}_{collection_suffix.strip()}"
-        
-        if self.vectorstore_type == 'PGVector':
-            vectorstore_params = {
-                "use_jsonb": True,
-                "collection_name": collection_name,
-                "create_extension": True,
-                "alita_sdk_options": {
-                    "target_schema": collection_name,
-                },
-                "connection_string": self.connection_string.get_secret_value()
-            }
-        elif self.vectorstore_type == 'Chroma':
-            vectorstore_params = {
-                "collection_name": collection_name,
-                "persist_directory": "./indexer_db"
-            }
+
+        # Get database-specific parameters using adapter
+        connection_string = self.connection_string.get_secret_value() if self.connection_string else None
+        vectorstore_params = self._adapter.get_vectorstore_params(collection_name, connection_string)
 
         return VectorStoreWrapper(
             llm=self.llm,
@@ -371,36 +364,17 @@ class BaseVectorStoreToolApiWrapper(BaseToolApiWrapper):
         )
 
     def remove_index(self, collection_suffix: str = ""):
-        """
-            Cleans the indexed data in the collection
-        """
-
-        self._init_vector_store(collection_suffix)._remove_collection()
+        """Cleans the indexed data in the collection."""
+        vectorstore_wrapper = self._init_vector_store(collection_suffix)
+        collection_name = f"{self.collection_name}_{collection_suffix}" if collection_suffix else str(self.collection_name)
+        self._adapter.remove_collection(vectorstore_wrapper, collection_name)
+        return (f"Collection '{collection_name}' has been removed from the vector store.\n"
+                f"Available collections: {self.list_collections()}")
 
     def list_collections(self):
-        """
-            Lists all collections in the vector store
-        """
-        if self.vectorstore_type.lower() == 'PGVector'.lower():
-            from sqlalchemy import text
-            from sqlalchemy.orm import Session
-
-            # schema_name = self.vectorstore.collection_name
-            with Session(self._init_vector_store().vectorstore.session_maker.bind) as session:
-                get_collections = text("""
-                    SELECT table_schema
-                    FROM information_schema.columns
-                    WHERE udt_name = 'vector';
-                """)
-
-                # Execute the raw SQL query
-                result = session.execute(get_collections)
-
-                # Fetch all rows from the result
-                docs = result.fetchall()
-            return str(docs)
-        vector_client = self._init_vector_store().vectoradapter.vectorstore._client
-        return ','.join([collection.name for collection in vector_client.list_collections()])
+        """Lists all collections in the vector store."""
+        vectorstore_wrapper = self._init_vector_store()
+        return self._adapter.list_collections(vectorstore_wrapper)
 
     def search_index(self,
                      query: str,

alita_sdk/tools/figma/__init__.py

@@ -22,7 +22,7 @@ def get_tools(tool):
             # indexer settings
             llm=tool['settings'].get('llm', None),
             connection_string = tool['settings'].get('connection_string', None),
-            collection_name=str(tool['id']),
+            collection_name=str(tool['toolkit_name']),
             doctype='doc',
             embedding_model="HuggingFaceEmbeddings",
             embedding_model_params={"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
@@ -57,6 +57,10 @@ class FigmaToolkit(BaseToolkit):
             connection_string = (Optional[SecretStr], Field(description="Connection string for vectorstore",
                                                             default=None,
                                                             json_schema_extra={'secret': True})),
+
+            # 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"})),
             __config__=ConfigDict(
                 json_schema_extra={
                      "metadata": {

alita_sdk/tools/github/__init__.py

@@ -25,7 +25,7 @@ def _get_toolkit(tool) -> BaseToolkit:
         llm=tool['settings'].get('llm', None),
         alita=tool['settings'].get('alita', None),
         connection_string=tool['settings'].get('connection_string', None),
-        collection_name=str(tool['id']),
+        collection_name=str(tool['toolkit_name']),
         doctype='code',
         embedding_model="HuggingFaceEmbeddings",
         embedding_model_params={"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
@@ -96,6 +96,10 @@ class AlitaGitHubToolkit(BaseToolkit):
                                                             default=None,
                                                             json_schema_extra={'secret': True})),
             
+            # 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"})),
+
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools}))
         )
 
@@ -123,4 +127,3 @@ class AlitaGitHubToolkit(BaseToolkit):
 
     def get_tools(self):
         return self.tools
-

alita_sdk/tools/gitlab/__init__.py

@@ -7,7 +7,6 @@ from pydantic import create_model, BaseModel, ConfigDict, SecretStr
 from pydantic.fields import Field
 
 from .api_wrapper import GitLabAPIWrapper
-# from .tools import __all__
 from ..utils import clean_string, TOOLKIT_SPLITTER, get_max_toolkit_length
 
 name = "gitlab"
@@ -24,7 +23,7 @@ def get_tools(tool):
         llm=tool['settings'].get('llm', None),
         alita=tool['settings'].get('alita', None),
         connection_string=tool['settings'].get('connection_string', None),
-        collection_name=f"{tool.get('toolkit_name')}_{str(tool['id'])}",
+        collection_name=str(tool['toolkit_name']),
         doctype='code',
         embedding_model="HuggingFaceEmbeddings",
         embedding_model_params={"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
@@ -51,6 +50,10 @@ class AlitaGitlabToolkit(BaseToolkit):
             connection_string=(Optional[SecretStr], Field(description="Connection string for vectorstore",
                                                           default=None,
                                                           json_schema_extra={'secret': True})),
+
+            # 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"})),
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
             __config__=ConfigDict(json_schema_extra={
                 'metadata': {

alita_sdk/tools/jira/__init__.py

@@ -25,7 +25,7 @@ def get_tools(tool):
         llm=tool['settings'].get('llm', None),
         alita=tool['settings'].get('alita', None),
         connection_string=tool['settings'].get('connection_string', None),
-        collection_name=f"{tool.get('toolkit_name')}_{str(tool['id'])}",
+        collection_name=str(tool['toolkit_name']),
         embedding_model="HuggingFaceEmbeddings",
         embedding_model_params={"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
         vectorstore_type="PGVector",
@@ -81,6 +81,16 @@ class JiraToolkit(BaseToolkit):
             )),
             verify_ssl=(bool, Field(description="Verify SSL", default=True)),
             additional_fields=(Optional[str], Field(description="Additional fields", default="")),
+
+            # indexer settings
+            connection_string=(Optional[SecretStr], Field(description="Connection string for vectorstore",
+                                                          default=None,
+                                                          json_schema_extra={'secret': True})),
+
+            # 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"})),
+
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
             __config__=ConfigDict(json_schema_extra={
                 'metadata': {

alita_sdk/tools/sharepoint/__init__.py

@@ -18,7 +18,7 @@ def get_tools(tool):
         llm=tool['settings'].get('llm'),
         # indexer settings
         connection_string=tool['settings'].get('connection_string', None),
-        collection_name=f"{tool.get('toolkit_name')}_{str(tool['id'])}",
+        collection_name=str(tool['toolkit_name']),
         embedding_model="HuggingFaceEmbeddings",
         embedding_model_params={"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
         vectorstore_type="PGVector")
@@ -43,6 +43,10 @@ class SharepointToolkit(BaseToolkit):
             connection_string = (Optional[SecretStr], Field(description="Connection string for vectorstore",
                                                             default=None,
                                                             json_schema_extra={'secret': True})),
+
+            # 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"})),
             __config__=ConfigDict(json_schema_extra={
                 'metadata': {
                     "label": "Sharepoint", "icon_url": "sharepoint.svg",

alita_sdk/tools/sharepoint/api_wrapper.py

@@ -172,7 +172,6 @@ class SharepointApiWrapper(BaseVectorStoreToolApiWrapper):
                 yield Document(page_content=str(content), metadata=new_metadata)
         else:
             document.page_content = str(doc_content)
-            yield document
 
     @extend_with_vector_tools
     def get_available_tools(self):

alita_sdk/tools/testrail/__init__.py

@@ -55,6 +55,10 @@ class TestrailToolkit(BaseToolkit):
             connection_string=(Optional[SecretStr], Field(description="Connection string for vectorstore",
                                                           default=None,
                                                           json_schema_extra={'secret': True})),
+
+            # 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"})),
             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/tools/vector_adapters/VectorStoreAdapter.py

@@ -0,0 +1,91 @@
+from abc import ABC, abstractmethod
+from typing import Any, Dict, Optional
+
+
+class VectorStoreAdapter(ABC):
+    """Abstract base class for vector store adapters."""
+
+    @abstractmethod
+    def get_vectorstore_params(self, collection_name: str, connection_string: Optional[str] = None) -> Dict[str, Any]:
+        """Get vector store specific parameters."""
+        pass
+
+    @abstractmethod
+    def list_collections(self, vectorstore_wrapper) -> str:
+        """List all collections in the vector store."""
+        pass
+
+    @abstractmethod
+    def remove_collection(self, vectorstore_wrapper, collection_name: str):
+        """Remove a collection from the vector store."""
+        pass
+
+
+class PGVectorAdapter(VectorStoreAdapter):
+    """Adapter for PGVector database operations."""
+
+    def get_vectorstore_params(self, collection_name: str, connection_string: Optional[str] = None) -> Dict[str, Any]:
+        return {
+            "use_jsonb": True,
+            "collection_name": collection_name,
+            "create_extension": True,
+            "alita_sdk_options": {
+                "target_schema": collection_name,
+            },
+            "connection_string": connection_string
+        }
+
+    def list_collections(self, vectorstore_wrapper) -> str:
+        from sqlalchemy import text
+        from sqlalchemy.orm import Session
+
+        with Session(vectorstore_wrapper.vectorstore.session_maker.bind) as session:
+            get_collections = text("""
+                SELECT table_schema
+                FROM information_schema.columns
+                WHERE udt_name = 'vector';
+            """)
+            result = session.execute(get_collections)
+            docs = result.fetchall()
+        return str(docs)
+
+    def remove_collection(self, vectorstore_wrapper, collection_name: str):
+        vectorstore_wrapper._remove_collection()
+
+
+class ChromaAdapter(VectorStoreAdapter):
+    """Adapter for Chroma database operations."""
+
+    def get_vectorstore_params(self, collection_name: str, connection_string: Optional[str] = None) -> Dict[str, Any]:
+        return {
+            "collection_name": collection_name,
+            "persist_directory": "./indexer_db"
+        }
+
+    def list_collections(self, vectorstore_wrapper) -> str:
+        vector_client = vectorstore_wrapper.vectoradapter.vectorstore._client
+        return ','.join([collection.name for collection in vector_client.list_collections()])
+
+    def remove_collection(self, vectorstore_wrapper, collection_name: str):
+        vectorstore_wrapper.vectoradapter.vectorstore.delete_collection()
+
+
+class VectorStoreAdapterFactory:
+    """Factory for creating vector store adapters."""
+
+    _adapters = {
+        'PGVector': PGVectorAdapter,
+        'Chroma': ChromaAdapter,
+    }
+
+    @classmethod
+    def create_adapter(cls, vectorstore_type: str) -> VectorStoreAdapter:
+        adapter_class = cls._adapters.get(vectorstore_type)
+        if not adapter_class:
+            raise ValueError(f"Unsupported vectorstore type: {vectorstore_type}")
+        return adapter_class()
+
+    @classmethod
+    def register_adapter(cls, vectorstore_type: str, adapter_class: type):
+        """Register a new adapter for a vector store type."""
+        cls._adapters[vectorstore_type] = adapter_class

alita_sdk/tools/xray/__init__.py

@@ -24,7 +24,7 @@ def get_tools(tool):
 
         # indexer settings
         connection_string=tool['settings'].get('connection_string', None),
-        collection_name=f"{tool.get('toolkit_name')}_{str(tool['id'])}",
+        collection_name=str(tool['toolkit_name']),
         embedding_model="HuggingFaceEmbeddings",
         embedding_model_params={"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
         vectorstore_type="PGVector"
@@ -45,6 +45,16 @@ class XrayToolkit(BaseToolkit):
             client_id=(str, Field(description="Client ID")),
             client_secret=(SecretStr, Field(description="Client secret", json_schema_extra={'secret': True})),
             limit=(Optional[int], Field(description="Limit", default=100)),
+
+            # indexer settings
+            connection_string=(Optional[SecretStr], Field(description="Connection string for vectorstore",
+                                                          default=None,
+                                                          json_schema_extra={'secret': True})),
+
+            # 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"})),
+
             selected_tools=(List[Literal[tuple(selected_tools)]], Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
             __config__={'json_schema_extra':
                             {

alita_sdk/tools/zephyr_enterprise/__init__.py

@@ -18,7 +18,7 @@ def get_tools(tool):
 
         # indexer settings
         connection_string=tool['settings'].get('connection_string', None),
-        collection_name=f"{tool.get('toolkit_name')}_{str(tool['id'])}",
+        collection_name=str(tool['toolkit_name']),
         embedding_model="HuggingFaceEmbeddings",
         embedding_model_params={"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
         vectorstore_type="PGVector"
@@ -41,6 +41,10 @@ class ZephyrEnterpriseToolkit(BaseToolkit):
             connection_string=(Optional[SecretStr], Field(description="Connection string for vectorstore",
                                                           default=None,
                                                           json_schema_extra={'secret': True})),
+
+            # 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"})),
             selected_tools=(List[Literal[tuple(selected_tools)]], []),
             __config__=ConfigDict(json_schema_extra={
                 'metadata': {
@@ -71,4 +75,3 @@ class ZephyrEnterpriseToolkit(BaseToolkit):
 
     def get_tools(self) -> List[BaseTool]:
         return self.tools
-

alita_sdk/tools/zephyr_enterprise/api_wrapper.py

@@ -1,15 +1,29 @@
 import logging
-from typing import Optional, List
+from importlib.metadata import metadata
+from operator import ifloordiv
+from typing import Optional, List, Generator
 
 from langchain_core.tools import ToolException
 from pydantic import create_model, model_validator, PrivateAttr, Field, SecretStr
 
+from langchain_core.documents import Document
 from .zephyr_enterprise import ZephyrClient
-from ..elitea_base import BaseToolApiWrapper
+from ..elitea_base import BaseToolApiWrapper, BaseVectorStoreToolApiWrapper, extend_with_vector_tools
 
 logger = logging.getLogger(__name__)
 
-class ZephyrApiWrapper(BaseToolApiWrapper):
+zql_description = """
+                    ZQL query to search for test cases. 
+                    Supported: estimatedTime, testcaseId, creator, release,
+                    project, priority, altId, version,
+                    versionId, automated, folder, contents,
+                    name, comment, tag
+
+                    It has to follow the syntax in examples:
+                    "folder=\"TestToolkit\"", "name~\"TestToolkit5\"
+                    """
+
+class ZephyrApiWrapper(BaseVectorStoreToolApiWrapper):
     base_url: str
     token: SecretStr
     _client: Optional[ZephyrClient] = PrivateAttr()
@@ -39,7 +53,7 @@ class ZephyrApiWrapper(BaseToolApiWrapper):
             return ToolException(f"Unable to retrieve Zephyr entities: {e}")
 
 
-    def get_testcases_by_zql(self, zql: str):
+    def get_testcases_by_zql(self, zql: str, return_as_list: bool = False):
 
         """
         Retrieve testcases by zql.
@@ -50,8 +64,12 @@ class ZephyrApiWrapper(BaseToolApiWrapper):
             testcases = self._client.get_testcases_by_zql(zql)
             parsed_test_cases = []
             if testcases['resultSize'] == 0:
+                if return_as_list:
+                    return []
                 return "No test cases found for the provided ZQL query."
             logger.info(f"Retrieved test cases: {testcases}")
+            if return_as_list:
+                return [test_case['testcase'] for test_case in testcases['results']]
             for test_case in testcases['results']:
                 parsed_test_cases.append(f"Test case ID: {test_case.get('id')}, Test case: {test_case['testcase']}")
             return "\n".join(parsed_test_cases)
@@ -69,6 +87,20 @@ class ZephyrApiWrapper(BaseToolApiWrapper):
         except Exception as e:
             return ToolException(f"Unable to retrieve Zephyr entities: {e}")
 
+    def get_test_case_steps(self, testcase_tree_id: str):
+        last_version = self.get_last_version(testcase_tree_id)
+        return self._client.get_testcase_steps(last_version)
+
+    def get_last_version(self, testcase_tree_id: str):
+        # get test case id
+        test_case_id = self._client.get_test_case(testcase_tree_id)['testcase']['testcaseId']
+        logger.info(f"Test case id: {test_case_id}")
+        # get test case version id
+        return self._client.get_testcase_versions(test_case_id)[-1]['id']
+
+    def get_test_case_steps_by_version(self, test_case_version_id: str):
+        return self._client.get_testcase_steps(test_case_version_id)
+
     def add_steps(self, testcase_tree_id: str, steps: Optional[List[dict]] = []):
 
         """
@@ -81,14 +113,10 @@ class ZephyrApiWrapper(BaseToolApiWrapper):
             if not steps:
                 return ToolException("Steps cannot be empty.")
 
-            # get test case id
-            test_case_id = self._client.get_test_case(testcase_tree_id)['testcase']['testcaseId']
-            logger.info(f"Test case id: {test_case_id}")
-            # get test case version id
-            test_case_version_id = self._client.get_testcase_versions(test_case_id)[-1]['id']
+            test_case_version_id = self.get_last_version(testcase_tree_id)
             logger.info(f"Test case version id: {test_case_version_id}")
             # get test case steps
-            test_case_steps = self._client.get_testcase_steps(test_case_version_id)
+            test_case_steps = self.get_test_case_steps_by_version(test_case_version_id)
             logger.info(f"Test case steps: {test_case_steps}")
             # add steps
             actual_steps = test_case_steps['steps'] if test_case_steps else None
@@ -120,6 +148,30 @@ class ZephyrApiWrapper(BaseToolApiWrapper):
         except Exception as e:
             return ToolException(f"Unable to retrieve Zephyr entities: {e}")
 
+    def _index_tool_params(self, **kwargs) -> dict[str, tuple[type, Field]]:
+        """
+        Returns a list of fields for index_data args schema.
+        """
+        return {
+            "zql": (str, Field(description=zql_description, examples=["folder=\"TestToolkit\"", "name~\"TestToolkit5\""))
+        }
+
+    def _base_loader(self, zql: str) -> Generator[Document, None, None]:
+        test_cases = self.get_testcases_by_zql(zql)
+        for test_case in test_cases:
+            metadata = {
+                ("updated_on" if k == "lastModifiedOn" else "id" if k == "testcaseId" else k): str(v)
+                for k, v in test_case
+                if k != "id"
+            }
+            yield Document(page_content='', metadata=metadata)
+
+    def _process_document(self, document: Document) -> Generator[Document, None, None]:
+        id = document.metadata['id']
+        test_case_content = self.get_test_case_steps(id)
+        document.page_content = test_case_content
+
+    @extend_with_vector_tools
     def get_available_tools(self):
         return [
             {
@@ -188,16 +240,7 @@ class ZephyrApiWrapper(BaseToolApiWrapper):
                 "description": self.get_testcases_by_zql.__doc__,
                 "args_schema": create_model(
                     "TestCaseByZqlModel",
-                    zql=(str, Field(description="""
-                    ZQL query to search for test cases. 
-                    Supported: estimatedTime, testcaseId, creator, release,
-                    project, priority, altId, version,
-                    versionId, automated, folder, contents,
-                    name, comment, tag
-                    
-                    It has to follow the syntax in examples:
-                    "folder=\"TestToolkit\"", "name~\"TestToolkit5\"
-                    """, examples=["folder=\"TestToolkit\"", "name~\"TestToolkit5\""]))
+                    zql=(str, Field(description=zql_description, examples=["folder=\"TestToolkit\"", "name~\"TestToolkit5\""]))
                 ),
                 "ref": self.get_testcases_by_zql,
             }

alita_sdk/tools/zephyr_essential/__init__.py

@@ -18,7 +18,7 @@ def get_tools(tool):
 
         # indexer settings
         connection_string = tool['settings'].get('connection_string', None),
-        collection_name = f"{tool.get('toolkit_name')}_{str(tool['id'])}",
+        collection_name=str(tool['toolkit_name']),
         embedding_model = "HuggingFaceEmbeddings",
         embedding_model_params = {"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
         vectorstore_type = "PGVector"
@@ -41,6 +41,10 @@ class ZephyrEssentialToolkit(BaseToolkit):
             connection_string=(Optional[SecretStr], Field(description="Connection string for vectorstore",
                                                           default=None,
                                                           json_schema_extra={'secret': True})),
+
+            # 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"})),
             __config__={'json_schema_extra': {'metadata': {"label": "Zephyr Essential", "icon_url": "zephyr.svg",
                             "categories": ["test management"],
                             "extra_categories": ["test automation", "test case management", "test planning"]
@@ -67,4 +71,3 @@ class ZephyrEssentialToolkit(BaseToolkit):
 
     def get_tools(self):
         return self.tools
-

alita_sdk/tools/zephyr_essential/api_wrapper.py

@@ -253,7 +253,6 @@ class ZephyrEssentialApiWrapper(BaseVectorStoreToolApiWrapper):
                 base_data['test_case_content'] = additional_content
 
             document.page_content = json.dumps(base_data)
-            yield document
         except json.JSONDecodeError as e:
             raise ToolException(f"Failed to decode JSON from document: {e}")
 

alita_sdk/tools/zephyr_scale/api_wrapper.py

@@ -1283,7 +1283,6 @@ class ZephyrScaleApiWrapper(BaseVectorStoreToolApiWrapper):
                 base_data['test_case_content'] = additional_content
 
             document.page_content = json.dumps(base_data)
-            yield document
         except json.JSONDecodeError as e:
             raise ToolException(f"Failed to decode JSON from document: {e}")
 

{alita_sdk-0.3.221.dist-info → alita_sdk-0.3.223.dist-info}/METADATA

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

@@ -69,7 +69,7 @@ alita_sdk/runtime/toolkits/vectorstore.py,sha256=BGppQADa1ZiLO17fC0uCACTTEvPHlod
 alita_sdk/runtime/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/runtime/tools/agent.py,sha256=m98QxOHwnCRTT9j18Olbb5UPS8-ZGeQaGiUyZJSyFck,3162
 alita_sdk/runtime/tools/application.py,sha256=mC2_ZFx4WLHc98Gzll88Vw6cqyx2cmbig2IeJBtHRdg,2836
-alita_sdk/runtime/tools/artifact.py,sha256=BWeLWnwlVgOaAVxJfREeMkdDXnPEiTas2L0iUujVBWI,8692
+alita_sdk/runtime/tools/artifact.py,sha256=K8TE4z5gYOLgY1ee51DkeN4rXt4neXdLFLpsxHX6MKE,8665
 alita_sdk/runtime/tools/datasource.py,sha256=pvbaSfI-ThQQnjHG-QhYNSTYRnZB0rYtZFpjCfpzxYI,2443
 alita_sdk/runtime/tools/echo.py,sha256=spw9eCweXzixJqHnZofHE1yWiSUa04L4VKycf3KCEaM,486
 alita_sdk/runtime/tools/function.py,sha256=ZFpd7TGwIawze2e7BHlKwP0NHwNw42wwrmmnXyJQJhk,2600
@@ -82,7 +82,7 @@ alita_sdk/runtime/tools/pgvector_search.py,sha256=NN2BGAnq4SsDHIhUcFZ8d_dbEOM8Qw
 alita_sdk/runtime/tools/prompt.py,sha256=nJafb_e5aOM1Rr3qGFCR-SKziU9uCsiP2okIMs9PppM,741
 alita_sdk/runtime/tools/router.py,sha256=wCvZjVkdXK9dMMeEerrgKf5M790RudH68pDortnHSz0,1517
 alita_sdk/runtime/tools/tool.py,sha256=lE1hGi6qOAXG7qxtqxarD_XMQqTghdywf261DZawwno,5631
-alita_sdk/runtime/tools/vectorstore.py,sha256=lpbpS2yukyT9xRkT2tNQl9YqnlO5F0rNCyj0nU7OJDE,38537
+alita_sdk/runtime/tools/vectorstore.py,sha256=-AMZaWHeMoM5k0GX62vDA9BThSM1e-5UwsLUM3b4WL8,38538
 alita_sdk/runtime/utils/AlitaCallback.py,sha256=E4LlSBuCHWiUq6W7IZExERHZY0qcmdjzc_rJlF2iQIw,7356
 alita_sdk/runtime/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/runtime/utils/constants.py,sha256=Xntx1b_uxUzT4clwqHA_U6K8y5bBqf_4lSQwXdcWrp4,13586
@@ -94,16 +94,16 @@ alita_sdk/runtime/utils/toolkit_runtime.py,sha256=MU63Fpxj0b5_r1IUUc0Q3-PN9VwL7r
 alita_sdk/runtime/utils/toolkit_utils.py,sha256=I9QFqnaqfVgN26LUr6s3XlBlG6y0CoHURnCzG7XcwVs,5311
 alita_sdk/runtime/utils/utils.py,sha256=CpEl3LCeLbhzQySz08lkKPm7Auac6IiLF7WB8wmArMI,589
 alita_sdk/tools/__init__.py,sha256=1AHqP2xyLjn92xVm70l9XIke6FkfHkLo5OoQVe4BuP8,10421
-alita_sdk/tools/elitea_base.py,sha256=HH02c1j7MtzJmuEumM8yQHxFuRL1Tc3ZavD27bvMoi4,31091
+alita_sdk/tools/elitea_base.py,sha256=anEPuN9dtOsSTJQlHpap5ho_hf-iPmBQpw4tmNUTP5k,30475
 alita_sdk/tools/ado/__init__.py,sha256=2NMQwt2pjIukSC9nSZ7CLocdGpK7002x7ixKr_wunxk,1313
 alita_sdk/tools/ado/utils.py,sha256=PTCludvaQmPLakF2EbCGy66Mro4-rjDtavVP-xcB2Wc,1252
-alita_sdk/tools/ado/repos/__init__.py,sha256=guYY95Gtyb0S4Jj0V1qO0x2jlRoH0H1cKjHXNwmShow,6388
+alita_sdk/tools/ado/repos/__init__.py,sha256=VvOapCG5okdyfCNdeTfOdcgRvl54cB1-sx_BF0TBWOw,6858
 alita_sdk/tools/ado/repos/repos_wrapper.py,sha256=SgnEUX1lTn5ncZI8EFfPibk0_gHstkApDSqTCxYsrUk,50097
-alita_sdk/tools/ado/test_plan/__init__.py,sha256=bVywTYTvdm1rUeP2krVVMRN-xDCY--ze7NFdTxJP9ow,4708
+alita_sdk/tools/ado/test_plan/__init__.py,sha256=QUNSOCVheUbILve4CxjBzBHKcIA28jZjoA6rRb8tD_Y,5483
 alita_sdk/tools/ado/test_plan/test_plan_wrapper.py,sha256=MIEgTcjXQ_Hmx9Thzb1VXrviiB-nusFe-ym3BYlwa_4,20175
-alita_sdk/tools/ado/wiki/__init__.py,sha256=WCIKOisU2h3E4SNDvGfWCMZ3nRMxfH_ZhIffmSHH3XI,4576
+alita_sdk/tools/ado/wiki/__init__.py,sha256=3cPNqUyOFwnZ2g5QW8wsdBC9sB0u39F-q4KKVjQqTN8,5351
 alita_sdk/tools/ado/wiki/ado_wrapper.py,sha256=JfEbq2OBVtQe-6igYj09ucxwHEl34SozPIJ6gRGWcU8,14157
-alita_sdk/tools/ado/work_item/__init__.py,sha256=Is-YHu7OZdxp4iJZzHqXHM5VANK2Y6a_C-oU7ryNiJ8,5038
+alita_sdk/tools/ado/work_item/__init__.py,sha256=dBDjP8CJnKe9xBbJg-EG1AvRo06mcfS2f0AVeXMaBmk,5507
 alita_sdk/tools/ado/work_item/ado_wrapper.py,sha256=7YrqNIDnVYYm8__joABAr3iIk3IvYoQ7NZ9Vi-iaICc,28233
 alita_sdk/tools/advanced_jira_mining/__init__.py,sha256=pUTzECqGvYaR5qWY3JPUhrImrZgc7pCXuqSe5eWIE80,4604
 alita_sdk/tools/advanced_jira_mining/data_mining_wrapper.py,sha256=nZPtuwVWp8VeHw1B8q9kdwf-6ZvHnlXTOGdcIMDkKpw,44211
@@ -117,7 +117,7 @@ alita_sdk/tools/azure_ai/search/__init__.py,sha256=RaaWHX9o2NMxRruQGQwsDfacvtEJ5
 alita_sdk/tools/azure_ai/search/api_wrapper.py,sha256=E4p6HPDlwgxfT_i6cvg9rN4Vn_47CVAyNBAKLIGq3mU,7265
 alita_sdk/tools/base/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/tools/base/tool.py,sha256=-N27AodZS49vdPCgFkU-bFS9bxoPopZBnNrmwInx3d0,864
-alita_sdk/tools/bitbucket/__init__.py,sha256=jLnTLdfDK-n3AoWD_ljVVYo8u2TuFmWxB5Cwn7p8PGQ,5051
+alita_sdk/tools/bitbucket/__init__.py,sha256=EEJUp965OT0JijI6x7U7DP-_5sgRC3Q3TbNnxky6hPo,5458
 alita_sdk/tools/bitbucket/api_wrapper.py,sha256=wZp00LAxRs-nsMAduXUHTGGV41ayThpAKqGKmm4g27U,10880
 alita_sdk/tools/bitbucket/bitbucket_constants.py,sha256=UsbhQ1iEvrKoxceTFPWTYhaXS1zSxbmjs1TwY0-P4gw,462
 alita_sdk/tools/bitbucket/cloud_api_wrapper.py,sha256=VELi65tLXvszwCGQSqVfyVal0ylx9DgAmAGpRQL_Zkg,15522
@@ -185,7 +185,7 @@ alita_sdk/tools/code/linter/api_wrapper.py,sha256=wylpwhAw02Jt8L18CqBq2He5PbwIkx
 alita_sdk/tools/code/loaders/codesearcher.py,sha256=XoXXZtIQZhvjIwZlnl_4wVGHC-3saYzFo5oDR_Zh3EY,529
 alita_sdk/tools/code/sonar/__init__.py,sha256=u8wpgXJ_shToLl3G9-XEtGDor5dhmsnurIImh1-e-U0,3165
 alita_sdk/tools/code/sonar/api_wrapper.py,sha256=nNqxcWN_6W8c0ckj-Er9HkNuAdgQLoWBXh5UyzNutis,2653
-alita_sdk/tools/confluence/__init__.py,sha256=1wGSP4CjPbfpdZLsjC1SVftTV7XSvW3QCAMHuh9tIxA,6885
+alita_sdk/tools/confluence/__init__.py,sha256=Osn9Sd8dURk8zms6J6MDYtH-MAQaXcI21MW_MOZS9pE,7293
 alita_sdk/tools/confluence/api_wrapper.py,sha256=VR0TEMJPqe3ylzl_DuFSZ3LS0oTIJ7SufjifRXIYEhg,84507
 alita_sdk/tools/confluence/loader.py,sha256=4bf5qrJMEiJzuZp2NlxO2XObLD1w7fxss_WyMUpe8sg,9290
 alita_sdk/tools/confluence/utils.py,sha256=Lxo6dBD0OlvM4o0JuK6qeB_4LV9BptiwJA9e1vqNcDw,435
@@ -193,16 +193,16 @@ alita_sdk/tools/custom_open_api/__init__.py,sha256=9aT5SPNPWcJC6jMZEM-3rUCXVULj_
 alita_sdk/tools/custom_open_api/api_wrapper.py,sha256=sDSFpvEqpSvXHGiBISdQQcUecfO3md-_F8hAi6p2dvg,4340
 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=O5Aiv1g9Zub1Z6bfxKRK3WuVKGmAh7sPD0rJ5WdsGK4,4822
+alita_sdk/tools/figma/__init__.py,sha256=VBimOTXVoq0KK4dXcHaWT0yDcPZxhHYpZ_J3AqL_4xU,5229
 alita_sdk/tools/figma/api_wrapper.py,sha256=Q5BbEVfzIPGGSAV-T9gv0a31ku4jqCCvQDF85etHJEo,19276
-alita_sdk/tools/github/__init__.py,sha256=YPpZPPhRUHWKJ9aaMJnkjl9xrnAij1YB9C2TMRnlaTI,6388
+alita_sdk/tools/github/__init__.py,sha256=QIu4RQcuM21kVQk9yhCVdUsE4RERKb1FIWFRO9YaSAc,6794
 alita_sdk/tools/github/api_wrapper.py,sha256=JRhn7Cgg2j6uEwlvuQCMeISNYvRV2Yahx-v-p8HspUQ,8767
 alita_sdk/tools/github/github_client.py,sha256=nxnSXsDul2PPbWvYZS8TmAFFmR-5ALyakNoV5LN2D4U,86617
 alita_sdk/tools/github/graphql_client_wrapper.py,sha256=d3AGjzLGH_hdQV2V8HeAX92dJ4dlnE5OXqUlCO_PBr0,71539
 alita_sdk/tools/github/schemas.py,sha256=yFsqivfjCPRk9GxFJrL8sTz6nnjFCZ0j5DIfPtGSsvA,13852
 alita_sdk/tools/github/tool.py,sha256=Jnnv5lenV5ds8AAdyo2m8hSzyJ117HZBjzHC6T1ck-M,1037
 alita_sdk/tools/github/tool_prompts.py,sha256=y6ZW_FpUCE87Uop3WuQAZVRnzxO5t7xjBOI5bCqiluw,30194
-alita_sdk/tools/gitlab/__init__.py,sha256=J_nKHitpD1EhoOtnI5D2Qn-IyiiWd1sxOT181YPpF2k,4507
+alita_sdk/tools/gitlab/__init__.py,sha256=DftTt4uVaRG4g2RFvLKyghG_A28ukGPHK9G2jojEGnM,4853
 alita_sdk/tools/gitlab/api_wrapper.py,sha256=kAGHmne6Wh8WrQQUdZ1d8bhgzgmN_fAz49WZsX7K1Ps,22396
 alita_sdk/tools/gitlab/tools.py,sha256=vOGTlSaGaFmWn6LS6YFP-FuTqUPun9vnv1VrUcUHAZQ,16500
 alita_sdk/tools/gitlab/utils.py,sha256=Z2XiqIg54ouqqt1to-geFybmkCb1I6bpE91wfnINH1I,2320
@@ -218,7 +218,7 @@ alita_sdk/tools/google/bigquery/schemas.py,sha256=Gb8KQZSoRkmjXiz21dTC95c1MHEHFc
 alita_sdk/tools/google/bigquery/tool.py,sha256=Esf9Hsp8I0e7-5EdkFqQ-bid0cfrg-bfSoHoW_IIARo,1027
 alita_sdk/tools/google_places/__init__.py,sha256=mHKc7u9P2gqGDzqqJNQC9qiZYEm5gncnM_1XjtrM17o,3152
 alita_sdk/tools/google_places/api_wrapper.py,sha256=7nZly6nk4f4Tm7s2MVdnnwlb-1_WHRrDhyjDiqoyPjA,4674
-alita_sdk/tools/jira/__init__.py,sha256=siRFageS8tm9Ak67cWq8BgfnhUWcVrPK52zj4p60ZvI,6239
+alita_sdk/tools/jira/__init__.py,sha256=ieBe08us8poRzjLnuIxHGoP33sJRSHbdr8Z-GbI4P6k,6922
 alita_sdk/tools/jira/api_wrapper.py,sha256=gZXEtOZtWvINHipHPj8Dg6uNyYKZariLo4Bs3_wLJrA,75932
 alita_sdk/tools/keycloak/__init__.py,sha256=0WB9yXMUUAHQRni1ghDEmd7GYa7aJPsTVlZgMCM9cQ0,3050
 alita_sdk/tools/keycloak/api_wrapper.py,sha256=cOGr0f3S3-c6tRDBWI8wMnetjoNSxiV5rvC_0VHb8uw,3100
@@ -271,8 +271,8 @@ alita_sdk/tools/salesforce/model.py,sha256=wzpbTdUx5mANApAZFQIKzq7xXtYBiiSlKvrTX
 alita_sdk/tools/servicenow/__init__.py,sha256=VHH3qpUbEJ0tdtrIiWakohCmbifUOPgCVXERN3jRwsU,4234
 alita_sdk/tools/servicenow/api_wrapper.py,sha256=WpH-bBLGFdhehs4g-K-WAkNuaD1CSrwsDpdgB3RG53s,6120
 alita_sdk/tools/servicenow/servicenow_client.py,sha256=Rdqfu-ll-qbnclMzChLZBsfXRDzgoX_FdeI2WLApWxc,3269
-alita_sdk/tools/sharepoint/__init__.py,sha256=JEKSMgpPxctjfLPkz0ndbUgasDGnU0D_-S6T6y8YcoY,3698
-alita_sdk/tools/sharepoint/api_wrapper.py,sha256=bLDWQWTpnOxNmO9IDz_H-UNZ1uHM5gbIHsZlRipTxNU,9110
+alita_sdk/tools/sharepoint/__init__.py,sha256=8ix4a4lWN6vpGS6bdcr4U7zdttk6ShgQ0ji4r7w3iKw,4073
+alita_sdk/tools/sharepoint/api_wrapper.py,sha256=YCD0DkdgTyrDs4cSOO_LQjTVfmoA8U_keGGJlefxSTo,9083
 alita_sdk/tools/sharepoint/authorization_helper.py,sha256=n-nL5dlBoLMK70nHu7P2RYCb8C6c9HMA_gEaw8LxuhE,2007
 alita_sdk/tools/sharepoint/utils.py,sha256=fZ1YzAu5CTjKSZeslowpOPH974902S8vCp1Wu7L44LM,446
 alita_sdk/tools/slack/__init__.py,sha256=mbP2JiHybGSAH0ay8pxvPCqeU2eb9CK_NaCKG1uhPE4,3894
@@ -282,11 +282,13 @@ alita_sdk/tools/sql/api_wrapper.py,sha256=Rky0_CX9HWDQ2mClHGAgP3LHjYVX4iymPuilZM
 alita_sdk/tools/sql/models.py,sha256=AKJgSl_kEEz4fZfw3kbvdGHXaRZ-yiaqfJOB6YOj3i0,641
 alita_sdk/tools/testio/__init__.py,sha256=qi12wyJXN02hrUXg08CbijcCL5pi30JMbJfiXjn1Zr0,2646
 alita_sdk/tools/testio/api_wrapper.py,sha256=BvmL5h634BzG6p7ajnQLmj-uoAw1gjWnd4FHHu1h--Q,21638
-alita_sdk/tools/testrail/__init__.py,sha256=sYDPE4wOv24dt8K6g57fNBd4toLZSrrhqM5g8sDlTDI,4445
+alita_sdk/tools/testrail/__init__.py,sha256=VEoFDsMgzMapOaifdlEL_S_YxzD0CjXokGYvqvfPMNw,4842
 alita_sdk/tools/testrail/api_wrapper.py,sha256=Aax0jspgidXYNxLIw6qTWu3dO2JOIS0ALIqsCzQuFbQ,32087
 alita_sdk/tools/utils/__init__.py,sha256=155xepXPr4OEzs2Mz5YnjXcBpxSv1X2eznRUVoPtyK0,3268
 alita_sdk/tools/utils/content_parser.py,sha256=j2eDFhLCPNw61lnt8uaJLO7-Md3qWlKjOejou5JGKdg,7327
-alita_sdk/tools/xray/__init__.py,sha256=oEfC64wuUrxv56UE8z5iksl76pN7hJJfJ9mKuhaHZEc,3648
+alita_sdk/tools/vector_adapters/VectorStoreAdapter.py,sha256=bsUFx-P3I5Rfzpk-mCrabhLtt3motAm02GaRK9C-WXA,3260
+alita_sdk/tools/vector_adapters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+alita_sdk/tools/xray/__init__.py,sha256=OYa1wveTm-lAhsJaGXMnwOrDQWl6ch--NjNLBeR63eM,4331
 alita_sdk/tools/xray/api_wrapper.py,sha256=A8PJmY2k7TowaD_vk6ZxkMnSUoZUt9A6g4TJrZfNTAw,32225
 alita_sdk/tools/yagmail/__init__.py,sha256=c4Qn3em0tLxzRmFKpzbBgY9W2EnOoKf0azoDJHng5CY,2208
 alita_sdk/tools/yagmail/yagmail_wrapper.py,sha256=SKoGVd1X4Ew3ad5tOdtPoY00M6jStNdT3q7GXEjQc5g,1952
@@ -294,19 +296,19 @@ alita_sdk/tools/zephyr/Zephyr.py,sha256=ODZbg9Aw0H0Rbv-HcDXLI4KHbPiLDHoteDofshw9
 alita_sdk/tools/zephyr/__init__.py,sha256=8B2Ibz5QTmB5WkV0q8Sq4kuj92FFaFWZLrT877zRRLg,2897
 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=lWnOuVmva8vWBSlnk-wv40oBowxJDXa7iumsiXACcA0,3511
-alita_sdk/tools/zephyr_enterprise/api_wrapper.py,sha256=Ir3zHljhbZQJRJJQOBzS_GL5xvxb3-Vq5VF8XIMkxck,9348
+alita_sdk/tools/zephyr_enterprise/__init__.py,sha256=XAWLj4WgMpEz3xnP8VW1lzHVW48K3CxIXdgeubjMvaw,3885
+alita_sdk/tools/zephyr_enterprise/api_wrapper.py,sha256=arPAvOGrurEsZaICvdVrcbspXBdDQl4XuNdfZSmm1lg,11204
 alita_sdk/tools/zephyr_enterprise/zephyr_enterprise.py,sha256=hV9LIrYfJT6oYp-ZfQR0YHflqBFPsUw2Oc55HwK0H48,6809
-alita_sdk/tools/zephyr_essential/__init__.py,sha256=LYLF9imlfeuW8KZDGLeENWGXT71x7WWDv-Ss8rG2v8Q,3256
-alita_sdk/tools/zephyr_essential/api_wrapper.py,sha256=ksg-2j_w74pt2pdoWuuSU-gF3E6IlNtv4wxPf8sJMWg,36812
+alita_sdk/tools/zephyr_essential/__init__.py,sha256=-VS-smytUSjqNLLi6HmqbsU3NfmNFTrzPjzNiEtOi5c,3628
+alita_sdk/tools/zephyr_essential/api_wrapper.py,sha256=zjZnAx1Fl1fYx5Vc1Q8JOc_pxUEkCngbRzzmea4sj7Y,36785
 alita_sdk/tools/zephyr_essential/client.py,sha256=bfNcUKNqj9MFWTludGbbqD4qZlxrBaC2JtWsCfZMqSY,9722
 alita_sdk/tools/zephyr_scale/__init__.py,sha256=2NTcdrfkx4GSegqyXhsPLsEpc4FlACuDy85b0fk6cAo,4572
-alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=9CzQqQKv45LqZCmwSe4zzEXvBtStIcGQ1b51EQG5rhQ,78032
+alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=ZpHVRpeWDE-kxOf491vHgNesozuwEVS4oB0cnPs4XMw,78005
 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.221.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.221.dist-info/METADATA,sha256=zuvtQKT7XslCi0pc408eocmlMiXws7sCY2IpTauO-wQ,18917
-alita_sdk-0.3.221.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.221.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.221.dist-info/RECORD,,
+alita_sdk-0.3.223.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.223.dist-info/METADATA,sha256=G8Oorskh9yb8AK6-VSNMDROWc51kjn1_EkR_rfvlUf4,18917
+alita_sdk-0.3.223.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.223.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.223.dist-info/RECORD,,