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 clearml import Logger, Task
 from langchain.tools import Tool
 from langchain.memory import ConversationBufferMemory, SimpleMemory
 from langchain_community.callbacks import ClearMLCallbackHandler
@@ -195,9 +195,6 @@ def execute(query):
             f"Process this user query: {query}")
         # 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)}")