gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Make ruff happy

Browse source
This commit is contained in:
oryx1729 2025-09-20 17:02:31 -07:00
parent 766b300fbc
commit 1f63b6db55
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/openai/adapter.py
View file

@ -32,6 +32,7 @@ logger = get_logger()
# litellm to drop unsupported params, e.g., reasoning_effort when not supported by the model. # litellm to drop unsupported params, e.g., reasoning_effort when not supported by the model.
litellm.drop_params = True litellm.drop_params = True
class OpenAIAdapter(LLMInterface): class OpenAIAdapter(LLMInterface):
""" """
Adapter for OpenAI's GPT-3, GPT-4 API. Adapter for OpenAI's GPT-3, GPT-4 API.
@ -140,7 +141,7 @@ class OpenAIAdapter(LLMInterface):
ContentFilterFinishReasonError, ContentFilterFinishReasonError,
ContentPolicyViolationError, ContentPolicyViolationError,
InstructorRetryException, InstructorRetryException,
) as error: ):
if not (self.fallback_model and self.fallback_api_key): if not (self.fallback_model and self.fallback_api_key):
raise ContentPolicyFilterError( raise ContentPolicyFilterError(
f"The provided input contains content that is not aligned with our content policy: {text_input}" f"The provided input contains content that is not aligned with our content policy: {text_input}"