hailin
/
faiss_rag_enterprise
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

.

This commit is contained in:
hailin 2025-05-09 20:34:01 +08:00
parent 467f85fc1b
commit 7872cb5197
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
scripts/rag_build_query.py
View File

@ -14,6 +14,7 @@ from scripts.permissions import get_user_allowed_indexes
import faiss import faiss
from typing import List from typing import List
import asyncio import asyncio
from sentence_transformers import SentenceTransformer
USER_INDEX_PATH = "index_data" USER_INDEX_PATH = "index_data"
USER_DOC_PATH = "docs" USER_DOC_PATH = "docs"
@ -21,7 +22,7 @@ USER_DOC_PATH = "docs"
class CustomEmbedding(BaseEmbedding): class CustomEmbedding(BaseEmbedding):
model: SentenceTransformer # ✅ 显式声明是字段 model: SentenceTransformer # ✅ 显式声明是字段
def __init__(self, model_name: str): def __init__(self, model_name: str):
from sentence_transformers import SentenceTransformer from sentence_transformers import SentenceTransformer
self.model = SentenceTransformer(model_name) self.model = SentenceTransformer(model_name)