debugging

app.py

@@ -76,7 +76,6 @@ def initialize_agent(
 
     checkpointer = MemorySaver()
     model = ChatOpenAI(model=model, temperature=temperature, top_p=top_p, **openai_kwargs)
-    print(model.invoke("Hey how are you?"))
     agent = ChatAgent(
         model,
         tools=list(tools_dict.values()),
@@ -104,8 +103,8 @@ if __name__ == "__main__":
         "DicomProcessorTool",
         "ChestXRayClassifierTool",
         "ChestXRaySegmentationTool",
-        # "ChestXRayReportGeneratorTool",
-        # "XRayVQATool",
+        "ChestXRayReportGeneratorTool",
+        "XRayVQATool",
         # "LlavaMedTool",
         # "XRayPhraseGroundingTool",
         # "ChestXRayGeneratorTool",
@@ -119,7 +118,6 @@ if __name__ == "__main__":
     if base_url := os.getenv("OPENAI_BASE_URL"):
         openai_kwargs["base_url"] = base_url
 
-    print(f"OPENAI arguments - {openai_kwargs}")
     agent, tools_dict = initialize_agent(
         "medrax/docs/system_prompts.txt",
         tools_to_use=selected_tools,

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: