Project5 / Lang_chain.py
Krithikesh77's picture
Upload 4 files
2cb3d78 verified
raw
history blame contribute delete
410 Bytes
from langchain_google_genai import ChatGoogleGenerativeAI
from dotenv import load_dotenv
import os
load_dotenv()
api_key=os.getenv("GOOGLE_API_KEY")
model="gemini-2.5-flash"
def hello_langchain(prompt, history):
llm = ChatGoogleGenerativeAI(api_key=api_key, model=model)
response = llm.invoke(prompt)
return response.content
if __name__=="__main__":
print(hello_langchain("Hello"))