c4a0e3ef53
- 流式输出: SSE 逐 token 接收, \\n\n\ 段落缓冲后 mdprint 彩色渲染 - 日志系统: charmbracelet/log v2 双写(stderr + log.yml), yunshu log 命令 - 会议室架构: dialog(main) + weather/profile/note(sub) 多 Agent 编排 - 泛型工具注册: NewTool[T] 反射推导 JSON Schema, 类型安全 - 安全加固: safeMemoryPath 三段校验(EvalSymlinks+Rel), maxToolCalls=2 - 性能优化: sync.Once 延迟加载, note 一步完成, obs/summary 合并 - Prompt 适配: 流式输出原则(先调工具不说话), 单 Agent 查询跳过 obs+summary - 文档: AGENTS.md + architecture.md + changelog.md 全部同步至 v2.3.0
100 lines
2.4 KiB
Go
100 lines
2.4 KiB
Go
package main
|
|
|
|
type MessageRole string
|
|
|
|
const (
|
|
RoleSystem MessageRole = "system"
|
|
RoleUser MessageRole = "user"
|
|
RoleAssistant MessageRole = "assistant"
|
|
RoleTool MessageRole = "tool"
|
|
)
|
|
|
|
type Message struct {
|
|
Role MessageRole `json:"role"`
|
|
Content string `json:"content"`
|
|
ToolCalls []ToolCall `json:"tool_calls,omitempty"`
|
|
ToolCallID string `json:"tool_call_id,omitempty"`
|
|
}
|
|
|
|
type ToolCall struct {
|
|
ID string `json:"id"`
|
|
Type string `json:"type"`
|
|
Function ToolCallFunction `json:"function"`
|
|
}
|
|
|
|
type ToolCallFunction struct {
|
|
Name string `json:"name"`
|
|
Arguments string `json:"arguments"`
|
|
}
|
|
|
|
type CacheDef struct {
|
|
TTL int `yaml:"ttl"`
|
|
Keys []string `yaml:"keys"`
|
|
}
|
|
|
|
type AgentDef struct {
|
|
Name string `yaml:"name"`
|
|
Description string `yaml:"description"`
|
|
Type string `yaml:"type"`
|
|
Cache *CacheDef `yaml:"cache,omitempty"`
|
|
Tools []string `yaml:"tools"`
|
|
SystemPrompt string
|
|
}
|
|
|
|
type Schema map[string]any
|
|
|
|
type ToolDef struct {
|
|
Name string
|
|
Description string
|
|
Parameters Schema
|
|
Execute func(args map[string]interface{}) (string, error)
|
|
}
|
|
|
|
type OpenAIMessage struct {
|
|
Role string `json:"role"`
|
|
Content *string `json:"content"`
|
|
ToolCalls []ToolCall `json:"tool_calls,omitempty"`
|
|
}
|
|
|
|
type OpenAIChoice struct {
|
|
Index int `json:"index"`
|
|
Message OpenAIMessage `json:"message"`
|
|
FinishReason string `json:"finish_reason"`
|
|
}
|
|
|
|
type OpenAIUsage struct {
|
|
PromptTokens int `json:"prompt_tokens"`
|
|
CompletionTokens int `json:"completion_tokens"`
|
|
TotalTokens int `json:"total_tokens"`
|
|
}
|
|
|
|
type OpenAIResponse struct {
|
|
ID string `json:"id"`
|
|
Object string `json:"object"`
|
|
Created int64 `json:"created"`
|
|
Model string `json:"model"`
|
|
Choices []OpenAIChoice `json:"choices"`
|
|
Usage OpenAIUsage `json:"usage"`
|
|
}
|
|
|
|
type OpenAIErrorResponse struct {
|
|
Error OpenAIError `json:"error"`
|
|
}
|
|
|
|
type OpenAIError struct {
|
|
Message string `json:"message"`
|
|
Type string `json:"type"`
|
|
Code string `json:"code"`
|
|
}
|
|
|
|
type OpenAITool struct {
|
|
Type string `json:"type"`
|
|
Function OpenAIToolFunc `json:"function"`
|
|
}
|
|
|
|
type OpenAIToolFunc struct {
|
|
Name string `json:"name"`
|
|
Description string `json:"description"`
|
|
Parameters interface{} `json:"parameters"`
|
|
}
|