update

aniketmaurya · May 24, 2024 · bacefc6 · bacefc6
1 parent cbd51d7
commit bacefc6
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/examples/server.py b/examples/server.py
@@ -10,6 +10,9 @@
     image_inspector,
 )
 from agents.tool_executor import need_tool_use
+import logging
+
+logging.basicConfig(level=logging.INFO, format="%(asctime)s %(message)s")
 
 
 class AgentAPI(ls.LitAPI):
@@ -23,6 +26,7 @@ def setup(self, device):
         ])
 
     def predict(self, messages):
+        logging.debug(messages[-1])
         output = self.llm.chat_completion(messages)
         if need_tool_use(output):
             tool_results = self.llm.run_tools(output)