Update app.py

app.py

@@ -20,7 +20,7 @@ chat_memory = None
 query_memory = None
 
 def initialize_components():
-    global llm, chat_memory, query_memory
+    global llm, chat_memory, query_memory, clearml_callback
     load_dotenv()
     
     OPENAI_API_KEY="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"
@@ -43,8 +43,21 @@ def initialize_components():
     )
     query_memory = SimpleMemory()
 
+    # Setup and use the ClearML Callback
+    clearml_callback = ClearMLCallbackHandler(
+    task_type="inference",
+    project_name="langchain_callback_demo",
+    task_name="llm",
+    tags=["test"],
+    # Change the following parameters based on the amount of detail you want tracked
+    visualize=True,
+    complexity_metrics=True,
+    stream_logs=True,)
+    
+    callbacks = [StdOutCallbackHandler(), clearml_callback]
+    
     # Initialize planning agent with both memories
-    planning_agent.initialize_planning_agent(llm, chat_memory, query_memory)
+    planning_agent.initialize_planning_agent(llm, chat_memory, query_memory, callbacks)
 
     logger.info("Components initialized successfully")