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(antigravity): add streaming support for Claude model requests

Browse Source
This commit is contained in:
Luis Pater
2025-12-17 22:16:57 +08:00
parent ae7b972649
commit 09923f654c
1 changed files with 19 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
internal/runtime/executor/antigravity_executor.go
View File

@@ -77,6 +77,25 @@ func (e *AntigravityExecutor) Execute(ctx context.Context, auth *cliproxyauth.Au
auth = updatedAuth auth = updatedAuth
} }
if strings.Contains(req.Model, "claude") {
stream, errExecuteStream := e.ExecuteStream(ctx, auth, req, opts)
if errExecuteStream != nil {
return resp, errExecuteStream
}
var buffer bytes.Buffer
for chunk := range stream {
if chunk.Err != nil {
return resp, chunk.Err
}
if len(chunk.Payload) > 0 {
_, _ = buffer.Write(chunk.Payload)
}
}
resp = cliproxyexecutor.Response{Payload: buffer.Bytes()}
return resp, nil
}
reporter := newUsageReporter(ctx, e.Identifier(), req.Model, auth) reporter := newUsageReporter(ctx, e.Identifier(), req.Model, auth)
defer reporter.trackFailure(ctx, &err) defer reporter.trackFailure(ctx, &err)