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

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/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.dist-info → alita_sdk-0.3.223.dist-info}/METADATA

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

@@ -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=cDJ4PDvhrqIpgkheKBOPPN-hUZT4-AJLxp2ankbiAmo,30477
+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
@@ -307,8 +307,8 @@ alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=ZpHVRpeWDE-kxOf491vHgNesozuwE
 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.222.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.222.dist-info/METADATA,sha256=DOZdr2MgwJLF1TIA4wp3BlWWTIo9rq1im3G_1U8Sv_U,18917
-alita_sdk-0.3.222.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.222.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.222.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,,