fix: names

henomis · May 14, 2024 · 60f4ad8 · 60f4ad8
1 parent ed42e47
commit 60f4ad8
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git 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
@@ -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
@@ -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,