import os

from langchain.chat_models import init_chat_model
from langchain_core.messages import HumanMessage, SystemMessage

os.environ['OPENAI_BASE_URL']='https://dashscope.aliyuncs.com/compatible-mode/v1'
os.environ['OPENAI_API_KEY']=os.getenv("DASHSCOPE_API_KEY")

model=init_chat_model('qwen-plus',model_provider='openai')

# 方法1：单条消息，调用llm
print(model.invoke('hello').content)

# 方法2：多组消息数据，调用llm
messages = [
            {"role": "system", "content": "回复不超过10个字"},
            {"role": "user", "content": "hello"}
        ]
print(model.invoke(messages).content)

# 方法3：使用message对象
messages=[
    SystemMessage("回复不超过10个字"),
    HumanMessage("hello")
]
print(model.invoke(messages).content)