From f0656b4370fbeb54c7ada419489d7b7a400bc5f7 Mon Sep 17 00:00:00 2001 From: LE Quoc Dat Date: Thu, 22 May 2025 12:35:27 +0200 Subject: [PATCH] small fix --- backend/agent/run.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/backend/agent/run.py b/backend/agent/run.py index 14c85c1a..2f5d2991 100644 --- a/backend/agent/run.py +++ b/backend/agent/run.py @@ -75,15 +75,15 @@ async def run_agent( thread_manager.add_tool(DataProvidersTool) - # Only include sample response if the model name does not contain "anthropic" - if "anthropic" not in model_name.lower(): + if "gemini-2.5-flash" in model_name.lower(): + system_message = { "role": "system", "content": get_gemini_system_prompt() } # example included + elif "anthropic" not in model_name.lower(): + # Only include sample response if the model name does not contain "anthropic" sample_response_path = os.path.join(os.path.dirname(__file__), 'sample_responses/1.txt') with open(sample_response_path, 'r') as file: sample_response = file.read() system_message = { "role": "system", "content": get_system_prompt() + "\n\n " + sample_response + "" } - elif "gemini-2.5-flash" in model_name.lower(): - system_message = { "role": "system", "content": get_gemini_system_prompt() } # example included else: system_message = { "role": "system", "content": get_system_prompt() }