You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
26 lines
765 B
26 lines
765 B
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) |