d21f9c3769
- Added a `noop` check in multiple logging methods (`LLMComplete`, `ToolStart`, `ToolComplete`, `HookStart`, `HookComplete`, and `HistoryLoad`) to prevent logging when the logger is in no-operation mode. - Improved command handling in `root.go` by removing minimum argument requirements for commands and providing help output when no arguments are given. - Introduced an `agent` command for better organization of agent-related functionalities in the CLI. - Implemented automatic detection of the application root directory in `run.go` to streamline the application startup process. - Cleaned up debug print statements in `config.go` to reduce clutter in the output.
72 lines
3.1 KiB
Go
72 lines
3.1 KiB
Go
package agent
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/yaoapp/kun/exception"
|
|
"github.com/yaoapp/yao/config"
|
|
)
|
|
|
|
var appPath string
|
|
var envFile string
|
|
|
|
var langs = map[string]string{
|
|
"Test an agent with input cases": "使用测试用例测试智能体",
|
|
"Test an agent with input cases from JSONL file or direct message": "使用 JSONL 文件或直接消息测试智能体",
|
|
"Application directory": "应用目录",
|
|
"Environment file": "环境变量文件",
|
|
"Input: JSONL file path or message (required)": "输入: JSONL 文件路径或消息 (必需)",
|
|
"Path to output file (default: output-{timestamp}.jsonl)": "输出文件路径 (默认: output-{timestamp}.jsonl)",
|
|
"Explicit agent ID (default: auto-detect)": "指定智能体 ID (默认: 自动检测)",
|
|
"Override connector": "覆盖连接器",
|
|
"Test user ID (default: test-user)": "测试用户 ID (默认: test-user)",
|
|
"Test team ID (default: test-team)": "测试团队 ID (默认: test-team)",
|
|
"Reporter agent ID for custom report": "自定义报告生成器智能体 ID",
|
|
"Number of runs for stability analysis": "稳定性分析的运行次数",
|
|
"Default timeout per test case": "每个测试用例的默认超时时间",
|
|
"Number of parallel test cases": "并行测试用例数",
|
|
"Verbose output": "详细输出",
|
|
"Stop on first failure": "遇到第一个失败时停止",
|
|
"Error: input is required (-i flag)": "错误: 需要输入 (-i 参数)",
|
|
"Error: failed to get current directory": "错误: 获取当前目录失败",
|
|
"Error: agent (-n) is required when using direct message input and not in an agent directory": "错误: 使用直接消息输入且不在智能体目录时需要指定 -n 参数",
|
|
"Hint: Make sure you're in a Yao application directory or specify --app flag": "提示: 确保在 Yao 应用目录中或使用 --app 参数指定",
|
|
"Error: invalid timeout format": "错误: 无效的超时格式",
|
|
}
|
|
|
|
// L Language switch
|
|
func L(words string) string {
|
|
var lang = os.Getenv("YAO_LANG")
|
|
if lang == "" {
|
|
return words
|
|
}
|
|
|
|
if trans, has := langs[words]; has {
|
|
return trans
|
|
}
|
|
return words
|
|
}
|
|
|
|
// Boot sets the configuration
|
|
func Boot() {
|
|
root := config.Conf.Root
|
|
if appPath != "" {
|
|
r, err := filepath.Abs(appPath)
|
|
if err != nil {
|
|
exception.New("Root error %s", 500, err.Error()).Throw()
|
|
}
|
|
root = r
|
|
}
|
|
if envFile != "" {
|
|
config.Conf = config.LoadFrom(envFile)
|
|
} else {
|
|
config.Conf = config.LoadFrom(filepath.Join(root, ".env"))
|
|
}
|
|
|
|
if config.Conf.Mode == "production" {
|
|
config.Production()
|
|
} else if config.Conf.Mode == "development" {
|
|
config.Development()
|
|
}
|
|
}
|