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

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.160.dist-info → alita_sdk-0.3.161.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.160
+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

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

@@ -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.160.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.160.dist-info/METADATA,sha256=zzsqORk7WyWsRCx16yd_XAAXbn6SZbLLrQuAj_oGgI4,18667
-alita_sdk-0.3.160.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.160.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.160.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,,