add prompt templates

This commit is contained in:
2026-02-11 16:44:41 +08:00
parent 9351672001
commit 8226f8c97c

View File

@@ -29,3 +29,62 @@ CREATE TABLE IF NOT EXISTS prompt_templates (
CREATE INDEX IF NOT EXISTS idx_prompt_templates_set_id
ON prompt_templates(prompt_set_id);
-- Seed: initial prompt set for lang_agent/graphs/routing.py
-- The pipeline_id can be used by RoutingConfig.pipeline_id to load these prompts.
INSERT INTO prompt_sets (pipeline_id, name, description, is_active, list)
SELECT
'routing',
'default',
'Initial prompt set for RoutingGraph nodes',
true,
''
WHERE NOT EXISTS (
SELECT 1
FROM prompt_sets
WHERE pipeline_id = 'routing'
AND name = 'default'
);
INSERT INTO prompt_templates (prompt_set_id, prompt_key, content)
SELECT ps.id, t.prompt_key, t.content
FROM prompt_sets ps
JOIN (
VALUES
('route_prompt', '决定用工具或者聊天'),
('chat_prompt', '正常聊天时说什么'),
('tool_prompt', '如何用工具')
) AS t(prompt_key, content)
ON true
WHERE ps.pipeline_id = 'routing'
AND ps.name = 'default'
ON CONFLICT (prompt_set_id, prompt_key)
DO UPDATE SET
content = EXCLUDED.content,
updated_at = now();
-- Seed: initial prompt set for lang_agent/graphs/react.py
-- ReactGraph uses prompt key "sys_prompt" (see default_key in build_prompt_store).
INSERT INTO prompt_sets (pipeline_id, name, description, is_active, list)
SELECT
'react',
'default',
'Initial prompt set for ReactGraph',
true,
''
WHERE NOT EXISTS (
SELECT 1
FROM prompt_sets
WHERE pipeline_id = 'react'
AND name = 'default'
);
INSERT INTO prompt_templates (prompt_set_id, prompt_key, content)
SELECT ps.id, 'sys_prompt', '如何用工具'
FROM prompt_sets ps
WHERE ps.pipeline_id = 'react'
AND ps.name = 'default'
ON CONFLICT (prompt_set_id, prompt_key)
DO UPDATE SET
content = EXCLUDED.content,
updated_at = now();