feat: enhance retriever trace summary by adding policies support and refactoring related methods

2026-04-12 10:59:59 +08:00
parent 7f70614739
commit 0b77d91dfe
3 changed files with 106 additions and 11 deletions
@@ -79,6 +79,7 @@ func (c *RuntimeTraceCollector) SetRetrieverSummary(summary RetrieverTraceSummar
 	c.Data.Retriever.EmbeddingMs = summary.EmbeddingMs
 	c.Data.Retriever.VectorSearchMs = summary.VectorSearchMs
 	c.Data.Retriever.HydrateMs = summary.HydrateMs
+	c.Data.Retriever.Policies = append([]RetrieverPolicyTraceItem(nil), summary.Policies...)
 }

 func (c *RuntimeTraceCollector) AddToolItem(item ToolTraceItem) {