from langchain_core.output_parsers import StrOutputParser, JsonOutputParser, SimpleJsonOutputParser
from langchain_core.prompts import PromptTemplate

from langchain_openai import ChatOpenAI

from env_util import DASHSCOPE_API_KEY, DASHSCOPE_BASE_URL

# llm对象就是调用大模型的对象
llm = ChatOpenAI(
    model = "qwen-plus",
    base_url=DASHSCOPE_BASE_URL,
    api_key=DASHSCOPE_API_KEY,
    temperature=0.8,
);

# 提示词模板
prompt_template = PromptTemplate.from_template("帮我生成一个简短的，关于{topic}的报幕词。")

# 指定字符串结构化输出
chain = prompt_template | llm | StrOutputParser()

# 调用大模型，然后给提示词模板赋值
resp = chain.invoke({"topic","迎新生晚会"})

# 输出结果
print(resp)