Refactor system prompt handling in LangGraphAgent and BasicAgent for improved clarity and consistency

agent.py

@@ -69,14 +69,7 @@ class LangGraphAgent:
 
     def __init__(self, *, model: Optional[object] = None, system_prompt: Optional[str] = None):
         # Guide the model to use tools and to output a clear final answer.
-        base_prompt = system_prompt or "You are a helpful assistant. Keep answers concise."
-        self.system_prompt = (
-            base_prompt
-            # + "\n\nGuidelines:\n"
-            #   "- Use tools when they can verify facts or fetch fresh data.\n"
-            #   "- Think privately; do not reveal chain-of-thought.\n"
-            #   "- Provide the final user-facing result prefixed exactly with 'FINAL ANSWER:'."
-        )
+        self.system_prompt = system_prompt or "You are a helpful assistant. Keep answers concise."
 
         # Choose an LLM if not provided
         if model is None:

app.py

@@ -3,16 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-
-# --- System Prompt Integration ---
-try:
-    from prompt import get_system_prompt as _get_system_prompt  # preferred
-except Exception:
-    _get_system_prompt = None  # type: ignore
-try:
-    from prompt import SYSTEM_PROMPT as _SYSTEM_PROMPT  # fallback constant
-except Exception:
-    _SYSTEM_PROMPT = None  # type: ignore
+from prompt import system_prompt
 
 
 def _load_system_prompt() -> str:
@@ -20,13 +11,8 @@ def _load_system_prompt() -> str:
     Resolve system prompt from prompt.py:
     - Prefer get_system_prompt() if available, else SYSTEM_PROMPT, else default.
     """
-    if callable(_get_system_prompt):
-        try:
-            return _get_system_prompt()  # type: ignore
-        except Exception:
-            pass
-    if isinstance(_SYSTEM_PROMPT, str) and _SYSTEM_PROMPT.strip():
-        return _SYSTEM_PROMPT  # type: ignore
+    if system_prompt is not None:
+        return system_prompt
     return "You are a helpful assistant. Keep answers concise."