Add set max_tokens attribute in OpenAIClient initialization (#330)

2025-04-07 12:33:15 -04:00 · 2025-04-07 12:33:15 -04:00 · 807a402ba4
commit 807a402ba4
parent 87fbf31a78
1 changed files with 2 additions and 0 deletions
--- a/graphiti_core/llm_client/openai_client.py
+++ b/graphiti_core/llm_client/openai_client.py
@ -87,6 +87,8 @@ class OpenAIClient(LLMClient):
        else:
            self.client = client

+        self.max_tokens = max_tokens
+
    async def _generate_response(
        self,
        messages: list[Message],