44 lines
1.1 KiB
Python
44 lines
1.1 KiB
Python
import os
|
||
from dotenv import load_dotenv
|
||
|
||
# 1. 加载.env文件
|
||
load_dotenv()
|
||
|
||
# 2. 解决 UUID v7 警告
|
||
try:
|
||
from langsmith import uuid7
|
||
import uuid
|
||
uuid.uuid4 = uuid7 # 全局替换
|
||
except ImportError:
|
||
pass # 如果没装 langsmith，省略
|
||
|
||
|
||
# 3. 解决TensorFlow & Numpy 兼容性告警
|
||
def silence_framework_warnings():
|
||
import os
|
||
import warnings
|
||
import logging
|
||
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
|
||
warnings.filterwarnings('ignore')
|
||
logging.getLogger('tensorflow').setLevel(logging.ERROR)
|
||
|
||
silence_framework_warnings()
|
||
|
||
|
||
|
||
|
||
# 4. 获取API_KEYS
|
||
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
|
||
LANGCHAIN_API_KEY = os.getenv("LANGCHAIN_API_KEY")
|
||
AMAP_MAPS_API_KEY = os.getenv("AMAP_MAPS_API_KEY")
|
||
CHATGPT_API_KEY = os.getenv("CHATGPT_API_KEY")
|
||
|
||
if not OPENAI_API_KEY:
|
||
raise ValueError("× 请在.env中设置OPENAI_API_KEY")
|
||
if not LANGCHAIN_API_KEY:
|
||
raise ValueError("× 请在.env中设置LANGCHAIN_API_KEY")
|
||
if not AMAP_MAPS_API_KEY:
|
||
raise ValueError("× 请在.env中设置AMAP_MAPS_API_KEY")
|
||
if not CHATGPT_API_KEY:
|
||
raise ValueError("× 请在.env中设置CHATGPT_API_KEY")
|