chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,14 @@
|
||||
from langchain_core.output_parsers import StrOutputParser
|
||||
from langchain_core.prompts import PromptTemplate
|
||||
from langchain_openai import ChatOpenAI
|
||||
|
||||
import mlflow
|
||||
|
||||
prompt = PromptTemplate(
|
||||
input_variables=["product"],
|
||||
template="What is {product}?",
|
||||
)
|
||||
llm = ChatOpenAI(temperature=0.1, stream_usage=True)
|
||||
chain = prompt | llm | StrOutputParser()
|
||||
|
||||
mlflow.models.set_model(chain)
|
||||
Reference in New Issue
Block a user