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

conductor-checkpoint-msg_01SwJkCDAScffk8116KPVpTd

Browse source
This commit is contained in:
Daniel Chalef 2025-10-29 23:09:38 -07:00
parent 8a24aa16c6
commit f88a5c4ed2
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
graphiti_core/llm_client/openai_client.py
View file

@ -73,13 +73,16 @@ class OpenAIClient(BaseOpenAIClient):
verbosity: str | None = None, verbosity: str | None = None,
): ):
"""Create a structured completion using OpenAI's beta parse API.""" """Create a structured completion using OpenAI's beta parse API."""
# Only pass reasoning parameter for reasoning models (gpt-5 family: o1, o3-mini, o3, gpt-5-*)
is_reasoning_model = model.startswith('gpt-5') or model.startswith('o1') or model.startswith('o3')
response = await self.client.responses.parse( response = await self.client.responses.parse(
model=model, model=model,
input=messages, # type: ignore input=messages, # type: ignore
temperature=temperature, temperature=temperature,
max_output_tokens=max_tokens, max_output_tokens=max_tokens,
text_format=response_model, # type: ignore text_format=response_model, # type: ignore
reasoning={'effort': reasoning} if reasoning is not None else None, # type: ignore reasoning={'effort': reasoning} if reasoning is not None and is_reasoning_model else None, # type: ignore
text={'verbosity': verbosity} if verbosity is not None else None, # type: ignore text={'verbosity': verbosity} if verbosity is not None else None, # type: ignore
) )