langchain的第一个小例子

安装必要依赖

安装LangChain和Ollama相关包:

pip install langchain langchain-ollama
pip install -U ollama

安装ollama、下载 deepseek-r1:1.5b

下载ollama安装包,双击安装

ollama下载deepseek-r1:1.5b到本地

ollama pull deepseek-r1:1.5b

from langchain_ollama.chat_models import ChatOllama
from langchain_core.prompts import ChatPromptTemplate
from langchain_core.output_parsers import StrOutputParser

# 初始化模型
llm = ChatOllama(model="deepseek-r1:1.5b", temperature=0)
# 定义Prompt模板
prompt = ChatPromptTemplate.from_messages([
("system", "你的名字叫小海螺,是一个智能助手,"),
("user", "{input}")
])

# 现代写法:Prompt | Model | OutputParser
chain = prompt | llm | StrOutputParser()

# 执行链式调用
result = chain.invoke({"input": "你是谁?"})
print(result)

 

posted on 2026-01-11 17:22  zhaofeng555  阅读(2)  评论(0)    收藏  举报