diff --git a/apps/edge/internal/openai/hot_path_cleanup.go b/apps/edge/internal/openai/hot_path_cleanup.go index 6cca5ec2..4a365dc2 100644 --- a/apps/edge/internal/openai/hot_path_cleanup.go +++ b/apps/edge/internal/openai/hot_path_cleanup.go @@ -335,7 +335,7 @@ func (c *logicalRequestCoordinator) commitCleanupByLineage( var target *logicalRequestRecord for _, record := range c.requests { if record.state == logicalRequestStateCleanup && record.cleanup && record.ownerEdgeID == ownerEdgeID && record.principalRef == principalRef && - record.lineage == lineage.Prefix && sameLogicalRequestResultIDs(record.expected, lineage.ResultIDs) { + sameLogicalRequestResultIDs(record.expected, lineage.ResultIDs) { target = record break } @@ -343,8 +343,9 @@ func (c *logicalRequestCoordinator) commitCleanupByLineage( if target == nil { return logicalRequestSnapshot{}, errLogicalRequestNotFound } - if err := validateLogicalRequestContinuationLineage(target.lineage, target.expectedIssuedCallHash, target.expected, lineage); err != nil { - return logicalRequestSnapshot{}, err + if target.lineage.Endpoint != lineage.Prefix.Endpoint || target.lineage.ToolsetDigest != lineage.Prefix.ToolsetDigest || + target.expectedIssuedCallHash != lineage.IssuedCallHash { + return logicalRequestSnapshot{}, errLogicalRequestLineage } snapshot := target.snapshot() delete(c.requests, target.id)