Tested openai

AgentOps-AI · May 10, 2024 · 609e2a9 · 609e2a9
1 parent 7a83865
commit 609e2a9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/agentops/llm_tracker.py b/agentops/llm_tracker.py
@@ -104,7 +104,7 @@ def generator():
 
         # v0.0.0 responses are dicts
         try:
-            self.llm_event.returns = response  # TODO: test
+            self.llm_event.returns = response
             self.llm_event.agent_id = check_call_stack_for_agent_id()
             self.llm_event.prompt = kwargs["messages"]
             self.llm_event.prompt_tokens = response['usage']['prompt_tokens']