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