langchain_google_vertexai_chat_model_quickstart_with_gemini.py

1import getpass
2import os
3
4from langchain_google_vertexai import ChatVertexAI
5from langchain_core.messages import HumanMessage
6
7# Set up credentials if not already configured in your environment
8if "GOOGLE_APPLICATION_CREDENTIALS" not in os.environ:
9    os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = "/path/to/your/service-account-file.json"
10
11# Initialize the Chat Model
12llm = ChatVertexAI(model_name="gemini-1.5-flash")
13
14# Define a message
15message = HumanMessage(content="Translate 'Hello, how are you?' into French.")
16
17# Invoke the model
18response = llm.invoke([message])
19
20# Print the result
21print(response.content)