fix: bugs

This commit is contained in:
李如威 2025-08-07 17:06:18 +08:00
parent 185344ee7a
commit d7ad7a8342
2 changed files with 8 additions and 0 deletions

View File

@ -5,8 +5,12 @@
import sys import sys
import os import os
import warnings
from pathlib import Path from pathlib import Path
# 过滤掉PyTorch的FutureWarning
warnings.filterwarnings('ignore', category=FutureWarning, module='torch')
# 添加源码路径 # 添加源码路径
sys.path.append(os.path.join(os.path.dirname(__file__), '..', 'src')) sys.path.append(os.path.join(os.path.dirname(__file__), '..', 'src'))

View File

@ -6,10 +6,14 @@ import os
import shutil import shutil
import sqlite3 import sqlite3
import hashlib import hashlib
import warnings
from datetime import datetime from datetime import datetime
from pathlib import Path from pathlib import Path
from enum import Enum from enum import Enum
# 过滤掉PyTorch的FutureWarning避免干扰用户体验
warnings.filterwarnings('ignore', category=FutureWarning, module='torch')
from langchain_huggingface import HuggingFaceEmbeddings from langchain_huggingface import HuggingFaceEmbeddings
from langchain.embeddings.base import Embeddings from langchain.embeddings.base import Embeddings
from langchain_chroma import Chroma from langchain_chroma import Chroma