feat(metadata): enhance metadata extraction with reasoning content support

2026-03-27 00:24:16 +02:00
parent 74e67526d1
commit f76d1bbd23
2 changed files with 185 additions and 8 deletions
--- a/internal/ai/compat/client_test.go
+++ b/internal/ai/compat/client_test.go
@@ -178,3 +178,75 @@ func TestExtractMetadataParsesCodeFencedJSON(t *testing.T) {
 		t.Fatalf("metadata people = %#v, want [Alice]", metadata.People)
 	}
 }
+
+func TestExtractMetadataParsesArrayContent(t *testing.T) {
+	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		_ = json.NewEncoder(w).Encode(map[string]any{
+			"choices": []map[string]any{
+				{
+					"message": map[string]any{
+						"content": []map[string]any{
+							{"type": "text", "text": "{\"people\":[],\"action_items\":[],\"dates_mentioned\":[],\"topics\":[\"auth\"],\"type\":\"reference\",\"source\":\"mcp\"}"},
+						},
+					},
+				},
+			},
+		})
+	}))
+	defer server.Close()
+
+	client := New(Config{
+		Name:           "test",
+		BaseURL:        server.URL,
+		APIKey:         "secret",
+		EmbeddingModel: "embed-model",
+		MetadataModel:  "meta-model",
+		HTTPClient:     server.Client(),
+		Log:            discardLogger(),
+	})
+
+	metadata, err := client.ExtractMetadata(context.Background(), "hello")
+	if err != nil {
+		t.Fatalf("ExtractMetadata() error = %v", err)
+	}
+	if metadata.Type != "reference" {
+		t.Fatalf("metadata type = %q, want reference", metadata.Type)
+	}
+}
+
+func TestExtractMetadataUsesReasoningContentWhenContentEmpty(t *testing.T) {
+	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		_ = json.NewEncoder(w).Encode(map[string]any{
+			"choices": []map[string]any{
+				{
+					"message": map[string]any{
+						"content":           "",
+						"reasoning_content": "{\"people\":[\"Hein\"],\"action_items\":[],\"dates_mentioned\":[],\"topics\":[\"profile\"],\"type\":\"person_note\",\"source\":\"mcp\"}",
+					},
+				},
+			},
+		})
+	}))
+	defer server.Close()
+
+	client := New(Config{
+		Name:           "test",
+		BaseURL:        server.URL,
+		APIKey:         "secret",
+		EmbeddingModel: "embed-model",
+		MetadataModel:  "meta-model",
+		HTTPClient:     server.Client(),
+		Log:            discardLogger(),
+	})
+
+	metadata, err := client.ExtractMetadata(context.Background(), "hello")
+	if err != nil {
+		t.Fatalf("ExtractMetadata() error = %v", err)
+	}
+	if metadata.Type != "person_note" {
+		t.Fatalf("metadata type = %q, want person_note", metadata.Type)
+	}
+	if len(metadata.People) != 1 || metadata.People[0] != "Hein" {
+		t.Fatalf("metadata people = %#v, want [Hein]", metadata.People)
+	}
+}