From de6d175f7ec8a0a4d5e72fc167b5a6ec9802cc22 Mon Sep 17 00:00:00 2001 From: Fam Zheng Date: Tue, 4 Nov 2025 22:05:45 +0000 Subject: [PATCH] fix rag --- api/products/rag_service.py | 10 +++++----- requirements.txt | 7 ++++++- 2 files changed, 11 insertions(+), 6 deletions(-) diff --git a/api/products/rag_service.py b/api/products/rag_service.py index 983dd54..a59534e 100644 --- a/api/products/rag_service.py +++ b/api/products/rag_service.py @@ -1,13 +1,13 @@ #!/usr/bin/env python3 """ -RAG Service Module for 徵象防伪验证平台 +RAG Service Module for Themblem 基于 LangChain 的实时知识库检索服务，支持缓存优化 """ -from langchain.text_splitter import RecursiveCharacterTextSplitter -from langchain.embeddings import HuggingFaceEmbeddings -from langchain.vectorstores import FAISS -from langchain.schema import Document +from langchain_text_splitters import RecursiveCharacterTextSplitter +from langchain_community.embeddings import HuggingFaceEmbeddings +from langchain_community.vectorstores import FAISS +from langchain_core.documents import Document from .models import Article from django.core.cache import cache from typing import List, Dict, Any diff --git a/requirements.txt b/requirements.txt index 65f66bf..e1dec28 100644 --- a/requirements.txt +++ b/requirements.txt @@ -12,7 +12,12 @@ requests # RAG dependencies langchain>=0.1.0 langchain-community>=0.1.0 +langchain-core>=0.1.0 +langchain-text-splitters>=0.1.0 sentence-transformers>=2.2.0 faiss-cpu>=1.7.4 beautifulsoup4>=4.12.0 -openai>=1.0.0 \ No newline at end of file +openai>=1.0.0 + +# Protobuf version constraint (compatibility with transformers/tensorflow) +protobuf>=3.19.0,<3.21.0 \ No newline at end of file