From d7ad7a8342c495fa278c41dea62849b1a37b1513 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E5=A6=82=E5=A8=81?= Date: Thu, 7 Aug 2025 17:06:18 +0800 Subject: [PATCH] fix: bugs --- examples/simple_test.py | 4 ++++ src/base_rag/core.py | 4 ++++ 2 files changed, 8 insertions(+) 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