Update agents.py

agents.py

@@ -11,10 +11,17 @@ from langchain_core.prompts import PromptTemplate
 
 load_dotenv()
 
-langfuse_handler = CallbackHandler(
+langfuse_news_analysis_handler = CallbackHandler(
     secret_key=os.getenv("LANGFUSE_SECRET_KEY"),
     public_key=os.getenv("LANGFUSE_PUBLIC_KEY"),
     host="https://cloud.langfuse.com", # 🇪🇺 EU region
+    session_id="news_analysis",
+)
+langfuse_post_generation_handler = CallbackHandler(
+    secret_key=os.getenv("LANGFUSE_SECRET_KEY"),
+    public_key=os.getenv("LANGFUSE_PUBLIC_KEY"),
+    host="https://cloud.langfuse.com", # 🇪🇺 EU region
+    session_id="post_generation",
 )
 langfuse = Langfuse()
 
@@ -41,11 +48,7 @@ def basic_analysis(news):
         try:
             response = analysis_llm.invoke(
                 prompt.compile(news_object = news), 
-                config={"callbacks": [langfuse_handler],
-                        "metadata": {
-                            "langfuse_session_id": "news_analysis",
-                         },
-                       }
+                config={"callbacks": [langfuse_news_analysis_handler]}
             )
             print("################ BASIC ANALYSIS AGENT RESPONSE ################")
             print(response.content)
@@ -88,12 +91,7 @@ def get_text_post_content(details, reference):
 
 
         user_query = prompt.compile(NEWS_CONTENT = details, CHAR_LENGTH = 490- len(reference))
-        response = post_content_llm.invoke(user_query, 
-                                           config={"callbacks": [langfuse_handler],
-                                                   "metadata": {
-                                                        "langfuse_session_id": "post_content_generation",
-                                                    },
-                                                  })
+        response = post_content_llm.invoke(user_query, config={"callbacks": [langfuse_post_generation_handler]})
 
         print("POST CONTENT RESPONSE:", response)