68 lines
2.4 KiB
Go
68 lines
2.4 KiB
Go
package model
|
|
|
|
// WorkflowType 工作流类型
|
|
type WorkflowType string
|
|
|
|
const (
|
|
WorkflowTypeLoop WorkflowType = "loop"
|
|
WorkflowTypeParallel WorkflowType = "parallel"
|
|
WorkflowTypeSequential WorkflowType = "sequential"
|
|
)
|
|
|
|
// AiAgentConfigTable 一个 Agent 配置表的顶层结构，对应 YAML 中 tables 下的每一项
|
|
type AiAgentConfigTable struct {
|
|
AppName string `yaml:"app-name" json:"appName"`
|
|
Agent AgentSummary `yaml:"agent" json:"agent"`
|
|
Module AgentModule `yaml:"module" json:"module"`
|
|
}
|
|
|
|
// AgentSummary Agent 摘要信息
|
|
type AgentSummary struct {
|
|
AgentID string `yaml:"agent-id" json:"agentId"`
|
|
AgentName string `yaml:"agent-name" json:"agentName"`
|
|
AgentDesc string `yaml:"agent-desc" json:"agentDesc"`
|
|
}
|
|
|
|
// AgentModule Agent 模块配置，包含 API、模型、Agent 定义、工作流和 Runner
|
|
type AgentModule struct {
|
|
AiAPI AiAPIConfig `yaml:"ai-api" json:"aiApi"`
|
|
ChatModel ChatModelConfig `yaml:"chat-model" json:"chatModel"`
|
|
Agents []AgentConfig `yaml:"agents" json:"agents"`
|
|
AgentWorkflows []AgentWorkflowConfig `yaml:"agent-workflows" json:"agentWorkflows"`
|
|
Runner RunnerConfig `yaml:"runner" json:"runner"`
|
|
}
|
|
|
|
// AiAPIConfig LLM API 连接配置
|
|
type AiAPIConfig struct {
|
|
BaseURL string `yaml:"base-url" json:"baseUrl"`
|
|
APIKey string `yaml:"api-key" json:"apiKey"`
|
|
CompletionsPath string `yaml:"completions-path" json:"completionsPath"`
|
|
}
|
|
|
|
// ChatModelConfig 聊天模型配置
|
|
type ChatModelConfig struct {
|
|
Model string `yaml:"model" json:"model"`
|
|
}
|
|
|
|
// AgentConfig 单个 Agent 的定义
|
|
type AgentConfig struct {
|
|
Name string `yaml:"name" json:"name"`
|
|
Instruction string `yaml:"instruction" json:"instruction"`
|
|
Description string `yaml:"description" json:"description"`
|
|
OutputKey string `yaml:"output-key" json:"outputKey"`
|
|
}
|
|
|
|
// AgentWorkflowConfig 工作流配置，支持 loop/parallel/sequential 三种类型
|
|
type AgentWorkflowConfig struct {
|
|
Type WorkflowType `yaml:"type" json:"type"`
|
|
Name string `yaml:"name" json:"name"`
|
|
SubAgents []string `yaml:"sub-agents" json:"subAgents"`
|
|
Description string `yaml:"description" json:"description"`
|
|
MaxIterations int `yaml:"max-iterations" json:"maxIterations"`
|
|
}
|
|
|
|
// RunnerConfig Runner 配置，指定入口 Agent 名称
|
|
type RunnerConfig struct {
|
|
AgentName string `yaml:"agent-name" json:"agentName"`
|
|
}
|