From 807a402ba40e8857cf3220e8446c16551459e090 Mon Sep 17 00:00:00 2001
From: Pavlo Paliychuk <paul@getzep.com>
Date: Mon, 7 Apr 2025 12:33:15 -0400
Subject: [PATCH] Add set max_tokens attribute in OpenAIClient initialization
 (#330)

---
 graphiti_core/llm_client/openai_client.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/graphiti_core/llm_client/openai_client.py b/graphiti_core/llm_client/openai_client.py
index 45b89a0f..6abf90f9 100644
--- 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],