alita-sdk 0.3.295__tar.gz → 0.3.296__tar.gz

{alita_sdk-0.3.295 → alita_sdk-0.3.296}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.295
+Version: 0.3.296
 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.295 → alita_sdk-0.3.296}/alita_sdk/runtime/langchain/assistant.py

@@ -125,7 +125,7 @@ class Assistant:
         if not memory_tool or self.store is not None:
             return
         from .store_manager import get_manager
-        conn_str = memory_tool.get('settings', {}).get('connection_string', '')
+        conn_str = memory_tool['settings'].get('pgvector_configuration', {}).get('connection_string', '')
         store = get_manager().get_store(conn_str)
         self.store = store
 

alita_sdk-0.3.296/alita_sdk/runtime/langchain/document_loaders/AlitaPythonLoader.py

@@ -0,0 +1,9 @@
+from pathlib import Path
+from typing import Union
+from langchain_community.document_loaders.python import PythonLoader
+
+class AlitaPythonLoader(PythonLoader):
+    """Load `Python` files, respecting any non-default encoding if specified."""
+
+    def __init__(self, file_path: Union[str, Path], **kwargs):
+        super().__init__(file_path)

{alita_sdk-0.3.295 → alita_sdk-0.3.296}/alita_sdk/runtime/langchain/document_loaders/constants.py

@@ -25,6 +25,7 @@ from .AlitaPDFLoader import AlitaPDFLoader
 from .AlitaPowerPointLoader import AlitaPowerPointLoader
 from .AlitaTextLoader import AlitaTextLoader
 from .AlitaMarkdownLoader import AlitaMarkdownLoader
+from .AlitaPythonLoader import AlitaPythonLoader
 from enum import Enum
 
 class LoaderProperties(Enum):
@@ -190,7 +191,7 @@ loaders_map = {
         'allowed_to_override': ['max_tokens', 'mode', LoaderProperties.LLM.value, LoaderProperties.PROMPT.value, LoaderProperties.PROMPT_DEFAULT.value]
     },
     '.py': {
-        'class': PythonLoader,
+        'class': AlitaPythonLoader,
         'is_multimodal_processing': False,
         'kwargs': {},
         'allowed_to_override': ['max_tokens']

{alita_sdk-0.3.295 → alita_sdk-0.3.296}/alita_sdk/runtime/toolkits/tools.py

@@ -64,7 +64,6 @@ def get_tools(tools_list: list, alita_client, llm, memory_store: BaseStore = Non
             tools += MemoryToolkit.get_toolkit(
                 namespace=tool['settings'].get('namespace', str(tool['id'])),
                 store=memory_store,
-                toolkit_name=tool.get('toolkit_name', '')
             ).get_tools()
         elif tool['type'] == 'artifact':
             tools.extend(ArtifactToolkit.get_toolkit(

{alita_sdk-0.3.295 → alita_sdk-0.3.296}/alita_sdk/tools/__init__.py

@@ -79,7 +79,6 @@ _safe_import_tool('carrier', 'carrier', 'get_tools', 'AlitaCarrierToolkit')
 _safe_import_tool('ocr', 'ocr', 'get_tools', 'OCRToolkit')
 _safe_import_tool('pptx', 'pptx', 'get_tools', 'PPTXToolkit')
 _safe_import_tool('postman', 'postman', 'get_tools', 'PostmanToolkit')
-_safe_import_tool('memory', 'memory', 'get_tools', 'MemoryToolkit')
 _safe_import_tool('zephyr_squad', 'zephyr_squad', 'get_tools', 'ZephyrSquadToolkit')
 _safe_import_tool('zephyr_essential', 'zephyr_essential', 'get_tools', 'ZephyrEssentialToolkit')
 _safe_import_tool('slack', 'slack', 'get_tools', 'SlackToolkit')

{alita_sdk-0.3.295 → alita_sdk-0.3.296}/alita_sdk/tools/memory/__init__.py

@@ -1,7 +1,9 @@
-from typing import Optional, List
+from typing import Optional, List, Literal
 
 from langchain_core.tools import BaseToolkit, BaseTool
 
+from alita_sdk.configurations.pgvector import PgVectorConfiguration
+
 try:
     from langmem import create_manage_memory_tool, create_search_memory_tool
 except ImportError:
@@ -15,7 +17,7 @@ from pydantic import create_model, BaseModel, ConfigDict, Field, SecretStr
 
 name = "memory"
 
-def get_tools(tools_list: list, alita_client, llm, memory_store=None):
+def get_tools(tools_list: list, memory_store=None):
     """
     Get memory tools for the provided tool configurations.
     
@@ -53,13 +55,18 @@ class MemoryToolkit(BaseToolkit):
 
     @staticmethod
     def toolkit_config_schema() -> BaseModel:
+        memory_tools = [create_manage_memory_tool('test'), create_search_memory_tool('test')]
+        selected_tools = {x.name: x.args_schema.schema() for x in memory_tools}
+
         return create_model(
-            'MemoryConfig',
+            'memory',
             namespace=(str, Field(description="Memory namespace", json_schema_extra={'toolkit_name': True})),
-            # username=(Optional[str], Field(description="Username", default='Tester', json_schema_extra={'hidden': True})),
-            # connection_string=(Optional[SecretStr], Field(description="Connection string for vectorstore",
-            #                                               default=None,
-            #                                               json_schema_extra={'secret': True})),
+            pgvector_configuration=(Optional[PgVectorConfiguration], Field(description="PgVector Configuration",
+                                                                           json_schema_extra={
+                                                                               'configuration_types': ['pgvector']})),
+            selected_tools=(List[Literal[tuple(selected_tools)]],
+                            Field(default=[], json_schema_extra={'args_schemas': selected_tools})),
+
             __config__=ConfigDict(json_schema_extra={
                 'metadata': {
                     "label": "Memory",
@@ -72,7 +79,7 @@ class MemoryToolkit(BaseToolkit):
         )
 
     @classmethod
-    def get_toolkit(cls, namespace: str, store=None, **kwargs):
+    def get_toolkit(cls, namespace: str, store=None):
         """
         Get toolkit with memory tools.
         

{alita_sdk-0.3.295 → alita_sdk-0.3.296}/alita_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.295
+Version: 0.3.296
 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.295 → alita_sdk-0.3.296}/alita_sdk.egg-info/SOURCES.txt

@@ -72,6 +72,7 @@ alita_sdk/runtime/langchain/document_loaders/AlitaJiraLoader.py
 alita_sdk/runtime/langchain/document_loaders/AlitaMarkdownLoader.py
 alita_sdk/runtime/langchain/document_loaders/AlitaPDFLoader.py
 alita_sdk/runtime/langchain/document_loaders/AlitaPowerPointLoader.py
+alita_sdk/runtime/langchain/document_loaders/AlitaPythonLoader.py
 alita_sdk/runtime/langchain/document_loaders/AlitaQtestLoader.py
 alita_sdk/runtime/langchain/document_loaders/AlitaTableLoader.py
 alita_sdk/runtime/langchain/document_loaders/AlitaTextLoader.py

{alita_sdk-0.3.295 → alita_sdk-0.3.296}/pyproject.toml

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