x/themblem
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix rag

Browse Source
This commit is contained in:
Fam Zheng 2025-11-04 22:05:45 +00:00
parent da86e7d670
commit de6d175f7e
2 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
api/products/rag_service.py
View File

@ -1,13 +1,13 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
""" """
RAG Service Module for 徵象防伪验证平台 RAG Service Module for Themblem
基于 LangChain 的实时知识库检索服务，支持缓存优化 基于 LangChain 的实时知识库检索服务，支持缓存优化
""" """
from langchain.text_splitter import RecursiveCharacterTextSplitter from langchain_text_splitters import RecursiveCharacterTextSplitter
from langchain.embeddings import HuggingFaceEmbeddings from langchain_community.embeddings import HuggingFaceEmbeddings
from langchain.vectorstores import FAISS from langchain_community.vectorstores import FAISS
from langchain.schema import Document from langchain_core.documents import Document
from .models import Article from .models import Article
from django.core.cache import cache from django.core.cache import cache
from typing import List, Dict, Any from typing import List, Dict, Any

7
requirements.txt
View File

@ -12,7 +12,12 @@ requests
# RAG dependencies # RAG dependencies
langchain>=0.1.0 langchain>=0.1.0
langchain-community>=0.1.0 langchain-community>=0.1.0
langchain-core>=0.1.0
langchain-text-splitters>=0.1.0
sentence-transformers>=2.2.0 sentence-transformers>=2.2.0
faiss-cpu>=1.7.4 faiss-cpu>=1.7.4
beautifulsoup4>=4.12.0 beautifulsoup4>=4.12.0
openai>=1.0.0 openai>=1.0.0
# Protobuf version constraint (compatibility with transformers/tensorflow)
protobuf>=3.19.0,<3.21.0