github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-24 12:45:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(prompt): Initialize prompt templates to ensure all templates are load (#2729)

Browse Source
This commit is contained in:
Eric Wong 2025-05-26 18:52:52 +08:00 committed by GitHub
parent 7e9885574a
commit 761c75a2c3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 48 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
packages/dbgpt-app/src/dbgpt_app/component_configs.py
View File

@ -57,6 +57,8 @@ def initialize_components(
register_serve_apps(system_app, param, web_config.host, web_config.port)
_initialize_operators()
_initialize_code_server(system_app)
# Initialize prompt templates - MUST be after serve apps registration
_initialize_prompt_templates()
def _initialize_model_cache(system_app: SystemApp, web_config: ServiceWebParameters):
@ -158,3 +160,49 @@ def _initialize_code_server(system_app: SystemApp):
from dbgpt.util.code.server import initialize_code_server
initialize_code_server(system_app)
def _initialize_prompt_templates():
"""Initialize all prompt templates by importing scene modules.
This ensures that all prompt templates are registered in the prompt registry
before the application starts serving requests.
"""
logger.info("Initializing prompt templates...")
try:
# Import all scene prompt modules to trigger registration
# This is the same list as in ChatFactory.get_implementation()
# Verify that templates are registered
from dbgpt._private.config import Config
from dbgpt_app.scene.chat_dashboard.prompt import prompt # noqa: F401
from dbgpt_app.scene.chat_data.chat_excel.excel_analyze.prompt import ( # noqa: F401,F811
prompt,
)
from dbgpt_app.scene.chat_data.chat_excel.excel_learning.prompt import ( # noqa: F401, F811
prompt,
)
from dbgpt_app.scene.chat_db.auto_execute.prompt import ( # noqa: F401,F811
prompt,
)
from dbgpt_app.scene.chat_db.professional_qa.prompt import ( # noqa: F401, F811
prompt,
)
from dbgpt_app.scene.chat_knowledge.refine_summary.prompt import ( # noqa: F401,F811
prompt,
)
from dbgpt_app.scene.chat_knowledge.v1.prompt import prompt # noqa: F401,F811
from dbgpt_app.scene.chat_normal.prompt import prompt # noqa: F401,F811
cfg = Config()
registry = cfg.prompt_template_registry.registry
registered_scenes = list(registry.keys())
logger.info(
f"Successfully initialized prompt templates for scenes: {registered_scenes}"
)
except Exception as e:
logger.error(f"Failed to initialize prompt templates: {e}")
# Don't raise exception to avoid breaking the application startup
# The templates will be loaded lazily when needed