fix: Return response

AgentOps-AI · Jun 5, 2024 · 25af3df · 25af3df
1 parent 566eb3d
commit 25af3df
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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