neo4j_graphrag.exceptions.SchemaExtractionError: LLM response is not valid JSON when using VertexAI with Simple KGPipeline

[xpilasneo4j]

Similar to #376, I have an exception when loading data using VertexAI about the return type
neo4j_graphrag.exceptions.SchemaExtractionError: LLM response is not valid JSON.

I fixed it by adding in line 98 on the vertexai_llm.py
self.response_mime_type = "application/json"
But I guess it will have to come from the object model_params and for each LLM be translated into the correct options