StochasticAI

This example goes over how to use LangChain to interact with StochasticAI models

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

template = """Question: {question}

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

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

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)