结合 langchain 一起使用的工具,可以显示 web 界面
pip install streamlit duckduckgo-search
运行命令
streamlit run D:\Python_project\NLP\大模型学习\test.py
import os
from dotenv import load_dotenv
from langchain_community.llms import Tongyi
load_dotenv('key.env') # 指定加载 env 文件
key = os.getenv('DASHSCOPE_API_KEY') # 获得指定环境变量
DASHSCOPE_API_KEY = os.environ["DASHSCOPE_API_KEY"] # 获得指定环境变量
model = Tongyi(temperature=1)
from langchain_core.prompts import ChatPromptTemplate
import streamlit as st
st.title('🦜🔗 中文小故事生成器')
prompt = ChatPromptTemplate.from_template("请编写一篇关于{topic}的中文小故事,不超过100字")
chain = prompt | model
with st.form('my_form'):
text = st.text_area('输入主题关键词:', '小白兔')
submitted = st.form_submit_button('提交')
if submitted:
print(st.info(chain.invoke({"topic": text})))
会报错:ModuleNotFoundError: No module named ‘streamlit.cli‘
解決:【Python BUG】ModuleNotFoundError: No module named ‘streamlit.cli‘-CSDN博客
部署雲端可以点击右上角,结合官网示例查看怎么部署