diff --git a/lightrag/llm/openai.py b/lightrag/llm/openai.py
index 5ab2ac87..9c3d0261 100644
--- a/lightrag/llm/openai.py
+++ b/lightrag/llm/openai.py
@@ -867,7 +867,7 @@ async def azure_openai_complete(
     return result
 
 
-@wrap_embedding_func_with_attrs(embedding_dim=1536)
+@wrap_embedding_func_with_attrs(embedding_dim=1536, max_token_size=8192)
 async def azure_openai_embed(
     texts: list[str],
     model: str | None = None,