open-source/cursor-byok
1
0
Fork 0
mirror of https://wget.la/https://github.com/leookun/cursor-byok synced 2026-08-17 19:47:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(cursor): update model handling to use agentv1 responses and enhance CLI model details

Browse Source 
- Refactored response handling in newProtoMessage to utilize agentv1 for GetUsableModelsResponse and GetDefaultModelForCliResponse.
- Added new tests for buildCLIModelDetails and encodeCLIModels to ensure correct model details and encoding.
- Updated buildUsableModelsPayload and buildDefaultModelForCliPayload to leverage buildCLIModelDetails for improved model data structure.
This commit is contained in:
leookun
2026-08-03 22:56:17 +08:00
parent 424235f11d
commit 5df3e8f512
3 changed files with 78 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/backend/server/upstream/client.go
+3 -2
View File
@@ -14,6 +14,7 @@ import (
"strings"
"time"
"cursor/gen/agentv1"
"cursor/gen/aiserverv1"
"cursor/internal/logger"
"cursor/internal/netproxy"
@@ -433,9 +434,9 @@ func newProtoMessage(typeName string) (proto.Message, error) {
case "aiserver.v1.ListMarketplacesResponse":
return &aiserverv1.ListMarketplacesResponse{}, nil
case "aiserver.v1.GetUsableModelsResponse":
return &aiserverv1.GetUsableModelsResponse{}, nil
return &agentv1.GetUsableModelsResponse{}, nil
case "aiserver.v1.GetDefaultModelForCliResponse":
return &aiserverv1.GetDefaultModelForCliResponse{}, nil
return &agentv1.GetDefaultModelForCliResponse{}, nil
case "aiserver.v1.GetDefaultModelResponse":
return &aiserverv1.GetDefaultModelResponse{}, nil
case "aiserver.v1.GetGlobalCommandsResponse":