moved files

This commit is contained in:
2026-03-03 14:07:01 +08:00
parent 65a1705280
commit 6b0e50c532
5 changed files with 0 additions and 0 deletions

View File

@@ -0,0 +1,12 @@
-- Create the messages table
CREATE TABLE IF NOT EXISTS messages (
id BIGSERIAL PRIMARY KEY,
conversation_id TEXT NOT NULL,
message_type VARCHAR(10) NOT NULL CHECK (message_type IN ('human', 'ai', 'tool')),
content TEXT NOT NULL,
sequence_number INTEGER NOT NULL CHECK (sequence_number >= 0),
created_at TIMESTAMPTZ DEFAULT NOW()
);
-- Index for fast retrieval of all messages in a conversation (in order)
CREATE INDEX IF NOT EXISTS idx_messages_conversation ON messages (conversation_id, sequence_number);

View File

@@ -0,0 +1,103 @@
-- A prompt_set groups a full collection of prompts together.
-- Each pipeline can have many sets (versions, A/B variants, etc.);
-- exactly one should be marked is_active per pipeline.
CREATE TABLE IF NOT EXISTS prompt_sets (
id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
pipeline_id VARCHAR(64) NOT NULL,
graph_id VARCHAR(64),
name VARCHAR(128) NOT NULL,
description TEXT DEFAULT '',
is_active BOOLEAN DEFAULT false,
created_at TIMESTAMPTZ DEFAULT now(),
updated_at TIMESTAMPTZ DEFAULT now(),
list Varchar(255) DEFAULT '', -- tool_set list for client_tool_manager
api_key TEXT DEFAULT '' -- provider api key used to run pipeline
);
-- Backward-compatible migration for existing deployments.
ALTER TABLE prompt_sets
ADD COLUMN IF NOT EXISTS graph_id VARCHAR(64);
ALTER TABLE prompt_sets
ADD COLUMN IF NOT EXISTS api_key TEXT DEFAULT '';
UPDATE prompt_sets
SET graph_id = pipeline_id
WHERE graph_id IS NULL;
-- Fast lookup of the active set for a pipeline
CREATE INDEX IF NOT EXISTS idx_prompt_sets_pipeline_active
ON prompt_sets(pipeline_id, is_active);
-- Each prompt belongs to a prompt_set, keyed by prompt_key.
CREATE TABLE IF NOT EXISTS prompt_templates (
id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
prompt_set_id UUID NOT NULL REFERENCES prompt_sets(id) ON DELETE CASCADE,
prompt_key VARCHAR(64) NOT NULL,
content TEXT NOT NULL,
created_at TIMESTAMPTZ DEFAULT now(),
updated_at TIMESTAMPTZ DEFAULT now(),
UNIQUE(prompt_set_id, prompt_key)
);
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, graph_id, name, description, is_active, list)
SELECT
'routing',
'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, graph_id, name, description, is_active, list)
SELECT
'react',
'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();

View File

@@ -0,0 +1,13 @@
-- Create a new database
CREATE DATABASE ai_conversations;
-- Create a dedicated user (role) for your app
CREATE USER myapp_user WITH PASSWORD 'secure_password_123';
-- Grant privileges
GRANT ALL PRIVILEGES ON DATABASE ai_conversations TO myapp_user;
-- Also needed: allow user to create schemas/tables
\c ai_conversations
ALTER USER myapp_user CREATEDB; -- optional but helpful during dev
GRANT ALL ON SCHEMA public TO myapp_user;

View File

@@ -0,0 +1,20 @@
-- Drop the index first (if it exists)
DROP INDEX IF EXISTS idx_messages_conversation;
-- Drop the messages table (if it exists)
DROP TABLE IF EXISTS messages;
-- Recreate the messages table with TEXT conversation_id
-- Note: UUID extension is no longer needed since conversation_id is TEXT
CREATE TABLE messages (
id BIGSERIAL PRIMARY KEY,
conversation_id TEXT NOT NULL,
message_type VARCHAR(10) NOT NULL CHECK (message_type IN ('human', 'ai', 'tool')),
content TEXT NOT NULL,
sequence_number INTEGER NOT NULL CHECK (sequence_number >= 0),
created_at TIMESTAMPTZ DEFAULT NOW()
);
-- Recreate the index for fast retrieval of all messages in a conversation (in order)
CREATE INDEX idx_messages_conversation ON messages (conversation_id, sequence_number);