chore: rename multiple files to improve importability and module structure
This commit is contained in:
@@ -0,0 +1,27 @@
|
||||
from config import OPENAI_API_KEY
|
||||
from langchain_core.prompts import ChatPromptTemplate
|
||||
from langchain_openai import ChatOpenAI
|
||||
from langchain_core.output_parsers import StrOutputParser
|
||||
|
||||
# 定义提示词模板
|
||||
prompt = ChatPromptTemplate.from_messages([
|
||||
("system", "你非常可爱,说话末尾会带个喵"),
|
||||
("human", "{input}") # {input}:占位符
|
||||
])
|
||||
|
||||
# 初始化模型
|
||||
llm = ChatOpenAI(
|
||||
model="deepseek-chat",
|
||||
api_key=OPENAI_API_KEY,
|
||||
base_url="https://api.deepseek.com"
|
||||
)
|
||||
|
||||
# 定义解析器,把LLM返回的AIMessage转成字符串
|
||||
parser = StrOutputParser()
|
||||
|
||||
# 组成Chain
|
||||
chain = prompt | llm | parser
|
||||
|
||||
# 最终调用
|
||||
result = chain.invoke({"input": "你好喵"})
|
||||
print(result)
|
||||
Reference in New Issue
Block a user