alita-sdk 0.3.227__py3-none-any.whl → 0.3.228__py3-none-any.whl

alita_sdk/runtime/toolkits/tools.py

@@ -75,7 +75,7 @@ def get_tools(tools_list: list, alita_client, llm, memory_store: BaseStore = Non
                 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=tool.get('toolkit_name'),
                 embedding_model=tool['settings'].get('embedding_model', None),
                 embedding_model_params=tool['settings'].get('embedding_model_params', None),
                 vectorstore_type="PGVector"

alita_sdk/tools/ado/__init__.py

@@ -15,13 +15,13 @@ def get_tools(tool_type, tool):
         "limit": tool['settings'].get('limit', 5),
         "toolkit_name": tool.get('toolkit_name', ''),
         # indexer settings
-        "llm":tool['settings'].get('llm', None),
-        "connection_string":tool['settings'].get('connection_string', None),
-        "collection_name":str(tool['id']),
-        "doctype":'doc',
-        "embedding_model":"HuggingFaceEmbeddings",
-        "embedding_model_params":{"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
-        "vectorstore_type":"PGVector"
+        "llm": tool['settings'].get('llm', None),
+        "connection_string": tool['settings'].get('connection_string', None),
+        "collection_name": tool['toolkit_name'],
+        "doctype": 'doc',
+        "embedding_model": "HuggingFaceEmbeddings",
+        "embedding_model_params": {"model_name": "sentence-transformers/all-MiniLM-L6-v2"},
+        "vectorstore_type": "PGVector"
     }
     if tool_type == 'ado_plans':
         return AzureDevOpsPlansToolkit().get_toolkit(**config_dict).get_tools()

alita_sdk/tools/ado/repos/__init__.py

@@ -23,7 +23,7 @@ def _get_toolkit(tool) -> BaseToolkit:
         active_branch=tool['settings'].get('active_branch', ""),
         toolkit_name=tool['settings'].get('toolkit_name', ""),
         connection_string=tool['settings'].get('connection_string', None),
-        collection_name=str(tool['id']),
+        collection_name=tool['toolkit_name'],
         doctype='code',
         embedding_model="HuggingFaceEmbeddings",
         embedding_model_params={"model_name": "sentence-transformers/all-MiniLM-L6-v2"},

alita_sdk/tools/ado/wiki/ado_wrapper.py

@@ -1,3 +1,4 @@
+import hashlib
 import logging
 from typing import Any, Dict, Generator, Optional
 
@@ -229,21 +230,26 @@ class AzureDevOpsApiWrapper(BaseVectorStoreToolApiWrapper):
             logger.error(f"Unable to modify wiki page: {str(e)}")
             return ToolException(f"Unable to modify wiki page: {str(e)}")
 
-    def _base_loader(self, wiki_identifier: str, **kwargs) -> Generator[Document, None, None]:
+    def _base_loader(self, wiki_identifier: str, title_contains: Optional[str] = None, **kwargs) -> Generator[Document, None, None]:
         pages = self._client.get_pages_batch(pages_batch_request={}, project=self.project, wiki_identifier=wiki_identifier)
         #
         for page in pages:
             content = self._client.get_page_by_id(project=self.project, wiki_identifier=wiki_identifier, id=page.id, include_content=True).page.content
-            yield Document(page_content=content, metadata={
-                'id': page.id, 
-                'path': page.path,
-                'updated_on': ''
-            })
+            content_hash = hashlib.sha256(content.encode("utf-8")).hexdigest()
+            title = page.path.rsplit("/", 1)[-1]
+            if not title_contains or (title_contains and title_contains.lower() in title.lower()):
+                yield Document(page_content=content, metadata={
+                    'id': str(page.id),
+                    'path': page.path,
+                    'title': title,
+                    'updated_on': content_hash
+                })
 
     def _index_tool_params(self):
         """Return the parameters for indexing data."""
         return {
-            "wiki_identifier": (str, Field(description="Wiki identifier to index, e.g., 'ABCProject.wiki'"))
+            "wiki_identifier": (str, Field(description="Wiki identifier to index, e.g., 'ABCProject.wiki'")),
+            'title_contains': (Optional[str], Field(default=None, description="Optional filter to include only pages with titles containing exact this string"))
         }
 
     @extend_with_vector_tools

alita_sdk/tools/confluence/api_wrapper.py

@@ -214,8 +214,11 @@ class ConfluenceAPIWrapper(BaseVectorStoreToolApiWrapper):
                 "`atlassian` package not found, please run "
                 "`pip install atlassian-python-api`"
             )
-
+        if not values.get('base_url'):
+            raise ValueError("Base URL is required for Confluence API Wrapper.")
         url = values['base_url']
+        if not (values.get('token') or (values.get('api_key') and values.get('username'))):
+            raise ValueError("Either 'token' or both 'api_key' and 'username' must be provided for authentication.")
         api_key = values.get('api_key')
         username = values.get('username')
         token = values.get('token')
@@ -844,8 +847,12 @@ class ConfluenceAPIWrapper(BaseVectorStoreToolApiWrapper):
             yield document
 
     def _process_document(self, document: Document) -> Generator[Document, None, None]:
-        for attachment in self.get_page_attachments(document.metadata.get('id')):
-            yield Document(page_content=attachment.get('content', ''), metadata=attachment.get('metadata', {}))
+        attachments = self.get_page_attachments(document.metadata.get('id'))
+        if isinstance(attachments, str):
+            logger.info(f" {document.metadata.get('id')}: {attachments}")
+            return
+        for attachment in attachments:
+            yield Document(page_content=attachment.get('content', '') or attachment.get('llm_analysis', ''), metadata=attachment.get('metadata', {}))
 
     def _download_image(self, image_url):
         """
@@ -1577,7 +1584,7 @@ class ConfluenceAPIWrapper(BaseVectorStoreToolApiWrapper):
         """Return the parameters for indexing data."""
         return {
             "content_format": (Literal['view', 'storage', 'export_view', 'editor', 'anonymous'],
-                            Field(description="The format of the content to be retrieved.")),
+                            Field(description="The format of the content to be retrieved.", default='view')),
             "page_ids": (Optional[List[str]], Field(description="List of page IDs to retrieve.", default=None)),
             "label": (Optional[str], Field(description="Label to filter pages.", default=None)),
             "cql": (Optional[str], Field(description="CQL query to filter pages.", default=None)),

alita_sdk/tools/elitea_base.py

@@ -374,7 +374,7 @@ class BaseVectorStoreToolApiWrapper(BaseToolApiWrapper):
     def list_collections(self):
         """Lists all collections in the vector store."""
         vectorstore_wrapper = self._init_vector_store()
-        return self._adapter.list_collections(vectorstore_wrapper)
+        return self._adapter.list_collections(vectorstore_wrapper, self.collection_name or "")
 
     def search_index(self,
                      query: str,

alita_sdk/tools/vector_adapters/VectorStoreAdapter.py

@@ -35,15 +35,16 @@ class PGVectorAdapter(VectorStoreAdapter):
             "connection_string": connection_string
         }
 
-    def list_collections(self, vectorstore_wrapper) -> str:
+    def list_collections(self, vectorstore_wrapper, collection_name) -> 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';
+            get_collections = text(f"""
+                 SELECT table_schema
+                 FROM information_schema.columns
+                 WHERE udt_name = 'vector'
+                   AND table_schema LIKE '%{collection_name}%';
             """)
             result = session.execute(get_collections)
             docs = result.fetchall()

{alita_sdk-0.3.227.dist-info → alita_sdk-0.3.228.dist-info}/METADATA

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

@@ -64,7 +64,7 @@ alita_sdk/runtime/toolkits/artifact.py,sha256=_vQOR4_xPeyOd3Gqcf5vDxTu2uLzViYt7i
 alita_sdk/runtime/toolkits/datasource.py,sha256=qk78OdPoReYPCWwahfkKLbKc4pfsu-061oXRryFLP6I,2498
 alita_sdk/runtime/toolkits/prompt.py,sha256=WIpTkkVYWqIqOWR_LlSWz3ug8uO9tm5jJ7aZYdiGRn0,1192
 alita_sdk/runtime/toolkits/subgraph.py,sha256=ZYqI4yVLbEPAjCR8dpXbjbL2ipX598Hk3fL6AgaqFD4,1758
-alita_sdk/runtime/toolkits/tools.py,sha256=qSVGfVJXNMe_9x2bwWvlXPJe5Q89FtgJooqx5UPdlxM,8052
+alita_sdk/runtime/toolkits/tools.py,sha256=ebvVIyNzLMvqvgW58uP5qGmWV5bLczuKIr58OgladUQ,8029
 alita_sdk/runtime/toolkits/vectorstore.py,sha256=BGppQADa1ZiLO17fC0uCACTTEvPHlodEDYEzUcBRbAA,2901
 alita_sdk/runtime/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alita_sdk/runtime/tools/agent.py,sha256=m98QxOHwnCRTT9j18Olbb5UPS8-ZGeQaGiUyZJSyFck,3162
@@ -94,15 +94,15 @@ 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=AmOks-hZfHFI2G0_fBh7Yyz3wjohyTA1sLxLEqW87EQ,30544
-alita_sdk/tools/ado/__init__.py,sha256=2NMQwt2pjIukSC9nSZ7CLocdGpK7002x7ixKr_wunxk,1313
+alita_sdk/tools/elitea_base.py,sha256=22P97EmUlO-eC8DoGPiSNjHNjTv8ru0Q05qnR3U9WWs,30572
+alita_sdk/tools/ado/__init__.py,sha256=j4W52bLXVMrm-mZuyD1r6obw_EjRXiLKG7mFDOs8Cpw,1325
 alita_sdk/tools/ado/utils.py,sha256=PTCludvaQmPLakF2EbCGy66Mro4-rjDtavVP-xcB2Wc,1252
-alita_sdk/tools/ado/repos/__init__.py,sha256=VvOapCG5okdyfCNdeTfOdcgRvl54cB1-sx_BF0TBWOw,6858
+alita_sdk/tools/ado/repos/__init__.py,sha256=wdMYwkWfxZ9gz-5NKisG1QnwG_9c_s7JsjcM51i-spo,6863
 alita_sdk/tools/ado/repos/repos_wrapper.py,sha256=SgnEUX1lTn5ncZI8EFfPibk0_gHstkApDSqTCxYsrUk,50097
 alita_sdk/tools/ado/test_plan/__init__.py,sha256=QUNSOCVheUbILve4CxjBzBHKcIA28jZjoA6rRb8tD_Y,5483
 alita_sdk/tools/ado/test_plan/test_plan_wrapper.py,sha256=WWSWQx97sQfkM_CALybj9BPmPelozdRlxwefbkTgxis,20185
 alita_sdk/tools/ado/wiki/__init__.py,sha256=3cPNqUyOFwnZ2g5QW8wsdBC9sB0u39F-q4KKVjQqTN8,5351
-alita_sdk/tools/ado/wiki/ado_wrapper.py,sha256=HoM2U2G8sKM8Hh6HpTFHdTiKAu4vLigB8VwE0_QtMhw,14167
+alita_sdk/tools/ado/wiki/ado_wrapper.py,sha256=8UIuEgn2G3vbyLIdsI-iwX-zXJXzHENuTAih5USYGrs,14679
 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
@@ -186,7 +186,7 @@ alita_sdk/tools/code/loaders/codesearcher.py,sha256=XoXXZtIQZhvjIwZlnl_4wVGHC-3s
 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=Osn9Sd8dURk8zms6J6MDYtH-MAQaXcI21MW_MOZS9pE,7293
-alita_sdk/tools/confluence/api_wrapper.py,sha256=VR0TEMJPqe3ylzl_DuFSZ3LS0oTIJ7SufjifRXIYEhg,84507
+alita_sdk/tools/confluence/api_wrapper.py,sha256=4WqjVeFWyFeb4-VD5v4_J69pbyjire4Op7cBSKU9EXw,85057
 alita_sdk/tools/confluence/loader.py,sha256=4bf5qrJMEiJzuZp2NlxO2XObLD1w7fxss_WyMUpe8sg,9290
 alita_sdk/tools/confluence/utils.py,sha256=Lxo6dBD0OlvM4o0JuK6qeB_4LV9BptiwJA9e1vqNcDw,435
 alita_sdk/tools/custom_open_api/__init__.py,sha256=9aT5SPNPWcJC6jMZEM-3rUCXVULj_3-qJLQKmnreKNo,2537
@@ -286,7 +286,7 @@ alita_sdk/tools/testrail/__init__.py,sha256=VEoFDsMgzMapOaifdlEL_S_YxzD0CjXokGYv
 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=5meTV5nbhlaeZG2lL6OVY8XMaTVg2T2dgWZrtIjRLL8,7526
-alita_sdk/tools/vector_adapters/VectorStoreAdapter.py,sha256=bsUFx-P3I5Rfzpk-mCrabhLtt3motAm02GaRK9C-WXA,3260
+alita_sdk/tools/vector_adapters/VectorStoreAdapter.py,sha256=wZ_MhVWPEgoPhKRvbPB6Qs34EPyWx0IW7ydY_GVBZFQ,3344
 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
@@ -307,8 +307,8 @@ alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=JAeWf-RXohsxheUpT0iMDClc_izj-
 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.227.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.227.dist-info/METADATA,sha256=Rdq53_hxX1XkPsuzWiYs9_74Sxo6BzniJu9PvooRS9A,18917
-alita_sdk-0.3.227.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.227.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.227.dist-info/RECORD,,
+alita_sdk-0.3.228.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.228.dist-info/METADATA,sha256=HZtNzXNfu495HqKx7PxLBBjbD8rJ5iXHtZYO32XI3oU,18917
+alita_sdk-0.3.228.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.228.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.228.dist-info/RECORD,,