Merge pull request #142 from gomate-community/pipeline

update@rename embedding

Author: yanqiangmiffy

trustrag/applications/rag_openai.py

@@ -18,7 +18,7 @@
 from trustrag.modules.prompt.templates import CHAT_PROMPT_TEMPLATES
 from trustrag.modules.judger.llm_judger import LLMJudger
 from trustrag.modules.retrieval.dense_retriever import DenseRetriever
-from trustrag.modules.retrieval.embedding import OpenAIEmbedding
+from trustrag.modules.vector.embedding import OpenAIEmbedding
 from trustrag.modules.retrieval.web_retriever import DuckduckSearcher
 from trustrag.modules.rewriter.llm_rewriter import LLMRewriter
 

trustrag/applications/rag_openai_bge.py

@@ -14,13 +14,13 @@
 from trustrag.modules.document.common_parser import CommonParser
 from trustrag.modules.generator.chat import DeepSeekChat
 from trustrag.modules.generator.chat import GptTurbo, GPT4_DMXAPI
-from trustrag.modules.generator.llm import PROMPT_TEMPLATE
+from trustrag.modules.prompt.templates import CHAT_PROMPT_TEMPLATES
 from trustrag.modules.rewriter.llm_rewriter import LLMRewriter
 from trustrag.modules.judger.llm_judger import LLMJudger
 from trustrag.modules.reranker.bge_reranker import BgeReranker
 from trustrag.modules.retrieval.dense_retriever import DenseRetriever
 from trustrag.modules.document.chunk import TextChunker
-from trustrag.modules.retrieval.embedding import FlagModelEmbedding
+from trustrag.modules.vector.embedding import FlagModelEmbedding
 from  trustrag.modules.retrieval.web_retriever import DuckduckSearcher
 from trustrag.modules.chunks.sentence_chunk import SentenceChunker
 

trustrag/modules/deepresearch/finder/searcher.py

@@ -8,7 +8,7 @@
 
 from trustrag.modules.engine.websearch import DuckduckEngine, SearxngEngine
 from trustrag.modules.engine.qdrant import QdrantEngine
-from trustrag.modules.retrieval.embedding import SentenceTransformerEmbedding
+from trustrag.modules.vector.embedding import SentenceTransformerEmbedding
 
 
 @dataclass