Skip to content

Commit b5c552f

Browse files
committed
update@fixed embedding bug
1 parent d10e3b1 commit b5c552f

File tree

2 files changed

+1
-2
lines changed

2 files changed

+1
-2
lines changed

trustrag/applications/rag.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
from trustrag.modules.reranker.bge_reranker import BgeReranker
1616
from trustrag.modules.retrieval.dense_retriever import DenseRetriever
1717
from trustrag.modules.document.chunk import TextChunker
18-
from trustrag.modules.retrieval.embedding import FlagModelEmbedding
18+
from trustrag.modules.vector.embedding import FlagModelEmbedding
1919
class ApplicationConfig():
2020
def __init__(self):
2121
self.retriever_config = None

trustrag/modules/engine/chroma.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
from typing import List, Dict, Any, Union
22
import numpy as np
33
import chromadb
4-
from chromadb.config import Settings
54
from trustrag.modules.vector.embedding import EmbeddingGenerator
65

76

0 commit comments

Comments
 (0)