From 609e2a988fb8c566cd660e3a0ed0508eeb2abc7f Mon Sep 17 00:00:00 2001 From: Howard Gil Date: Fri, 10 May 2024 12:08:48 -0700 Subject: [PATCH] Tested openai --- agentops/llm_tracker.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/agentops/llm_tracker.py b/agentops/llm_tracker.py index f4e8c18e..250aa41c 100644 --- 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']