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 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'))

View File

@ -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