fix naive_query

2024-12-09 17:45:01 +08:00
parent e083854d47
commit 2ba20910bb
3 changed files with 2 additions and 2 deletions
--- a/lightrag/llm.py
+++ b/lightrag/llm.py
@@ -616,6 +616,7 @@ async def jina_embedding(
    data_list = await fetch_data(url, headers, data)
    return np.array([dp["embedding"] for dp in data_list])

+
@wrap_embedding_func_with_attrs(embedding_dim=2048, max_token_size=512)
@retry(
    stop=stop_after_attempt(3),