alita-sdk 0.3.159__py3-none-any.whl → 0.3.161__py3-none-any.whl

alita_sdk/runtime/langchain/assistant.py

@@ -61,8 +61,8 @@ class Assistant:
                                 "Review toolkits configuration or use pipeline as master agent.")
 
         # configure memory store if memory tool is defined
-        memory_tool = next((tool for tool in data['tools'] if tool['type'] == 'memory'), None)
-        self._configure_store(memory_tool)
+        # memory_tool = next((tool for tool in data['tools'] if tool['type'] == 'memory'), None)
+        # self._configure_store(memory_tool)
         
         # Lazy import to avoid circular dependency
         from ..toolkits.tools import get_tools

alita_sdk/runtime/toolkits/tools.py

@@ -67,6 +67,7 @@ def get_tools(tools_list: list, alita_client, llm, memory_store: BaseStore = Non
                 selected_tools=[],
                 llm=llm
             ))
+            # move on tools level
         # elif tool['type'] == 'memory':
         #     if memory_store is None:
         #         raise ToolException(f"Memory store is not provided for memory tool: {tool['name']}")
@@ -94,7 +95,7 @@ def get_tools(tools_list: list, alita_client, llm, memory_store: BaseStore = Non
     # Add community tools
     tools += community_tools(tools_list, alita_client, llm)
     # Add alita tools
-    tools += alita_tools(tools_list, alita_client, llm, memory_store)
+    tools += alita_tools(tools_list, alita_client, llm)
     # Add MCP tools
     tools += _mcp_tools(tools_list, alita_client)
     

alita_sdk/tools/__init__.py

@@ -1,8 +1,5 @@
 import logging
 from importlib import import_module
-from typing import Optional
-
-from langgraph.store.base import BaseStore
 
 logger = logging.getLogger(__name__)
 
@@ -77,14 +74,13 @@ _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')
 
 # Log import summary
 available_count = len(AVAILABLE_TOOLS)
 total_attempted = len(AVAILABLE_TOOLS) + len(FAILED_IMPORTS)
 logger.info(f"Tool imports completed: {available_count}/{total_attempted} successful")
 
-def get_tools(tools_list, alita, llm, store: Optional[BaseStore] = None, *args, **kwargs):
+def get_tools(tools_list, alita, llm, *args, **kwargs):
     tools = []
     for tool in tools_list:
         # validate tool name syntax - it cannot be started with _
@@ -94,7 +90,6 @@ def get_tools(tools_list, alita, llm, store: Optional[BaseStore] = None, *args,
         
         tool['settings']['alita'] = alita
         tool['settings']['llm'] = llm
-        tool['settings']['store'] = store
         tool_type = tool['type']
         
         # Check if tool is available and has get_tools function

alita_sdk/tools/memory/__init__.py

@@ -15,13 +15,6 @@ from pydantic import create_model, BaseModel, ConfigDict, Field, SecretStr
 
 name = "memory"
 
-def get_tools(tool):
-    return MemoryToolkit().get_toolkit(
-        namespace=tool['settings'].get('namespace', str(tool['id'])),
-        store=tool['settings'].get('store', None),
-        toolkit_name=tool.get('toolkit_name', '')
-).get_tools()
-
 class MemoryToolkit(BaseToolkit):
     tools: List[BaseTool] = []
 

alita_sdk/tools/postman/api_wrapper.py

@@ -1370,33 +1370,30 @@ class PostmanApiWrapper(BaseToolApiWrapper):
     def update_request_tests(self, request_path: str, tests: str, **kwargs) -> str:
         """Update request test scripts."""
         try:
-            # Get the request ID
-            _, request_id, _ = self._get_request_item_and_id(request_path)
-            
-            # Get current request to preserve existing data
-            current_request = self._make_request('GET', f'/collections/{self.collection_id}/requests/{request_id}')
-            request_data = current_request.get("data", {})
+            # Get request item and ID
+            request_item, request_id, _ = self._get_request_item_and_id(request_path)
             
-            # Prepare the events array - preserve any non-test events
-            events = [event for event in request_data.get("events", []) if event.get("listen") != "test"]
+            # Get existing events and preserve non-test events
+            existing_events = request_item.get("event", [])
+            events = [event for event in existing_events if event.get("listen") != "test"]
             
-            # Add the new test script
+            # Add the new test script using the official API format
             events.append({
                 "listen": "test",
                 "script": {
-                    "type": "text/javascript",
-                    "exec": tests.strip().split('\n')
+                    "exec": tests.strip().split('\n'),
+                    "type": "text/javascript"
                 }
             })
             
-            # Update the events array in the request data
-            request_data["events"] = events
-            
-            # Update the request using the individual request endpoint
+            # Create update payload using the events array format from official spec
+            request_update = {
+                "events": events
+            }
+
+            # Update using the individual request endpoint with proper events format
             response = self._make_request('PUT', f'/collections/{self.collection_id}/requests/{request_id}',
-                                        json=request_data)
-            
-            logger.info(f"Test script updated successfully for request '{request_path}'")
+                                          json=request_update)
             return json.dumps({"success": True, "message": f"Request '{request_path}' tests updated successfully"}, indent=2)
         except Exception as e:
             stacktrace = format_exc()
@@ -1407,33 +1404,30 @@ class PostmanApiWrapper(BaseToolApiWrapper):
     def update_request_pre_script(self, request_path: str, pre_request_script: str, **kwargs) -> str:
         """Update request pre-request scripts."""
         try:
-            # Get the request ID
-            _, request_id, _ = self._get_request_item_and_id(request_path)
-            
-            # Get current request to preserve existing data
-            current_request = self._make_request('GET', f'/collections/{self.collection_id}/requests/{request_id}')
-            request_data = current_request.get("data", {})
+            # Get request item and ID
+            request_item, request_id, _ = self._get_request_item_and_id(request_path)
             
-            # Prepare the events array - preserve any non-prerequest events
-            events = [event for event in request_data.get("events", []) if event.get("listen") != "prerequest"]
+            # Get existing events and preserve non-prerequest events
+            existing_events = request_item.get("event", [])
+            events = [event for event in existing_events if event.get("listen") != "prerequest"]
             
-            # Add the new prerequest script
+            # Add the new prerequest script using the official API format
             events.append({
                 "listen": "prerequest",
                 "script": {
-                    "type": "text/javascript",
-                    "exec": pre_request_script.strip().split('\n')
+                    "exec": pre_request_script.strip().split('\n'),
+                    "type": "text/javascript"
                 }
             })
             
-            # Update the events array in the request data
-            request_data["events"] = events
-            
-            # Update the request using the individual request endpoint
+            # Create update payload using the events array format from official spec
+            request_update = {
+                "events": events
+            }
+
+            # Update using the individual request endpoint with proper events format
             response = self._make_request('PUT', f'/collections/{self.collection_id}/requests/{request_id}',
-                                        json=request_data)
-            
-            logger.info(f"Pre-request script updated successfully for request '{request_path}'")
+                                          json=request_update)
             return json.dumps({"success": True, "message": f"Request '{request_path}' pre-script updated successfully"}, indent=2)
         except Exception as e:
             stacktrace = format_exc()
@@ -1612,16 +1606,14 @@ class PostmanApiWrapper(BaseToolApiWrapper):
             The script content as JSON string, or an error message if the script doesn't exist
         """
         try:
-            # Get the request ID and fetch current request data
-            _, request_id, _ = self._get_request_item_and_id(request_path)
-            
-            # Get current request to have the latest version with updated scripts
-            current_request = self._make_request('GET', f'/collections/{self.collection_id}/requests/{request_id}')
-            request_data = current_request.get("data", {})
+            # Get the request item from the collection and also try individual endpoint
+            request_item, request_id, _ = self._get_request_item_and_id(request_path)
             
-            # Find the script by type
             script_content = None
-            for event in request_data.get("events", []):
+            
+            # Method 1: Check events array (modern format)
+            events = request_item.get("event", [])
+            for event in events:
                 if event.get("listen") == script_type:
                     script = event.get("script", {})
                     exec_content = script.get("exec", [])
@@ -1631,13 +1623,25 @@ class PostmanApiWrapper(BaseToolApiWrapper):
                         script_content = str(exec_content)
                     break
             
+            # Method 2: If not found in events, try individual request endpoint for direct fields
             if script_content is None:
+                try:
+                    individual_request = self._make_request('GET', f'/collections/{self.collection_id}/requests/{request_id}')
+                    if script_type == "test":
+                        script_content = individual_request.get("tests", "")
+                    elif script_type == "prerequest":
+                        script_content = individual_request.get("preRequestScript", "")
+                except:
+                    # If individual endpoint fails, that's okay, we'll fall back to not found
+                    pass
+            
+            if not script_content or script_content.strip() == "":
                 return json.dumps({"success": False, "message": f"No {script_type} script found for request '{request_path}'"}, indent=2)
             
             return json.dumps({
                 "success": True,
                 "script_type": script_type, 
-                "script_content": script_content, 
+                "script_content": script_content.strip(), 
                 "request_path": request_path
             }, indent=2)
                 

{alita_sdk-0.3.159.dist-info → alita_sdk-0.3.161.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.159
+Version: 0.3.161
 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>
 License-Expression: Apache-2.0
@@ -31,8 +31,8 @@ Requires-Dist: langchain_community~=0.3.7; extra == "runtime"
 Requires-Dist: langchain-openai~=0.3.0; extra == "runtime"
 Requires-Dist: langgraph-checkpoint-sqlite~=2.0.0; extra == "runtime"
 Requires-Dist: langgraph-checkpoint-postgres~=2.0.1; extra == "runtime"
-Requires-Dist: langsmith>=0.3.45; extra == "runtime"
-Requires-Dist: langgraph>=0.4.8; extra == "runtime"
+Requires-Dist: langsmith==0.1.144; extra == "runtime"
+Requires-Dist: langgraph~=0.2.53; extra == "runtime"
 Requires-Dist: langchain_chroma~=0.2.2; extra == "runtime"
 Requires-Dist: langchain-unstructured~=0.1.6; extra == "runtime"
 Requires-Dist: langchain-postgres~=0.0.13; extra == "runtime"
@@ -122,7 +122,6 @@ Requires-Dist: yagmail==0.15.293; extra == "tools"
 Requires-Dist: pysnc==1.1.10; extra == "tools"
 Requires-Dist: shortuuid==1.0.13; extra == "tools"
 Requires-Dist: yarl==1.17.1; extra == "tools"
-Requires-Dist: langmem==0.0.27; extra == "tools"
 Provides-Extra: community
 Requires-Dist: retry-extended==0.2.3; extra == "community"
 Requires-Dist: browser-use==0.1.43; extra == "community"

{alita_sdk-0.3.159.dist-info → alita_sdk-0.3.161.dist-info}/RECORD

@@ -48,7 +48,7 @@ alita_sdk/runtime/clients/client.py,sha256=jbC_M72CybwZgFfMRL6paj-NmICrSuk1vVnVT
 alita_sdk/runtime/clients/datasource.py,sha256=HAZovoQN9jBg0_-lIlGBQzb4FJdczPhkHehAiVG3Wx0,1020
 alita_sdk/runtime/clients/prompt.py,sha256=li1RG9eBwgNK_Qf0qUaZ8QNTmsncFrAL2pv3kbxZRZg,1447
 alita_sdk/runtime/langchain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-alita_sdk/runtime/langchain/assistant.py,sha256=QJEMiEOrFMJ4GpnK24U2pKFblrvdQpKFdfhZsI2wAUI,7507
+alita_sdk/runtime/langchain/assistant.py,sha256=1G4yBhBc-tXgqerujUVu5Z8T49m_7ov-7zYYsm-jGb4,7511
 alita_sdk/runtime/langchain/chat_message_template.py,sha256=kPz8W2BG6IMyITFDA5oeb5BxVRkHEVZhuiGl4MBZKdc,2176
 alita_sdk/runtime/langchain/constants.py,sha256=eHVJ_beJNTf1WJo4yq7KMK64fxsRvs3lKc34QCXSbpk,3319
 alita_sdk/runtime/langchain/indexer.py,sha256=0ENHy5EOhThnAiYFc7QAsaTNp9rr8hDV_hTK8ahbatk,37592
@@ -102,7 +102,7 @@ alita_sdk/runtime/toolkits/artifact.py,sha256=7fTr9VpGd2zwCB3EwW4aqWa5jVKRTunqV3
 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=iDWnGpEnmiD226osSlMy7l5suUxOl6vDo8BTtzb8oCw,6128
+alita_sdk/runtime/toolkits/tools.py,sha256=gCIEtdeD9u-za-oIZtJ916r9oSR9_0gCWE5FIKynWdU,6148
 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
@@ -129,7 +129,7 @@ alita_sdk/runtime/utils/logging.py,sha256=svPyiW8ztDfhqHFITv5FBCj8UhLxz6hWcqGIY6
 alita_sdk/runtime/utils/save_dataframe.py,sha256=i-E1wp-t4wb17Zq3nA3xYwgSILjoXNizaQAA9opWvxY,1576
 alita_sdk/runtime/utils/streamlit.py,sha256=z4J_bdxkA0zMROkvTB4u379YBRFCkKh-h7PD8RlnZWQ,85644
 alita_sdk/runtime/utils/utils.py,sha256=dM8whOJAuFJFe19qJ69-FLzrUp6d2G-G6L7d4ss2XqM,346
-alita_sdk/tools/__init__.py,sha256=F1Mrl8jEUUnmfg_VwrNcVSwJKPBp-xmh9vv9goPRA0o,9933
+alita_sdk/tools/__init__.py,sha256=qsF21SiBa7P5gcWybCLP4K3xqA34LW9TVuM1QKaU-xc,9716
 alita_sdk/tools/elitea_base.py,sha256=NQaIxPX6DVIerHCb18jwUR6maZxxk73NZaTsFHkBQWE,21119
 alita_sdk/tools/ado/__init__.py,sha256=mD6GHcYMTtffPJkJvFPe2rzvye_IRmXmWfI7xYuZhO4,912
 alita_sdk/tools/ado/utils.py,sha256=PTCludvaQmPLakF2EbCGy66Mro4-rjDtavVP-xcB2Wc,1252
@@ -248,7 +248,7 @@ alita_sdk/tools/llm/llm_utils.py,sha256=v3_lWP_Nk6tJLkj0BYohOun0OWNfvzqLjPdPAMl-
 alita_sdk/tools/localgit/__init__.py,sha256=NScO0Eu-wl-rc63jjD5Qv1RXXB1qukSIJXx-yS_JQLI,2529
 alita_sdk/tools/localgit/local_git.py,sha256=gsAftNcK7nMCd8VsIkwDLs2SoG0MgpYdkQG5tmoynkA,18074
 alita_sdk/tools/localgit/tool.py,sha256=It_B24rMvFPurB355Oy5IShg2BsZTASsEoSS8hu2SXw,998
-alita_sdk/tools/memory/__init__.py,sha256=SOB5Lhf8v8v0-IDUXUgb1KNdv5je-ooi6oGor8iYPpI,2148
+alita_sdk/tools/memory/__init__.py,sha256=QBzuOQapovmbcFS4nG39p3g-fUPp3kQrjh8EGk6VmBs,1901
 alita_sdk/tools/ocr/__init__.py,sha256=pvslKVXyJmK0q23FFDNieuc7RBIuzNXTjTNj-GqhGb0,3335
 alita_sdk/tools/ocr/api_wrapper.py,sha256=08UF8wj1sR8DcW0z16pw19bgLatLkBF8dySW-Ds8iRk,29649
 alita_sdk/tools/ocr/text_detection.py,sha256=1DBxt54r3_HdEi93QynSIVta3rH3UpIvy799TPtDTtk,23825
@@ -273,7 +273,7 @@ alita_sdk/tools/pandas/statsmodels/descriptive.py,sha256=APdofBnEiRhMrn6tLKwH076
 alita_sdk/tools/pandas/statsmodels/hypothesis_testing.py,sha256=fdNAayMB3W7avMfKJCcbf2_P54vUXbq8KVebOB48348,10508
 alita_sdk/tools/pandas/statsmodels/regression.py,sha256=Y1pWK4u_qzrfA740K-FX0nZ5FREGGPk8mfvykPIYoiI,9164
 alita_sdk/tools/postman/__init__.py,sha256=W0HdtACnTZw6tnzj7_qY_X5RoRyX3czcUSVaZJjBW-Y,4236
-alita_sdk/tools/postman/api_wrapper.py,sha256=4Hf_aOvUB1G_BdlKvNaAqnQaoCnKDph-E7v4VEwaw5Y,77933
+alita_sdk/tools/postman/api_wrapper.py,sha256=DvdZtLPpe6LpsGfsF38UmNyDHjORwWWutisd5AVIogg,78094
 alita_sdk/tools/postman/postman_analysis.py,sha256=2d-Oi2UORosIePIUyncSONw9hY7dw8Zc7BQvCd4aqpg,45115
 alita_sdk/tools/pptx/__init__.py,sha256=LNSTQk0BncfdWLXAOGX2WXezG3D4qSEuYwLpokmF9iM,3438
 alita_sdk/tools/pptx/pptx_wrapper.py,sha256=yyCYcTlIY976kJ4VfPo4dyxj4yeii9j9TWP6W8ZIpN8,29195
@@ -317,8 +317,8 @@ alita_sdk/tools/zephyr_enterprise/api_wrapper.py,sha256=Ir3zHljhbZQJRJJQOBzS_GL5
 alita_sdk/tools/zephyr_enterprise/zephyr_enterprise.py,sha256=hV9LIrYfJT6oYp-ZfQR0YHflqBFPsUw2Oc55HwK0H48,6809
 alita_sdk/tools/zephyr_scale/__init__.py,sha256=2NTcdrfkx4GSegqyXhsPLsEpc4FlACuDy85b0fk6cAo,4572
 alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=UHVQUVqcBc3SZvDfO78HSuBzwAsRw2cCDQa-xMOzndE,68663
-alita_sdk-0.3.159.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.159.dist-info/METADATA,sha256=C4uRfDlRMstPa_EMg1hsUN1lHDkB80Xnme6REnRJ_Yk,18714
-alita_sdk-0.3.159.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.159.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.159.dist-info/RECORD,,
+alita_sdk-0.3.161.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.161.dist-info/METADATA,sha256=ybWYeKeLOJiQcnasyPl7i0jE6OTcqEmuVfmid62TT3A,18667
+alita_sdk-0.3.161.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.161.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.161.dist-info/RECORD,,