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)