logan
cd319ce7bd
- 侧栏 Ask|Agent 分段；Ask 仍走 /api/ai/chat，Agent 走 /api/ai/agent - RunAgent 多轮 tool call → 进程内 aitools（与 MCP 同源只读/白名单/审计） - 回复下步骤时间线；openai-chat tools（DeepSeek）；agent-accept 冒烟
61 lines
1.8 KiB
Go
61 lines
1.8 KiB
Go
package ai
|
|
|
|
import "context"
|
|
|
|
// Message 对话消息(OpenAI 兼容协议)。Agent 多轮可带 tool_calls / tool 角色。
|
|
type Message struct {
|
|
Role string `json:"role"`
|
|
Content string `json:"content,omitempty"`
|
|
ToolCalls []ToolCall `json:"tool_calls,omitempty"`
|
|
ToolCallID string `json:"tool_call_id,omitempty"`
|
|
Name string `json:"name,omitempty"`
|
|
}
|
|
|
|
// ToolCall OpenAI function tool_call。
|
|
type ToolCall struct {
|
|
ID string `json:"id"`
|
|
Type string `json:"type"`
|
|
Function ToolCallFunc `json:"function"`
|
|
}
|
|
|
|
type ToolCallFunc struct {
|
|
Name string `json:"name"`
|
|
Arguments string `json:"arguments"`
|
|
}
|
|
|
|
// ChatReq 在 stub 时代字段上演进,前端契约不变;新增 schema/model 可选字段。
|
|
type ChatReq struct {
|
|
Action string `json:"action"`
|
|
Prompt string `json:"prompt"`
|
|
SQL string `json:"sql"`
|
|
Driver string `json:"driver"`
|
|
Database string `json:"database"`
|
|
Error string `json:"error"`
|
|
Schema string `json:"schema,omitempty"` // 前端注入的库表列上下文
|
|
Model string `json:"model,omitempty"`
|
|
}
|
|
|
|
type ChatRes struct {
|
|
Content string `json:"content"`
|
|
SQL string `json:"sql,omitempty"`
|
|
}
|
|
|
|
// Provider AI 提供方抽象。本轮只做非流式 Complete;
|
|
// Stream(ctx, req) (<-chan string, error) 预留,SSE 流式留待后续版本。
|
|
type Provider interface {
|
|
Name() string
|
|
Complete(ctx context.Context, req ChatReq) (ChatRes, error)
|
|
}
|
|
|
|
// Default 按当前配置选择 Provider;stub 时或远端配置缺 key(免鉴权除外)回落 stub(离线占位)。
|
|
func Default() Provider {
|
|
cfg := LoadConfig()
|
|
if cfg.Provider == "" || cfg.Provider == "stub" {
|
|
return stubProvider{}
|
|
}
|
|
if cfg.APIKey == "" && cfg.AuthStyle != AuthNone {
|
|
return stubProvider{}
|
|
}
|
|
return NewRemoteProvider(cfg)
|
|
}
|