christoph/ai-agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chuns erhoeht

Browse Source
This commit is contained in:
Christoph K.
2026-03-12 19:24:14 +01:00
parent 75003b9534
commit 3c8d3873dc
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
internal/brain/ingest.go
View File

@@ -234,7 +234,8 @@ func ingestChunks(ctx context.Context, embClient *openai.Client, pointsClient pb
return err return err
} }
// IngestChatMessage speichert eine einzelne Chat-Nachricht in Qdrant. // IngestChatMessage speichert eine Chat-Nachricht in Qdrant.
// Lange Nachrichten (> maxChunkSize) werden automatisch aufgeteilt.
func IngestChatMessage(text, author, source string) error { func IngestChatMessage(text, author, source string) error {
ctx := context.Background() ctx := context.Background()
ctx = metadata.AppendToOutgoingContext(ctx, "api-key", config.Cfg.Qdrant.APIKey) ctx = metadata.AppendToOutgoingContext(ctx, "api-key", config.Cfg.Qdrant.APIKey)
@@ -246,9 +247,12 @@ func IngestChatMessage(text, author, source string) error {
ensureCollection(ctx, pb.NewCollectionsClient(conn)) ensureCollection(ctx, pb.NewCollectionsClient(conn))
pointsClient := pb.NewPointsClient(conn) pointsClient := pb.NewPointsClient(conn)
fullText := fmt.Sprintf("[%s] %s", author, text) prefix := fmt.Sprintf("[%s] ", author)
c := chunk{Text: fullText, Source: source, Type: "chat"} var chunks []chunk
return ingestChunks(ctx, embClient, pointsClient, []chunk{c}) for _, part := range splitLongSection(text) {
chunks = append(chunks, chunk{Text: prefix + part, Source: source, Type: "chat"})
}
return ingestChunks(ctx, embClient, pointsClient, chunks)
} }
func boolPtr(b bool) *bool { return &b } func boolPtr(b bool) *bool { return &b }