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),
--- a/lightrag/operate.py
+++ b/lightrag/operate.py
@@ -1103,7 +1103,6 @@ async def naive_query(
    response = await use_model_func(
        query,
        system_prompt=sys_prompt,
-        mode=query_param.mode,
    )

    if len(response) > len(sys_prompt):