test fix

AgentOps-AI · Jun 26, 2024 · 9120b79 · 9120b79
1 parent 94a5866
commit 9120b79
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/tests/test_record_function.py b/tests/test_record_function.py
@@ -192,8 +192,8 @@ async def async_add(x, y):
         assert mock_req.last_request.headers["X-Agentops-Api-Key"] == self.api_key
         assert mock_req.last_request.headers["Authorization"] == f"Bearer some_jwt2"
         assert request_json["events"][0]["action_type"] == self.event_type
-        assert request_json["events"][0]["params"] == {"x": 1, "y": 2, "z": 3}
-        assert request_json["events"][0]["returns"] == 6
+        assert request_json["events"][0]["params"] == {"x": 1, "y": 2}
+        assert request_json["events"][0]["returns"] == 3
 
         second_last_request_json = mock_req.request_history[-2].json()
         assert (
@@ -206,9 +206,8 @@ async def async_add(x, y):
         assert second_last_request_json["events"][0]["params"] == {
             "x": 1,
             "y": 2,
-            "z": 3,
         }
-        assert second_last_request_json["events"][0]["returns"] == 6
+        assert second_last_request_json["events"][0]["returns"] == 3
 
         session_1.end_session(end_state="Success")
         session_2.end_session(end_state="Success")