import os

from dotenv import load_dotenv

from langchain_openai import ChatOpenAI

from langchain_core.messages import SystemMessage,HumanMessage,AIMessage

load_dotenv()

model = ChatOpenAI(

    # 将大模型数据添加到环境变量中

    model=os.environ["MODEL_NAME"],

    api_key=os.environ["API_KEY"],

    base_url=os.environ["BASE_URL"]

)

messages = [

    SystemMessage(content='你是一个乐于助人的 AI 助手'),

    HumanMessage(content='你好，我是reawavetop'),

    AIMessage(content='你好，很高兴认识你'),

    HumanMessage(content='我刚才说我是谁')

]

response = model.invoke(messages)

print(type(response))

print(response.content)