sidagarwal04 commited on
Commit
0040bb8
·
verified ·
1 Parent(s): 393494f

Delete test.py

Browse files
Files changed (1) hide show
  1. test.py +0 -21
test.py DELETED
@@ -1,21 +0,0 @@
1
- import os
2
-
3
- from dotenv import load_dotenv
4
- from langchain_google_genai import ChatGoogleGenerativeAI
5
- from langchain.prompts import PromptTemplate
6
- from langchain.chains import LLMChain
7
-
8
- load_dotenv()
9
-
10
- GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
11
-
12
- llm = ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=GOOGLE_API_KEY)
13
-
14
- tweet_prompt = PromptTemplate.from_template("You are a content creator. Write me a tweet about {topic}.")
15
-
16
- tweet_chain = LLMChain(llm=llm, prompt=tweet_prompt, verbose=True)
17
-
18
- if __name__=="__main__":
19
- topic = "how ai is really cool"
20
- resp = tweet_chain.run(topic=topic)
21
- print(resp)