Add max_token_size parameter to azure_openai_embed wrapper

This commit is contained in:
yangdx 2025-11-28 13:41:01 +08:00
parent 4f12fe121d
commit 6e2946e78a

View file

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