alita-sdk 0.3.310__tar.gz → 0.3.311__tar.gz

{alita_sdk-0.3.310 → alita_sdk-0.3.311}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.310
+Version: 0.3.311
 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.310 → alita_sdk-0.3.311}/alita_sdk/runtime/toolkits/tools.py

@@ -176,7 +176,7 @@ def _init_single_mcp_tool(toolkit_name, available_tool, alita, toolkit_settings)
         tool_name = available_tool["name"]
         return McpServerTool(
             name=tool_name,
-            description=available_tool.get("description", ""),
+            description=f"MCP for a tool '{tool_name}': {available_tool.get("description", "")}",
             args_schema=McpServerTool.create_pydantic_model_from_schema(
                 available_tool.get("inputSchema", {})
             ),

{alita_sdk-0.3.310 → alita_sdk-0.3.311}/alita_sdk/tools/base_indexer_toolkit.py

@@ -161,7 +161,7 @@ class BaseIndexerToolkit(VectorStoreWrapperBase):
         self._log_tool_event(f"Indexing data into collection with suffix '{collection_suffix}'. It can take some time...")
         self._log_tool_event(f"Loading the documents to index...{kwargs}")
         documents = self._base_loader(**kwargs)
-        self._log_tool_event(f"Base documents were loaded. "
+        self._log_tool_event(f"Base documents were pre-loaded. "
                              f"Search for possible document duplicates and remove them from the indexing list...")
         documents = self._reduce_duplicates(documents, collection_suffix)
         self._log_tool_event(f"Duplicates were removed. "
@@ -216,6 +216,8 @@ class BaseIndexerToolkit(VectorStoreWrapperBase):
 
     def _collect_dependencies(self, documents: Generator[Document, None, None]):
         for document in documents:
+            self._log_tool_event(message=f"Collecting the dependencies for document ID "
+                                         f"'{document.metadata.get('id', 'N/A')}' to collect dependencies if any...")
             dependencies = self._process_document(document)
             yield document
             for dep in dependencies:

{alita_sdk-0.3.310 → alita_sdk-0.3.311}/alita_sdk/tools/confluence/api_wrapper.py

@@ -848,7 +848,8 @@ class ConfluenceAPIWrapper(BaseVectorStoreToolApiWrapper):
             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', {}))
+            # TODO need to retrive content from other fields/sources if both attachment.get('content', '') and attachment.get('llm_analysis', '') are None
+            yield Document(page_content=attachment.get('content', '') or attachment.get('llm_analysis', '') or '', metadata=attachment.get('metadata', {}))
 
     def _download_image(self, image_url):
         """

{alita_sdk-0.3.310 → alita_sdk-0.3.311}/alita_sdk/tools/figma/api_wrapper.py

@@ -301,6 +301,12 @@ class FigmaApiWrapper(NonCodeIndexerToolkit):
             figma_pages = [node for node in figma_pages if ('id' in node and node['id'].replace(':', '-') in node_ids_include)]
         elif node_ids_exclude:
             figma_pages = [node for node in figma_pages if ('id' in node and node['id'].replace(':', '-') not in node_ids_exclude)]
+
+        # if node_types_include is not provided, default to 'frame'
+        # to avoid downloading too many images and nodes which co=annot be rendered as images
+        if not node_types_include:
+            node_types_include = ['frame']
+
         node_ids = [
             child['id']
             for page in figma_pages
@@ -371,7 +377,7 @@ class FigmaApiWrapper(NonCodeIndexerToolkit):
                 default=None)),
             'node_types_include': (Optional[List[str]], Field(
                 description="List type of nodes to include in index: i.e. ['FRAME', 'COMPONENT', 'RECTANGLE', 'COMPONENT_SET', 'INSTANCE', 'VECTOR', ...].",
-                default=None)),
+                default=['frame'])),
             'node_types_exclude': (Optional[List[str]], Field(
                 description="List type of nodes to exclude from index. It is applied only if node_types_include is not provided: i.e. ['FRAME', 'COMPONENT', 'RECTANGLE', 'COMPONENT_SET', 'INSTANCE', 'VECTOR', ...]",
                 default=None))

{alita_sdk-0.3.310 → alita_sdk-0.3.311}/alita_sdk/tools/memory/__init__.py

@@ -61,7 +61,7 @@ class MemoryToolkit(BaseToolkit):
         return create_model(
             'memory',
             namespace=(str, Field(description="Memory namespace", json_schema_extra={'toolkit_name': True})),
-            pgvector_configuration=(Optional[PgVectorConfiguration], Field(description="PgVector Configuration",
+            pgvector_configuration=(PgVectorConfiguration, Field(description="PgVector Configuration",
                                                                            json_schema_extra={
                                                                                'configuration_types': ['pgvector']})),
             selected_tools=(List[Literal[tuple(selected_tools)]],
@@ -79,7 +79,7 @@ class MemoryToolkit(BaseToolkit):
         )
 
     @classmethod
-    def get_toolkit(cls, namespace: str, store=None):
+    def get_toolkit(cls, namespace: str, store=None, **kwargs):
         """
         Get toolkit with memory tools.
         
@@ -95,6 +95,12 @@ class MemoryToolkit(BaseToolkit):
                 "PostgreSQL dependencies (psycopg) are required for MemoryToolkit. "
                 "Install with: pip install psycopg[binary]"
             )
+
+        if store is None:
+            # The store is not provided, attempt to create it from configuration
+            from ...runtime.langchain.store_manager import get_manager
+            conn_str = (kwargs.get('pgvector_configuration') or {}).get('connection_string', '')
+            store = get_manager().get_store(conn_str)
         
         # Validate store type
         if store is not None and not isinstance(store, PostgresStore):

{alita_sdk-0.3.310 → alita_sdk-0.3.311}/alita_sdk.egg-info/PKG-INFO

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

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