跳至主要內容

StochasticAI

Stochastic Acceleration Platform 旨在簡化深度學習模型的生命週期。從上傳和版本控制模型,到訓練、壓縮和加速,再到將其投入生產。

此範例說明如何使用 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.chains import LLMChain
from langchain_community.llms import StochasticAI
from langchain_core.prompts import PromptTemplate
template = """Question: {question}

Answer: Let's think step by step."""

prompt = PromptTemplate.from_template(template)
llm = StochasticAI(api_url=YOUR_API_URL)
llm_chain = LLMChain(prompt=prompt, llm=llm)
question = "What NFL team won the Super Bowl in the year Justin Beiber was born?"

llm_chain.run(question)
"\n\nStep 1: In 1999, the St. Louis Rams won the Super Bowl.\n\nStep 2: In 1999, Beiber was born.\n\nStep 3: The Rams were in Los Angeles at the time.\n\nStep 4: So they didn't play in the Super Bowl that year.\n"

此頁面是否對您有幫助? (Was this page helpful?)