INNER CODE UNIT · Go
loadPrompt
voocel/ainovel-cli · cmd/ainovel-cli/main.go:225
func loadPrompt(opts cliOptions) (string, error) {
return loadPromptFrom(opts, os.Stdin)
}
func loadPromptFrom(opts cliOptions, stdin io.Reader) (string, error) {
if opts.PromptFile == "" {
return strings.TrimSpace(opts.Prompt), nil
}
if opts.PromptFile == "-" {
data, err := io.ReadAll(stdin)
if err != nil {
return "", fmt.Errorf("读取 prompt 失败: %w", err)
}
return strings.TrimSpace(string(data)), nil
}
return startup.LoadPromptFile(opts.PromptFile)
}