File size: 761 Bytes
24c1914
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
# import openai
from langchain_openai.llms import OpenAI
from dotenv import load_dotenv
import os

from langchain_core.prompts import PromptTemplate
from langchain.chains import LLMChain


# os.environ["OPENAPI_KEY"]='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'
load_dotenv()
api_key = os.environ['OPEN_API_KEY'];
llm = OpenAI(openai_api_key=api_key)

mytemplate = """Question: {question}
Answer: Let's think step by step."""

# prompt = PromptTemplate.from_template(mytemplate)

# mychain = LLMChain(llm=llm, prompt=prompt)

# question ="who is fastest man on earth?"
# print(mychain.run(question))

result= llm.generate(['two hit movies of bollywood ', 'two flop movies of bollywood'])

print(result.llm_output)
print(result.generations[1][0].text)