From 60f4ad8759643834756fee86b6aee92f78804790 Mon Sep 17 00:00:00 2001
From: Simone Vellei <henomis@gmail.com>
Date: Wed, 15 May 2024 01:41:59 +0200
Subject: [PATCH] fix: names

---
 rag/rag.go          | 6 +++---
 rag/rag_fusion.go   | 2 +-
 rag/sub_document.go | 2 +-
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/rag/rag.go b/rag/rag.go
index 83571a55..5f8764a9 100644
--- a/rag/rag.go
+++ b/rag/rag.go
@@ -85,7 +85,7 @@ func (r *RAG) WithLoader(sourceRegexp *regexp.Regexp, loader Loader) *RAG {
 func (r *RAG) AddSources(ctx context.Context, sources ...string) error {
 	ctx, span, err := r.startObserveSpan(
 		ctx,
-		"RAG AddSources",
+		"rag-add-source",
 		types.M{
 			"chunkSize":    r.chunkSize,
 			"chunkOverlap": r.chunkOverlap,
@@ -118,7 +118,7 @@ func (r *RAG) AddSources(ctx context.Context, sources ...string) error {
 func (r *RAG) AddDocuments(ctx context.Context, documents ...document.Document) error {
 	ctx, span, err := r.startObserveSpan(
 		ctx,
-		"RAG AddDocument",
+		"rag-add-document",
 		types.M{
 			"chunkSize":    r.chunkSize,
 			"chunkOverlap": r.chunkOverlap,
@@ -144,7 +144,7 @@ func (r *RAG) AddDocuments(ctx context.Context, documents ...document.Document)
 func (r *RAG) Retrieve(ctx context.Context, query string) ([]string, error) {
 	ctx, span, err := r.startObserveSpan(
 		ctx,
-		"RAG Retrieve",
+		"rag-retrieve",
 		types.M{
 			"query": query,
 			"topK":  r.topK,
diff --git a/rag/rag_fusion.go b/rag/rag_fusion.go
index 35c48a12..2401b58c 100644
--- a/rag/rag_fusion.go
+++ b/rag/rag_fusion.go
@@ -34,7 +34,7 @@ func NewFusion(index *index.Index, llm LLM) *Fusion {
 func (r *Fusion) Retrieve(ctx context.Context, query string) ([]string, error) {
 	ctx, span, err := r.startObserveSpan(
 		ctx,
-		"RAGFusion Retrieve",
+		"rag-fusion-retrieve",
 		types.M{
 			"query": query,
 			"topK":  r.topK,
diff --git a/rag/sub_document.go b/rag/sub_document.go
index ed3b290d..fe799ebe 100644
--- a/rag/sub_document.go
+++ b/rag/sub_document.go
@@ -64,7 +64,7 @@ func (r *SubDocumentRAG) WithLoader(sourceRegexp *regexp.Regexp, loader Loader)
 func (r *SubDocumentRAG) AddSources(ctx context.Context, sources ...string) error {
 	ctx, span, err := r.startObserveSpan(
 		ctx,
-		"SubDocumentRAG AddSources",
+		"rag-subdocument-add-sources",
 		types.M{
 			"chunkSize":      r.chunkSize,
 			"childChunkSize": r.childChunkSize,