alita-sdk 0.3.376__py3-none-any.whl → 0.3.377__py3-none-any.whl

alita_sdk/runtime/tools/vectorstore.py

@@ -215,7 +215,7 @@ class VectorStoreWrapper(BaseToolApiWrapper):
         """List all collections in the vectorstore.
         Returns a list of collection names, or if no collections exist,
         returns a dict with an empty list and a message."""
-        raw = self.vector_adapter.list_indexes(self)
+        raw = self.vector_adapter.list_collections(self)
         # Normalize raw result to a list of names
         if not raw:
             # No collections found

alita_sdk/runtime/tools/vectorstore_base.py

@@ -208,10 +208,10 @@ class VectorStoreWrapperBase(BaseToolApiWrapper):
             logger.error(f"Error during similarity search: {str(e)}")
             raise ToolException(f"Search failed: {str(e)}")
 
-    def list_indexes(self) -> List[str]:
+    def list_collections(self) -> List[str]:
         """List all collections in the vectorstore."""
 
-        collections = self.vector_adapter.list_indexes(self)
+        collections = self.vector_adapter.list_collections(self)
         if not collections:
             return "No indexed collections"
         return collections

alita_sdk/tools/base_indexer_toolkit.py

@@ -343,7 +343,7 @@ class BaseIndexerToolkit(VectorStoreWrapperBase):
         """Cleans the indexed data in the collection."""
         super()._clean_collection(index_name=index_name)
         return (f"Collection '{index_name}' has been removed from the vector store.\n"
-                f"Available collections: {self.list_indexes()}") if index_name \
+                f"Available collections: {self.list_collections()}") if index_name \
             else "All collections have been removed from the vector store." 
 
     def _build_collection_filter(self, filter: dict | str, index_name: str = "") -> dict:
@@ -385,7 +385,7 @@ class BaseIndexerToolkit(VectorStoreWrapperBase):
         """ Searches indexed documents in the vector store."""
         # build filter on top of index_name
 
-        available_collections = super().list_indexes()
+        available_collections = super().list_collections()
         if index_name and index_name not in available_collections:
             return f"Collection '{index_name}' not found. Available collections: {available_collections}"
 
@@ -547,10 +547,10 @@ class BaseIndexerToolkit(VectorStoreWrapperBase):
                 "args_schema": RemoveIndexParams
             },
             {
-                "name": "list_indexes",
-                "mode": "list_indexes",
-                "ref": self.list_indexes,
-                "description": self.list_indexes.__doc__,
+                "name": "list_collections",
+                "mode": "list_collections",
+                "ref": self.list_collections,
+                "description": self.list_collections.__doc__,
                 "args_schema": create_model("ListCollectionsParams")  # No parameters
             },
         ]

alita_sdk/tools/elitea_base.py

@@ -17,7 +17,7 @@ from ..runtime.utils.utils import IndexerKeywords
 
 logger = logging.getLogger(__name__)
 
-INDEX_TOOL_NAMES = ['index_data', 'remove_index', 'list_indexes', 'search_index', 'stepback_search_index',
+INDEX_TOOL_NAMES = ['index_data', 'remove_index', 'list_collections', 'search_index', 'stepback_search_index',
                             'stepback_summary_index']
 
 LoaderSchema = create_model(
@@ -403,9 +403,9 @@ class BaseVectorStoreToolApiWrapper(BaseToolApiWrapper):
         """Cleans the indexed data in the collection."""
         self._init_vector_store()._clean_collection(index_name=index_name)
         return (f"Collection '{index_name}' has been removed from the vector store.\n"
-                f"Available collections: {self.list_indexes()}")
+                f"Available collections: {self.list_collections()}")
 
-    def list_indexes(self):
+    def list_collections(self):
         """Lists all collections in the vector store."""
         vectorstore_wrapper = self._init_vector_store()
         return vectorstore_wrapper.list_collections()
@@ -538,9 +538,9 @@ class BaseVectorStoreToolApiWrapper(BaseToolApiWrapper):
             },
             {
                 "name": "list_indexes",
-                "mode": "list_indexes",
-                "ref": self.list_indexes,
-                "description": self.list_indexes.__doc__,
+                "mode": "list_collections",
+                "ref": self.list_collections,
+                "description": self.list_collections.__doc__,
                 "args_schema": create_model("ListCollectionsParams")  # No parameters
             },
 

alita_sdk/tools/vector_adapters/VectorStoreAdapter.py

@@ -307,7 +307,7 @@ class ChromaAdapter(VectorStoreAdapter):
 
     def list_collections(self, vectorstore_wrapper) -> str:
         vector_client = vectorstore_wrapper.vectorstore._client
-        return ','.join([collection.name for collection in vector_client.list_indexes()])
+        return ','.join([collection.name for collection in vector_client.list_collections()])
 
     def remove_collection(self, vectorstore_wrapper, collection_name: str):
         vectorstore_wrapper.vectorstore.delete_collection()

{alita_sdk-0.3.376.dist-info → alita_sdk-0.3.377.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.376
+Version: 0.3.377
 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 <lifedj27@gmail.com>, Artem Dubrovskiy <ad13box@gmail.com>
 License-Expression: Apache-2.0

{alita_sdk-0.3.376.dist-info → alita_sdk-0.3.377.dist-info}/RECORD

@@ -123,8 +123,8 @@ alita_sdk/runtime/tools/prompt.py,sha256=nJafb_e5aOM1Rr3qGFCR-SKziU9uCsiP2okIMs9
 alita_sdk/runtime/tools/router.py,sha256=p7e0tX6YAWw2M2Nq0A_xqw1E2P-Xz1DaJvhUstfoZn4,1584
 alita_sdk/runtime/tools/sandbox.py,sha256=0OjCNsDVO1N0cFNEFVr6GVICSaqGWesUzF6LcYg-Hn0,11349
 alita_sdk/runtime/tools/tool.py,sha256=lE1hGi6qOAXG7qxtqxarD_XMQqTghdywf261DZawwno,5631
-alita_sdk/runtime/tools/vectorstore.py,sha256=iRiB1VDDrgEJ4JmgsPtbkY6qldDfRci7spcikLFdY_U,34323
-alita_sdk/runtime/tools/vectorstore_base.py,sha256=7F_kpDkuVvkurCS2Z5oMKWnVjbzmli-D-gz3jXVCrVI,28244
+alita_sdk/runtime/tools/vectorstore.py,sha256=FsnxdnvMK5bUEFxz0eeSHeNpVOk2gxOeXjoSlvCo8rs,34327
+alita_sdk/runtime/tools/vectorstore_base.py,sha256=lNz6bOMpHOY8JiHT7BkoDbyj3kLykcKlCx4zOu_IgPE,28252
 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
@@ -136,9 +136,9 @@ 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=BVEVLkYiiotcUD0XsHyx-wACpHfALsQg7PLZpObqvK8,1008
 alita_sdk/tools/__init__.py,sha256=jUj1ztC2FbkIUB-YYmiqaz_rqW7Il5kWzDPn1mJmj5w,10545
-alita_sdk/tools/base_indexer_toolkit.py,sha256=ZYQONHUjrpK_UBd2HU14WpE0QYECKSnzDD32ey_Xc4A,26608
+alita_sdk/tools/base_indexer_toolkit.py,sha256=_Q8K5fVg5gBFlZ94wYvBrPWJ56VobnNrfr15Knyobu4,26632
 alita_sdk/tools/code_indexer_toolkit.py,sha256=p3zVnCnQTUf7JUGra9Rl6GEK2W1-hvvz0Xsgz0v0muM,7292
-alita_sdk/tools/elitea_base.py,sha256=jbgWUXHif0CS2dTZj87cPX_17dGk_WvYkjDwn1x4AwA,34389
+alita_sdk/tools/elitea_base.py,sha256=nV4sNVctJGgLqWTrqkI2iMc07k69GQ6uF5hXrLmsshg,34413
 alita_sdk/tools/non_code_indexer_toolkit.py,sha256=6Lrqor1VeSLbPLDHAfg_7UAUqKFy1r_n6bdsc4-ak98,1315
 alita_sdk/tools/ado/__init__.py,sha256=NnNYpNFW0_N_v1td_iekYOoQRRB7PIunbpT2f9ZFJM4,1201
 alita_sdk/tools/ado/utils.py,sha256=PTCludvaQmPLakF2EbCGy66Mro4-rjDtavVP-xcB2Wc,1252
@@ -332,7 +332,7 @@ alita_sdk/tools/testrail/api_wrapper.py,sha256=tQcGlFJmftvs5ZiO4tsP19fCo4CrJeq_U
 alita_sdk/tools/utils/__init__.py,sha256=xB9OQgW65DftadrSpoAAitnEIbIXZKBOCji0NDe7FRM,3923
 alita_sdk/tools/utils/available_tools_decorator.py,sha256=IbrdfeQkswxUFgvvN7-dyLMZMyXLiwvX7kgi3phciCk,273
 alita_sdk/tools/utils/content_parser.py,sha256=7ohj8HeL_-rmc-Fv0TS8IpxIQC8tOpfuhyT3XlWx-gQ,15368
-alita_sdk/tools/vector_adapters/VectorStoreAdapter.py,sha256=TmMgrvmUUP7pWAH0iEaHkMH6fwNsU_X44AFmotqXaHY,19015
+alita_sdk/tools/vector_adapters/VectorStoreAdapter.py,sha256=eDezsk41b5ql3CISQ6Xk-qE3foO-PjY0VSWeZnVxHPE,19019
 alita_sdk/tools/vector_adapters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/tools/xray/__init__.py,sha256=eOMWP8VamFbbJgt1xrGpGPqB9ByOTA0Cd3LCaETzGk4,4376
 alita_sdk/tools/xray/api_wrapper.py,sha256=uj5kzUgPdo_Oct9WCNMOpkb6o_3L7J4LZrEGtrwYMmc,30157
@@ -353,8 +353,8 @@ alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=kT0TbmMvuKhDUZc0i7KO18O38JM9S
 alita_sdk/tools/zephyr_squad/__init__.py,sha256=0ne8XLJEQSLOWfzd2HdnqOYmQlUliKHbBED5kW_Vias,2895
 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.376.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.376.dist-info/METADATA,sha256=zOktQPOXmbGzvlxCp6JnqppE-1adGSuHWO8GuBSA0mk,19071
-alita_sdk-0.3.376.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.376.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.376.dist-info/RECORD,,
+alita_sdk-0.3.377.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.377.dist-info/METADATA,sha256=rUHtuRzi9cHmWrlCZLzi2LfWfbJ-23kMyK7nP2Y3KRs,19071
+alita_sdk-0.3.377.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.377.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.377.dist-info/RECORD,,