WeKnora deployment source with the reviewed wiki identity, summary and recovery fixes

internal/tracing/langfuse/retrieval_obs_test.go

@@ -0,0 +1,49 @@
+package langfuse
+
+import (
+	"testing"
+
+	"github.com/Tencent/WeKnora/internal/types"
+)
+
+func TestSummarizeRetrieveOutput_countsAndPreviews(t *testing.T) {
+	out := SummarizeRetrieveOutput([]*types.RetrieveResult{
+		{
+			RetrieverEngineType: types.PostgresRetrieverEngineType,
+			RetrieverType:       types.VectorRetrieverType,
+			Results: []*types.IndexWithScore{
+				{ChunkID: "c1", KnowledgeID: "k1", Score: 0.9, Content: "alpha"},
+				{ChunkID: "c2", KnowledgeID: "k1", Score: 0.5, Content: "beta"},
+			},
+		},
+		{
+			RetrieverEngineType: types.PostgresRetrieverEngineType,
+			RetrieverType:       types.KeywordsRetrieverType,
+			Results: []*types.IndexWithScore{
+				{ChunkID: "c3", KnowledgeID: "k2", Score: 0.7, Content: "gamma"},
+			},
+		},
+	})
+
+	if out["total_hits"] != 3 {
+		t.Fatalf("total_hits = %v, want 3", out["total_hits"])
+	}
+	if out["vector_hits"] != 2 || out["keyword_hits"] != 1 {
+		t.Fatalf("vector/keyword hits = %v/%v", out["vector_hits"], out["keyword_hits"])
+	}
+	hits := out["top_hits"].([]map[string]interface{})
+	if len(hits) != 3 || hits[0]["chunk_id"] != "c1" {
+		t.Fatalf("unexpected top_hits: %#v", hits)
+	}
+}
+
+func TestSummarizeSearchResults_sortsByScore(t *testing.T) {
+	out := SummarizeSearchResults([]*types.SearchResult{
+		{ID: "low", Score: 0.2, Content: "low"},
+		{ID: "high", Score: 0.9, Content: "high"},
+	}, 10)
+	hits := out["top_hits"].([]map[string]interface{})
+	if len(hits) != 2 || hits[0]["chunk_id"] != "high" {
+		t.Fatalf("unexpected hits: %#v", hits)
+	}
+}