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

3
api/core/rag/rerank/rerank_model.py
View file

@ -2,9 +2,10 @@ from typing import Optional
from core.model_manager import ModelInstance
from core.rag.models.document import Document
from core.rag.rerank.rerank_base import BaseRerankRunner
class RerankModelRunner:
class RerankModelRunner(BaseRerankRunner):
def __init__(self, rerank_model_instance: ModelInstance) -> None:
self.rerank_model_instance = rerank_model_instance