debugging

interface.py

@@ -143,7 +143,6 @@ class ChatInterface:
         if message is not None:
             messages.append({"role": "user", "content": [{"type": "text", "text": message}]})
 
-        print(f"Messages: {messages}")
         try:
             for event in self.agent.workflow.stream(
                 {"messages": messages}, {"configurable": {"thread_id": self.current_thread_id}}

medrax/agent/agent.py

@@ -117,11 +117,11 @@ class ChatAgent:
             Dict[str, List[AnyMessage]]: A dictionary containing the model's response.
         """
         messages = state["messages"]
-        # print('process_request input', state)
+        print('process_request input', state)
         if self.prompts:
             messages = [SystemMessage(content=self.prompts["MEDICAL_ASSISTANT"])] + messages
         response = self.model.invoke(messages)
-        # print('process_request output', response)
+        print('process_request output', response)
         return {"messages": [response]}
 
     def has_tool_calls(self, state: AgentState) -> bool: