fix:修复docker镜像构建问题

This commit is contained in:
xiamuceer
2025-11-04 17:27:39 +08:00
parent ad95105417
commit 29d8d5fb94
7 changed files with 40 additions and 10 deletions
+2
View File
@@ -13,6 +13,8 @@ logger = get_logger(__name__)
# 配置离线模式,避免联网检查
os.environ['TRANSFORMERS_OFFLINE'] = '1'
os.environ['HF_DATASETS_OFFLINE'] = '1'
os.environ['HF_HUB_OFFLINE'] = '1' # 禁用HuggingFace Hub联网
os.environ['SENTENCE_TRANSFORMERS_HOME'] = 'data/models' # 设置模型缓存目录
class MemoryService:
+10 -1
View File
@@ -19,6 +19,15 @@ anthropic==0.72.0
httpx==0.28.1
python-dotenv==1.0.0
# NumPy版本锁定(兼容性要求)
numpy==1.26.4
# 向量数据库和Embedding (长期记忆系统)
chromadb==1.3.2
sentence-transformers>=2.3.1
# Transformers(锁定兼容版本)
transformers==4.35.2
# Sentence Transformers(基于PyTorch的文本embedding库)
sentence-transformers==2.3.1