mirror of
https://wget.la/https://github.com/leookun/cursor-byok
synced 2026-08-18 03:57:06 +08:00
fix: 修复502导致的内存泄漏问题
This commit is contained in:
@@ -25,6 +25,7 @@ func (gateway *DefaultProviderGateway) StartStream(ctx context.Context, req Prov
|
||||
if ctx == nil {
|
||||
ctx = context.Background()
|
||||
}
|
||||
defer releaseArtifactSession(req.Observer, req.RequestID, req.ModelCallID)
|
||||
requestKnobs := make(map[string]any, len(req.RequestKnobs)+2)
|
||||
for key, value := range req.RequestKnobs {
|
||||
requestKnobs[key] = value
|
||||
@@ -60,3 +61,15 @@ func (gateway *DefaultProviderGateway) StartStream(ctx context.Context, req Prov
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
type artifactSessionCleaner interface {
|
||||
ClearActiveArtifacts(requestID string, modelCallID string)
|
||||
}
|
||||
|
||||
func releaseArtifactSession(observer modeladapter.LLMArtifactObserver, requestID string, modelCallID string) {
|
||||
cleaner, ok := observer.(artifactSessionCleaner)
|
||||
if !ok {
|
||||
return
|
||||
}
|
||||
cleaner.ClearActiveArtifacts(requestID, modelCallID)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user