yujunjun/LightRAG
1
0
Fork 0
mirror of https://github.com/HKUDS/LightRAG.git synced 2025-08-31 20:03:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

update max_token_size according to openai doc

Browse Source
This commit is contained in:
Ahmad Hatahet 2024-11-30 17:16:07 +01:00
parent 7fea7d7b5e
commit 23cabbe7a3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lightrag/llm.py
View File

@ -556,7 +556,7 @@ async def openai_embedding(
return np.array([dp.embedding for dp in response.data]) return np.array([dp.embedding for dp in response.data])
@wrap_embedding_func_with_attrs(embedding_dim=1536, max_token_size=8192) @wrap_embedding_func_with_attrs(embedding_dim=1536, max_token_size=8191)
@retry( @retry(
stop=stop_after_attempt(3), stop=stop_after_attempt(3),
wait=wait_exponential(multiplier=1, min=4, max=10), wait=wait_exponential(multiplier=1, min=4, max=10),