ValueError: Model ollama/llama2:latest does not support response schema

[irtiq7]

Describe the bug
When I generating dataset using ollama, I get the above message.

To Reproduce

Run the following code:

import plexe

MODEL_ID = "ollama/llama2:latest"

# Generate synthetic data
dataset = plexe.DatasetGenerator(
    description="Example dataset with features and target",
    provider=MODEL_ID,
    schema={"features": str, "target": int}
)
dataset.generate(500)  # Generate 500 samples
print("Generated dataset:", dataset)

Expected behavior

Got this error:
ValueError: Model ollama/llama2:latest does not support response schema

[Maharshi2403]

Hi @irtiq7,
I successfully reproduced the error using the model "ollama/llama2:latest". The issue occurs specifically in the validation check where supports_response_schema(model=self.model) evaluates to False, indicating that LiteLLM does not currently support response schemas for this model.
For reference, here's the relevant code snippet where the error is raised:
if "response_format" not in litellm.get_supported_openai_params(model=self.model): raise ValueError(f"Model {self.model} does not support passing response_format") if not supports_response_schema(model=self.model): raise ValueError(f"Model {self.model} does not support response schema")