Vertexai & Langchain

[pgvetrivel]

Hi Team,

I am new to langchain. I am facing problem in importing vertexai.

from langchain_google_vertexai import VertexAI
from google.cloud import aiplatform

# Initialize Vertex AI
aiplatform.init(project='my-project', location='us-central1')  

# Create an instance of the VertexAI LLM
vertex_ai_llm = VertexAI(model_name="gemini-1.5-flash-002")

it throws the following error.
PydanticUndefinedAnnotation: name 'SafetySetting' is not defined

My Versions:
langchain: 0.3.7
langchain-core: 0.3.19
langchain-google-genai : 2.0.5
langchain-google-vertexai : 2.0.7
python :3.12.4

i am having google application credential in json file and already accessed it in the code.
i even tried in virtual environment. still i face the problem. please guide me

[feijoes]

@pgvetrivel It can be resolved by downgrading to pydantic==2.9.0

[pgvetrivel]

Thank you. Got Resolved