修改环境配置支持公司大模型配置
This commit is contained in:
+5
-7
@@ -2,13 +2,11 @@
|
||||
# DB_URL=postgresql+psycopg://ai:ai@localhost:5532/ai
|
||||
|
||||
# OpenAI API配置
|
||||
API_KEY=sk-oftybjqntjpxlhkcxkieluljb
|
||||
EMBEDDING_MODEL=BAAI/bge-large-zh-v1.5
|
||||
EMBEDDING_BASE_URL=https://api.siliconflow.cn/v1
|
||||
MODEL_LIST=Qwen2.5-72B=openai:Qwen/Qwen2.5-72B-Instruct&gpt-4o=openai:gpt-4o
|
||||
|
||||
# 模型配置
|
||||
MODEL_BASE_URL=https://api.siliconflow.cn/v1
|
||||
API_KEY=sk-oftybjqntjpxlhkc
|
||||
EMBEDDING_MODEL=bge-m3
|
||||
EMBEDDING_BASE_URL=http://10.1.16.39:9995/v1
|
||||
MODEL_LIST=Qwen2.5-72B=openai:Qwen2.5-72B-Instruct-GPTQ-Int8
|
||||
MODEL_BASE_URL=http://172.20.0.145:9995/v1
|
||||
|
||||
# 文件路径配置
|
||||
KNOWLEDGE_SOURCE_DIR=data
|
||||
|
||||
@@ -31,6 +31,7 @@
|
||||
from pathlib import Path
|
||||
|
||||
from agno.document.chunking.document import DocumentChunking
|
||||
from agno.models.deepseek import DeepSeek
|
||||
from dotenv import load_dotenv
|
||||
|
||||
# 加载.env文件
|
||||
@@ -107,6 +108,8 @@ def get_model_by_provider(provider: str, model_name: str):
|
||||
# return Claude(id=model_name)
|
||||
# elif provider == "groq":
|
||||
# return Groq(id=model_name)
|
||||
elif provider == "deepseek":
|
||||
return DeepSeek(id=model_name, base_url=model_baseUrl, api_key=api_key)
|
||||
else:
|
||||
raise ValueError(f"Unsupported model provider: {provider}")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user