diff --git a/examples/simple_test.py b/examples/simple_test.py index 6d366dc..9c4f2c7 100644 --- a/examples/simple_test.py +++ b/examples/simple_test.py @@ -5,8 +5,12 @@ import sys import os +import warnings from pathlib import Path +# 过滤掉PyTorch的FutureWarning +warnings.filterwarnings('ignore', category=FutureWarning, module='torch') + # 添加源码路径 sys.path.append(os.path.join(os.path.dirname(__file__), '..', 'src')) diff --git a/src/base_rag/core.py b/src/base_rag/core.py index 249da5a..05a6964 100644 --- a/src/base_rag/core.py +++ b/src/base_rag/core.py @@ -6,10 +6,14 @@ import os import shutil import sqlite3 import hashlib +import warnings from datetime import datetime from pathlib import Path from enum import Enum +# 过滤掉PyTorch的FutureWarning,避免干扰用户体验 +warnings.filterwarnings('ignore', category=FutureWarning, module='torch') + from langchain_huggingface import HuggingFaceEmbeddings from langchain.embeddings.base import Embeddings from langchain_chroma import Chroma