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

fix: Resolve issues with GPT5 models

Browse source
This commit is contained in:
Igor Ilic 2025-09-29 14:11:06 +02:00
parent 74f7a65110
commit f2e216cdf7
1 changed files with 2 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -29,9 +29,6 @@ observe = get_observe()
logger = get_logger() logger = get_logger()
# litellm to drop unsupported params, e.g., reasoning_effort when not supported by the model.
litellm.drop_params = True
class OpenAIAdapter(LLMInterface): class OpenAIAdapter(LLMInterface):
""" """
@ -76,8 +73,8 @@ class OpenAIAdapter(LLMInterface):
fallback_api_key: str = None, fallback_api_key: str = None,
fallback_endpoint: str = None, fallback_endpoint: str = None,
): ):
self.aclient = instructor.from_litellm(litellm.acompletion) self.aclient = instructor.from_litellm(litellm.acompletion, mode=instructor.Mode.JSON)
self.client = instructor.from_litellm(litellm.completion) self.client = instructor.from_litellm(litellm.completion, mode=instructor.Mode.JSON)
self.transcription_model = transcription_model self.transcription_model = transcription_model
self.model = model self.model = model
self.api_key = api_key self.api_key = api_key
@ -135,7 +132,6 @@ class OpenAIAdapter(LLMInterface):
api_version=self.api_version, api_version=self.api_version,
response_model=response_model, response_model=response_model,
max_retries=self.MAX_RETRIES, max_retries=self.MAX_RETRIES,
reasoning_effort="minimal",
) )
except ( except (
ContentFilterFinishReasonError, ContentFilterFinishReasonError,
@ -223,7 +219,6 @@ class OpenAIAdapter(LLMInterface):
api_base=self.endpoint, api_base=self.endpoint,
api_version=self.api_version, api_version=self.api_version,
response_model=response_model, response_model=response_model,
reasoning_effort="minimal",
max_retries=self.MAX_RETRIES, max_retries=self.MAX_RETRIES,
) )