chuan/CLIProxyAPI
1
0
Fork 0
mirror of https://github.com/router-for-me/CLIProxyAPI.git synced 2026-02-03 13:00:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(vertex): add usage source resolution for Vertex projects

Browse Source 
Extend `resolveUsageSource` to support Vertex projects by extracting and normalizing `project_id` or `project` from the metadata for accurate source resolution.
This commit is contained in:
Luis Pater
2025-11-12 08:43:02 +08:00
parent 09ecba6dab
commit 592f6fc66b
1 changed files with 15 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
internal/runtime/executor/usage_helpers.go
View File

@@ -129,6 +129,21 @@ func apiKeyFromContext(ctx context.Context) string {
func resolveUsageSource(auth *cliproxyauth.Auth, ctxAPIKey string) string { func resolveUsageSource(auth *cliproxyauth.Auth, ctxAPIKey string) string {
if auth != nil { if auth != nil {
provider := strings.TrimSpace(auth.Provider)
if strings.EqualFold(provider, "vertex") {
if auth.Metadata != nil {
if projectID, ok := auth.Metadata["project_id"].(string); ok {
if trimmed := strings.TrimSpace(projectID); trimmed != "" {
return trimmed
}
}
if project, ok := auth.Metadata["project"].(string); ok {
if trimmed := strings.TrimSpace(project); trimmed != "" {
return trimmed
}
}
}
}
if _, value := auth.AccountInfo(); value != "" { if _, value := auth.AccountInfo(); value != "" {
return strings.TrimSpace(value) return strings.TrimSpace(value)
} }