import os
from langchain_openai import ChatOpenAI
from langchain.chat_models import init_chat_model

from dotenv import load_dotenv
# 加载环境变量
loaded = load_dotenv()
if not loaded:
    print("警告：.env 文件未找到")



model_zp = init_chat_model(
    model="glm-4.5-air",
    model_provider="openai",
    base_url="https://open.bigmodel.cn/api/paas/v4",
    api_key=os.getenv("ZHIPUAI_API_KEY"),
    temperature=0.3,
    streaming=False,
    max_tokens=1024,
    timeout=60,  # 改为 60 秒
    max_retries=3
)

model_zp_ai = ChatOpenAI(
    model="glm-4.5-air",
    api_key=os.getenv("ZHIPUAI_API_KEY"),
    base_url="https://open.bigmodel.cn/api/paas/v4",
    temperature=0.3,
    streaming=False,
)


# content=model_zp.invoke("说说杭州最近的天气")
# print(content)

content=model_zp_ai.invoke("说说杭州最近的天气")
print(content)