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

feat(iflow): Add User-Agent header to API requests

Browse Source
This commit is contained in:
hkfires
2025-10-05 18:50:35 +08:00
parent 64f4c18fea
commit c8029b7166
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
internal/runtime/executor/iflow_executor.go
View File

@@ -20,7 +20,10 @@ import (
"github.com/tidwall/sjson" "github.com/tidwall/sjson"
) )
const iflowDefaultEndpoint = "/chat/completions" const (
iflowDefaultEndpoint = "/chat/completions"
iflowUserAgent = "iFlow-Cli"
)
// IFlowExecutor executes OpenAI-compatible chat completions against the iFlow API using API keys derived from OAuth. // IFlowExecutor executes OpenAI-compatible chat completions against the iFlow API using API keys derived from OAuth.
type IFlowExecutor struct { type IFlowExecutor struct {
@@ -215,6 +218,7 @@ func (e *IFlowExecutor) Refresh(ctx context.Context, auth *cliproxyauth.Auth) (*
func applyIFlowHeaders(r *http.Request, apiKey string, stream bool) { func applyIFlowHeaders(r *http.Request, apiKey string, stream bool) {
r.Header.Set("Content-Type", "application/json") r.Header.Set("Content-Type", "application/json")
r.Header.Set("Authorization", "Bearer "+apiKey) r.Header.Set("Authorization", "Bearer "+apiKey)
r.Header.Set("User-Agent", iflowUserAgent)
if stream { if stream {
r.Header.Set("Accept", "text/event-stream") r.Header.Set("Accept", "text/event-stream")
} else { } else {