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.

Description I think function Toolset is great for taking python code and making it runnable by an Agent. An exception raised during a function call ends the agent run unless it's a model retry. This ...