alita-sdk 0.3.222__tar.gz → 0.3.224__tar.gz

{alita_sdk-0.3.222 → alita_sdk-0.3.224}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.222
+Version: 0.3.224
 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.222 → alita_sdk-0.3.224}/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
         
@@ -705,9 +705,8 @@ class VectorStoreWrapper(BaseToolApiWrapper):
             combined_items = [item for item in combined_items if abs(item[1]) >= cut_off]
         
         # Sort by score and limit results
-
-        # for chroma we want ascending order (lower score is better), for others descending
-        combined_items.sort(key=lambda x: x[1], reverse= self.vectorstore_type.lower() != 'chroma')
+        # DISABLED: for chroma we want ascending order (lower score is better), for others descending
+        # combined_items.sort(key=lambda x: x[1], reverse= self.vectorstore_type.lower() != 'chroma')
         combined_items = combined_items[:search_top]
 
         # Format output based on doctype

{alita_sdk-0.3.222 → alita_sdk-0.3.224}/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-0.3.222 → alita_sdk-0.3.224}/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-0.3.222 → alita_sdk-0.3.224}/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-0.3.222 → alita_sdk-0.3.224}/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-0.3.222 → alita_sdk-0.3.224}/alita_sdk/tools/elitea_base.py

@@ -60,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(
@@ -112,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")),

{alita_sdk-0.3.222 → alita_sdk-0.3.224}/alita_sdk/tools/utils/content_parser.py

@@ -77,6 +77,10 @@ def parse_file_content(file_name=None, file_content=None, is_capture_image: bool
     extension = Path(file_path if file_path else file_name).suffix
 
     loader_object = loaders_map.get(extension)
+    if not loader_object:
+        raise ToolException(
+            f"Not supported type ({extension}) of files entered. "
+            f"Supported types are TXT, DOCX, PDF, PPTX, XLSX and XLS only.")
     loader_kwargs = loader_object['kwargs']
     loader_kwargs.update({
         "file_path": file_path,

{alita_sdk-0.3.222 → alita_sdk-0.3.224}/alita_sdk.egg-info/PKG-INFO

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

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