StochasticAI

随机加速平台旨在简化深度学习模型的生命周期。从上传和版本控制模型，通过训练、压缩和加速，到将其投入生产。

本示例介绍如何使用LangChain与StochasticAI模型进行交互。

您需要在此处获取API_KEY和API_URL。

from getpass import getpass

STOCHASTICAI_API_KEY = getpass()

     ········

import os

os.environ["STOCHASTICAI_API_KEY"] = STOCHASTICAI_API_KEY

YOUR_API_URL = getpass()

     ········

from langchain.llms import StochasticAI
from langchain import PromptTemplate, LLMChain

template = """问题：{question}

回答：让我们一步一步来思考。"""

prompt = PromptTemplate(template=template, input_variables=["question"])

llm = StochasticAI(api_url=YOUR_API_URL)

llm_chain = LLMChain(prompt=prompt, llm=llm)

question = "贾斯汀·比伯出生的那一年，哪支NFL球队赢得了超级碗？"

llm_chain.run(question)

    "\n\n第一步：1999年，圣路易斯公羊队赢得了超级碗。\n\n第二步：1999年，比伯出生。\n\n第三步：当时公羊队在洛杉矶。\n\n第四步：所以他们那年没有参加超级碗。\n"