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

Add type ignore for response_format to fix pyright error

Browse source
This commit is contained in:
Daniel Chalef 2025-11-09 20:54:54 -08:00
parent f8a17ce435
commit 552c9d9634
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
graphiti_core/llm_client/openai_generic_client.py
View file

@ -17,7 +17,7 @@ limitations under the License.
import json import json
import logging import logging
import typing import typing
from typing import ClassVar from typing import Any, ClassVar
import openai import openai
from openai import AsyncOpenAI from openai import AsyncOpenAI
@ -100,7 +100,7 @@ class OpenAIGenericClient(LLMClient):
openai_messages.append({'role': 'system', 'content': m.content}) openai_messages.append({'role': 'system', 'content': m.content})
try: try:
# Prepare response format # Prepare response format
response_format = {'type': 'json_object'} response_format: dict[str, Any] = {'type': 'json_object'}
if response_model is not None: if response_model is not None:
schema_name = getattr(response_model, '__name__', 'structured_response') schema_name = getattr(response_model, '__name__', 'structured_response')
json_schema = response_model.model_json_schema() json_schema = response_model.model_json_schema()
@ -117,7 +117,7 @@ class OpenAIGenericClient(LLMClient):
messages=openai_messages, messages=openai_messages,
temperature=self.temperature, temperature=self.temperature,
max_tokens=self.max_tokens, max_tokens=self.max_tokens,
response_format=response_format, response_format=response_format, # type: ignore[arg-type]
) )
result = response.choices[0].message.content or '' result = response.choices[0].message.content or ''
return json.loads(result) return json.loads(result)