jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: move the embedding to the rag module and abstract the rerank runner for extension (#9423)

Browse source
This commit is contained in:
zhuhao 2024-10-17 19:12:42 +08:00 • committed by GitHub
commit b90ad587c2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
61 changed files with 135 additions and 78 deletions
Download patch file Download diff file Expand all files Collapse all files

2
api/core/rag/datasource/retrieval_service.py
View file

@ -6,7 +6,7 @@ from flask import Flask, current_app
from core.rag.data_post_processor.data_post_processor import DataPostProcessor
from core.rag.datasource.keyword.keyword_factory import Keyword
from core.rag.datasource.vdb.vector_factory import Vector
from core.rag.rerank.constants.rerank_mode import RerankMode
from core.rag.rerank.rerank_type import RerankMode
from core.rag.retrieval.retrieval_methods import RetrievalMethod
from extensions.ext_database import db
from models.dataset import Dataset