Something went wrong. Try again.
sloproxy
Something went wrong. Try again.
1.5 kB · 48 lines
Go
at main
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849package helps
import ( "context"
cliproxyauth "github.com/router-for-me/CLIProxyAPI/v7/sdk/cliproxy/auth" cliproxyexecutor "github.com/router-for-me/CLIProxyAPI/v7/sdk/cliproxy/executor" "github.com/tidwall/gjson")
// EmitWebSocketResponseEvent delivers an upstream WebSocket response frame to any configured observer.func EmitWebSocketResponseEvent(ctx context.Context, opts cliproxyexecutor.Options, auth *cliproxyauth.Auth, provider, model string, payload []byte) { if opts.WebSocketResponseObserver == nil || len(payload) == 0 { return } var authID, authLabel, authType string if auth != nil { authID = auth.ID authLabel = auth.Label authType, _ = auth.AccountInfo() } eventType := gjson.GetBytes(payload, "type").String() var requestID string var traceID string if opts.Metadata != nil { if reqID, ok := opts.Metadata["request_id"].(string); ok { requestID = reqID } if trID, ok := opts.Metadata["trace_id"].(string); ok { traceID = trID } } requestedModel := PayloadRequestedModel(opts, model) opts.WebSocketResponseObserver(ctx, cliproxyexecutor.WebSocketResponseEvent{ RequestID: requestID, TraceID: traceID, SourceFormat: opts.SourceFormat.String(), Model: model, RequestedModel: requestedModel, Provider: provider, AuthID: authID, AuthLabel: authLabel, AuthType: authType, EventType: eventType, Payload: payload, Metadata: opts.Metadata, })}