Lang-Smith-Implementation / 2_sequential_chain.py
Abhisingh-18's picture
Mirror of github.com/Abhisingh18/Lang-Smith-Implementation
d8ec195 verified
Raw
History Blame Contribute Delete
655 Bytes
from langchain_openai import ChatOpenAI
from dotenv import load_dotenv
from langchain_core.prompts import PromptTemplate
from langchain_core.output_parsers import StrOutputParser
load_dotenv()
prompt1 = PromptTemplate(
template='Generate a detailed report on {topic}',
input_variables=['topic']
)
prompt2 = PromptTemplate(
template='Generate a 5 pointer summary from the following text \n {text}',
input_variables=['text']
)
model = ChatOpenAI()
parser = StrOutputParser()
chain = prompt1 | model | parser | prompt2 | model | parser
result = chain.invoke({'topic': 'Unemployment in India'})
print(result)