83 lines
3.2 KiB
Python
83 lines
3.2 KiB
Python
# config.py
|
|
import os
|
|
from pathlib import Path
|
|
|
|
# --- OpenAI API Configuration ---
|
|
|
|
# !! 安全警告 !! 直接将 API 密钥写入代码风险很高。请优先考虑使用环境变量。
|
|
# !! SECURITY WARNING !! Hardcoding API keys is highly discouraged due to security risks. Prefer environment variables.
|
|
# 如果你确定要硬编码，请取消下一行的注释并填入你的密钥
|
|
# OPENAI_API_KEY_CONFIG = "sk-YOUR_REAL_API_KEY_HERE" # <--- 在这里直接填入你的 OpenAI Key
|
|
|
|
# 如果 OPENAI_API_KEY_CONFIG 未定义 (被注释掉了), 则尝试从环境变量获取
|
|
# This provides a fallback mechanism, but the primary request was to hardcode.
|
|
# Uncomment the line above and fill it to hardcode the key.
|
|
# OPENAI_API_KEY_FROM_CONFIG = os.getenv("OPENAI_API_KEY", "YOUR_API_KEY_PLACEHOLDER_IF_NOT_IN_ENV") # Fallback if not hardcoded above
|
|
# If you absolutely want to force using only a hardcoded key from here, use:
|
|
OPENAI_API_KEY_FROM_CONFIG = os.getenv("DOUBAO_API_KEY", "YOUR_API_KEY_PLACEHOLDER_IF_NOT_IN_ENV")
|
|
|
|
|
|
# 直接在此处配置 API base URL
|
|
# 设置为 None 使用官方地址, 或设置为你的代理/Azure 地址
|
|
# Example for a proxy: "http://localhost:1234/v1"
|
|
# Example for Azure: "https://YOUR_RESOURCE_NAME.openai.azure.com/"
|
|
OPENAI_API_BASE_URL_CONFIG = (
|
|
"https://ark.cn-beijing.volces.com/api/v3" # <--- 在这里直接填入你的 Base URL 或者 None
|
|
)
|
|
|
|
# --- 选择模型和维度 ---
|
|
OPENAI_EMBEDDING_MODEL = "doubao-embedding-large-text-250515"
|
|
OPENAI_EMBEDDING_DIM = 2048
|
|
OPENAI_EMBEDDING_KEY = os.getenv("DOUBAO_API_KEY", "YOUR_API_KEY_PLACEHOLDER_IF_NOT_IN_ENV")
|
|
OPENAI_EMBEDDING_BASE = "https://ark.cn-beijing.volces.com/api/v3"
|
|
OPENAI_LLM_MODEL = "doubao-seed-1-6-250615"
|
|
|
|
# ---huggingface模型---
|
|
HUGGINGFACE_KEY = "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"
|
|
HUGGINGFACE_EMBEDDING_MODEL = "BAAI/bge-m3"
|
|
|
|
# --- 豆包 TTS Configuration ---
|
|
DOUBAO_TTS_API_URL = "https://openspeech.bytedance.com/api/v3/tts/unidirectional"
|
|
DOUBAO_TTS_APP_ID = os.getenv("DOUBAO_TTS_APP_ID", "3842625790")
|
|
DOUBAO_TTS_ACCESS_KEY = os.getenv("DOUBAO_TTS_KEY", "YOUR_ACCESS_KEY")
|
|
DOUBAO_TTS_RESOURCE_ID = "seed-tts-1.0" # 豆包语音合成模型1.0 字符版
|
|
DOUBAO_TTS_SPEAKER = "ICL_zh_female_aojiaonvyou_tob"
|
|
DOUBAO_TTS_FORMAT = "mp3"
|
|
DOUBAO_TTS_SAMPLE_RATE = 24000
|
|
|
|
# --- Milvus Lite Configuration (保持不变) ---
|
|
USER_ID_PREFIX = "user"
|
|
MILVUS_PERSIST_BASE_DIR = Path("./milvus_user_data_openai")
|
|
MILVUS_INDEX_PARAMS = {"index_type": "FLAT", "metric_type": "L2", "params": {}}
|
|
MILVUS_SEARCH_PARAMS = {"metric_type": "L2", "params": {}}
|
|
MILVUS_STAND_URI = ""
|
|
|
|
MEM0_CONFIG = {
|
|
"vector_store": {
|
|
"provider": "milvus",
|
|
"config": {
|
|
"embedding_model_dims": 2048,
|
|
}
|
|
},
|
|
"llm": {
|
|
"provider": "openai",
|
|
"config": {
|
|
"api_key": OPENAI_API_KEY_FROM_CONFIG,
|
|
"model": "doubao-seed-1-6-250615",
|
|
"openai_base_url": OPENAI_API_BASE_URL_CONFIG
|
|
}
|
|
},
|
|
"embedder": {
|
|
"provider": "openai",
|
|
"config": {
|
|
"api_key": OPENAI_EMBEDDING_KEY,
|
|
"model": "doubao-embedding-large-text-250515",
|
|
"openai_base_url": OPENAI_EMBEDDING_BASE
|
|
}
|
|
},
|
|
}
|
|
|
|
|
|
# --- Application Settings (保持不变) ---
|
|
DEFAULT_USER_ID = "user_openai"
|