Skip to content

Commit c0ff5ae

Browse files
authored
Merge pull request #23 from Shekswess/develop
[Fix] Missed check for response schema for OpenAI models and Anthropic models
2 parents 88bdb23 + 10c93e8 commit c0ff5ae

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

synthgenai/llm.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -232,6 +232,9 @@ def check_response_format(self) -> bool:
232232
custom_llm_provider = None
233233
if self.model.startswith("gpt") or self.model.startswith("claude"):
234234
if "response_format" in get_supported_openai_params(
235+
model=self.model,
236+
custom_llm_provider=custom_llm_provider,
237+
) and supports_response_schema(
235238
model=self.model, custom_llm_provider=custom_llm_provider
236239
):
237240
logger.info(f"JSON format is supported by the LLM model: {self.model}")

0 commit comments

Comments
 (0)