avoid ctxKeyIsStreaming is nil

cr7258 · Dec 21, 2024 · 8399b41 · 8399b41
1 parent 4643342
commit 8399b41
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/plugins/wasm-go/extensions/ai-proxy/provider/failover.go b/plugins/wasm-go/extensions/ai-proxy/provider/failover.go
@@ -543,7 +543,7 @@ func (c *ProviderConfig) OnRequestFailed(activeProvider Provider, ctx wrapper.Ht
 	if c.isFailoverEnabled() {
 		c.handleUnavailableApiToken(ctx, apiTokenInUse, log)
 	}
-	if c.isRetryOnFailureEnabled() && !ctx.GetContext(ctxKeyIsStreaming).(bool) {
+	if c.isRetryOnFailureEnabled() && ctx.GetContext(ctxKeyIsStreaming) != nil && !ctx.GetContext(ctxKeyIsStreaming).(bool) {
 		c.retryFailedRequest(activeProvider, ctx, log)
 		return types.HeaderStopAllIterationAndWatermark
 	}