From 25af3df4e5168ee61860c0159b54f0f5434fa942 Mon Sep 17 00:00:00 2001
From: Siddharth Prajosh <sprajosh@gmail.com>
Date: Wed, 5 Jun 2024 11:54:10 +0530
Subject: [PATCH] fix: Return response

---
 agentops/llm_tracker.py | 2 ++
 1 file changed, 2 insertions(+)

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