Merge pull request #2461 from wucm667/fix/image-gen-upstream-context-detach
fix(gateway): 修复图片生成上游请求过早取消
This commit is contained in:
commit
57603e169e
@ -592,7 +592,7 @@ func (s *OpenAIGatewayService) forwardOpenAIImagesAPIKey(
|
|||||||
setOpsUpstreamRequestBody(c, forwardBody)
|
setOpsUpstreamRequestBody(c, forwardBody)
|
||||||
}
|
}
|
||||||
|
|
||||||
upstreamCtx, releaseUpstreamCtx := detachStreamUpstreamContext(ctx, parsed.Stream)
|
upstreamCtx, releaseUpstreamCtx := detachUpstreamContext(ctx)
|
||||||
defer releaseUpstreamCtx()
|
defer releaseUpstreamCtx()
|
||||||
|
|
||||||
token, _, err := s.GetAccessToken(upstreamCtx, account)
|
token, _, err := s.GetAccessToken(upstreamCtx, account)
|
||||||
|
|||||||
@ -967,7 +967,7 @@ func (s *OpenAIGatewayService) forwardOpenAIImagesOAuth(
|
|||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
upstreamCtx, releaseUpstreamCtx := detachStreamUpstreamContext(ctx, parsed.Stream)
|
upstreamCtx, releaseUpstreamCtx := detachUpstreamContext(ctx)
|
||||||
defer releaseUpstreamCtx()
|
defer releaseUpstreamCtx()
|
||||||
|
|
||||||
token, _, err := s.GetAccessToken(upstreamCtx, account)
|
token, _, err := s.GetAccessToken(upstreamCtx, account)
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user