vrr/open-code-review
2
0
Fork 0
mirror of https://github.com/alibaba/open-code-review.git synced 2026-07-10 17:59:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

feat: 为 llm test 增加来源和 url

Browse source
This commit is contained in:
kite 2026-05-13 14:15:24 +08:00
parent 92c4119d7d
commit f40872f26e
2 changed files with 9 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

4
cmd/opencodereview/llm_cmd.go
View file

@ -77,7 +77,9 @@ func runLLMTest() error {
if resp.Model != "" {
model = resp.Model
}
fmt.Printf("Model: %s\n", model)
fmt.Printf("Source: %s\n", ep.Source)
fmt.Printf("URL: %s\n", ep.URL)
fmt.Printf("Model: %s\n", model)
fmt.Printf("%s\n", resp.Content())
return nil
}

10
internal/llm/resolver.go
View file

@ -15,6 +15,7 @@ type ResolvedEndpoint struct {
Token string
Model string
Protocol string // "anthropic" or "openai"
Source string // human-readable config source label
}
// Environment variable names for OCR-specific configuration.
@ -52,6 +53,7 @@ func ResolveEndpoint(configPath string) (ResolvedEndpoint, error) {
return ResolvedEndpoint{}, fmt.Errorf("resolve %s: %w", s.name, err)
}
if ok && ep.URL != "" && ep.Token != "" && ep.Model != "" {
ep.Source = s.name
return ep, nil
}
}
@ -79,7 +81,7 @@ func tryOCREnv() (ResolvedEndpoint, bool, error) {
protocol = "openai"
}
return ResolvedEndpoint{URL: url, Token: token, Model: model, Protocol: protocol}, true, nil
return ResolvedEndpoint{URL: url, Token: token, Model: model, Protocol: protocol, Source: "OCR environment"}, true, nil
}
// llmFileConfig represents the llm section in config.json.
@ -123,7 +125,7 @@ func tryOCRConfig(path string) (ResolvedEndpoint, bool, error) {
protocol = "openai"
}
return ResolvedEndpoint{URL: cfg.Llm.URL, Token: cfg.Llm.AuthToken, Model: cfg.Llm.Model, Protocol: protocol}, true, nil
return ResolvedEndpoint{URL: cfg.Llm.URL, Token: cfg.Llm.AuthToken, Model: cfg.Llm.Model, Protocol: protocol, Source: "OCR config file"}, true, nil
}
// tryCCEnv reads Claude Code environment variables.
@ -137,7 +139,7 @@ func tryCCEnv() (ResolvedEndpoint, bool, error) {
url := ensureMessagesSuffix(baseURL)
return ResolvedEndpoint{URL: url, Token: token, Model: model, Protocol: "anthropic"}, true, nil
return ResolvedEndpoint{URL: url, Token: token, Model: model, Protocol: "anthropic", Source: "Claude Code environment"}, true, nil
}
// tryShellRC parses ~/.zshrc and ~/.bashrc for ANTHROPIC_* exports.
@ -213,7 +215,7 @@ func parseShellRC(path string) (ResolvedEndpoint, bool, error) {
url := ensureMessagesSuffix(baseURL)
return ResolvedEndpoint{URL: url, Token: token, Model: model, Protocol: "anthropic"}, true, nil
return ResolvedEndpoint{URL: url, Token: token, Model: model, Protocol: "anthropic", Source: "Shell rc file"}, true, nil
}
// ensureMessagesSuffix appends /v1/messages to base URLs that lack a versioned path.