test_setup.py

import os
from dotenv import load_dotenv
from langchain_openai import ChatOpenAI

# 1. 加载 .env 里的 API Key
# 这行代码会自动寻找同级目录下的 .env 文件并读取
load_dotenv()

# 检查 key 是否加载成功 (为了安全，只打印前5位)
api_key = os.getenv("OPENAI_API_KEY")
if not api_key:
    print("❌ 错误: 未找到 API Key，请检查 .env 文件！")
else:
    print(f"✅ API Key 加载成功: {api_key[:5]}...")

# 2. 初始化模型
# 这里我们用 gpt-4o-mini，它便宜且速度快，非常适合测试
print("正在呼叫 OpenAI...")
try:
    llm = ChatOpenAI(model="gpt-4o-mini")

    # 3. 发送一个测试问题
    response = llm.invoke(
        "你好，请用一句话通过'机会成本'的概念解释为什么我要做Side Project？"
    )

    # 4. 打印结果
    print("-" * 30)
    print("🤖 AI 回复:")
    print(response.content)
    print("-" * 30)
    print("🎉 环境测试完美通过！")

except Exception as e:
    print(f"❌ 连接失败: {e}")