Update workflow.py

workflow.py

@@ -94,6 +94,7 @@ class ResearchWorkflow:
         try:
             domain = state["context"].get("domain", "Biomedical Research").strip().lower()
             fallback_analyses = ResearchConfig.DOMAIN_FALLBACKS
+            # If the domain matches a fallback, return it.
             if domain in fallback_analyses:
                 logger.info(f"Using fallback analysis for domain: {state['context'].get('domain')}")
                 return {
@@ -148,10 +149,12 @@ class ResearchWorkflow:
             current_analysis = state["messages"][-1].content
             refinement_history.append(current_analysis)
             difficulty_level = max(0, 3 - state["context"]["refine_count"])
+            domain = state["context"].get("domain", "Biomedical Research")
             logger.info(f"Refinement iteration: {state['context']['refine_count']}, Difficulty level: {difficulty_level}")
 
             if state["context"]["refine_count"] >= 3:
                 meta_prompt = (
+                    f"Domain: {domain}\n"
                     "You are given the following series of refinement outputs:\n" +
                     "\n---\n".join(refinement_history) +
                     "\n\nSynthesize the above into a final, concise, and high-quality technical analysis report. "
@@ -165,6 +168,7 @@ class ResearchWorkflow:
                 }
             else:
                 refinement_prompt = (
+                    f"Domain: {domain}\n"
                     f"Refine this analysis (current difficulty level: {difficulty_level}):\n{current_analysis}\n\n"
                     "First, critically evaluate the analysis and identify its weaknesses, such as inaccuracies, unsupported claims, or lack of clarity. Summarize these weaknesses in a short paragraph.\n\n"
                     "Then, improve the following aspects:\n"