diff --git a/agentops/llm_tracker.py b/agentops/llm_tracker.py index 5f8a0dca..03c90e96 100644 --- a/agentops/llm_tracker.py +++ b/agentops/llm_tracker.py @@ -434,6 +434,8 @@ def _handle_response_ollama(self, response, kwargs, init_timestamp): self.llm_event.prompt_tokens = 0 self.llm_event.completion_tokens = 0 + return response + def override_openai_v1_completion(self): from openai.resources.chat import completions