implementing genAI using langchain future work

yours7himanshu · Jan 14, 2025 · 9cd9f28 · 9cd9f28
1 parent b630f3b
commit 9cd9f28
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/genAI/.gitignore b/genAI/.gitignore
@@ -0,0 +1,2 @@
+.ipynb_checkpoints
+venv
diff --git a/genAI/main.py b/genAI/main.py
@@ -0,0 +1,23 @@
+import os
+from dotenv import load_dotenv
+from langchain_google_vertexai import ChatVertexAI
+from langchain_core.messages import HumanMessage, SystemMessage
+
+
+load_dotenv();
+gimini_api_key=os.getenv("GIMINI_API_KEY");
+langchain_api_key = os.getenv("LANGCHAIN_API_KEY");
+
+model = ChatVertexAI(model="gemini-1.5-flash")
+
+# Define the conversation
+messages = [
+    SystemMessage("Always there to help you"),
+    HumanMessage("hi!"),
+]
+
+# Invoke the model
+response = model.invoke(messages)
+
+# Print the response
+print(response)