From 6e2946e78aa1213d23749aac23fe3513f5c957db Mon Sep 17 00:00:00 2001
From: yangdx <gzdaniel@me.com>
Date: Fri, 28 Nov 2025 13:41:01 +0800
Subject: [PATCH] Add max_token_size parameter to azure_openai_embed wrapper

---
 lightrag/llm/openai.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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,