Update agent/planning_agent.py

agent/planning_agent.py

@@ -1,6 +1,6 @@
 from langchain_openai import ChatOpenAI
 from langchain.agents import initialize_agent, AgentType
-from langchain.agents import AgentType, initialize_agent, load_tools
+
 from langchain.tools import Tool
 from langchain.memory import ConversationBufferMemory, SimpleMemory
 from langchain_community.callbacks import ClearMLCallbackHandler
@@ -193,8 +193,11 @@ def execute(query):
         query_memory.memories['original_query'] = query 
         response = agent.run(
             f"Process this user query: {query}")
-        # clearml_callback.flush_tracker(langchain_asset=llm,name="Planning agent", finish=True)
-        clearml_callback.flush_tracker(langchain_asset=llm,name="Planning agent")
+        # clearml_callback.flush_tracker(langchain_asset=agent,name="Planning agent", finish=True)
+        #clearml_callback.flush_tracker(langchain_asset=llm,name="Planning agent")
+        a_task = Task.get_task(project_name='langchain_callback_demo', task_name='llm')
+        logger = a_task.get_logger()
+        logger.report_text(response)
         return response
     except Exception as e:
         logger.error(f"Error in planning agent: {str(e)}")