1  from langchain_core.output_parsers import StrOutputParser
 2  from langchain_core.prompts import PromptTemplate
 3  from langchain_openai import ChatOpenAI
 4  
 5  import mlflow
 6  
 7  prompt = PromptTemplate(
 8      input_variables=["product"],
 9      template="What is {product}?",
10  )
11  llm = ChatOpenAI(temperature=0.1, stream_usage=True)
12  chain = prompt | llm | StrOutputParser()
13  
14  mlflow.models.set_model(chain)