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],