View file src/colab/langchain_chat.py - Download
# -*- coding: utf-8 -*-
"""langchain_chat.ipynb
Automatically generated by Colaboratory.
Original file is located at
https://colab.research.google.com/drive/1SQQB4iiNYdAOsNVd9mp_3MIeCkZPRaPA
## Text generation with LangChain
"""
# Commented out IPython magic to ensure Python compatibility.
!pip install openai
!pip install langchain
from openai import OpenAI
from langchain.chat_models import ChatOpenAI
from langchain.prompts import ChatPromptTemplate
# Put your OpenAI key below
# %env OPENAI_API_KEY=
model = ChatOpenAI()
prompt = ChatPromptTemplate.from_template("tell me a joke about {foo}")
chain = prompt | model
print(chain.invoke({"foo": "bears"}).content)