gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix naive_query

Browse Source
This commit is contained in:
Larfii
2024-12-09 17:45:01 +08:00
parent e083854d47
commit 2ba20910bb
3 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
lightrag/llm.py
View File

@@ -616,6 +616,7 @@ async def jina_embedding(
data_list = await fetch_data(url, headers, data) data_list = await fetch_data(url, headers, data)
return np.array([dp["embedding"] for dp in data_list]) return np.array([dp["embedding"] for dp in data_list])
@wrap_embedding_func_with_attrs(embedding_dim=2048, max_token_size=512) @wrap_embedding_func_with_attrs(embedding_dim=2048, max_token_size=512)
@retry( @retry(
stop=stop_after_attempt(3), stop=stop_after_attempt(3),

1
lightrag/operate.py
View File

@@ -1103,7 +1103,6 @@ async def naive_query(
response = await use_model_func( response = await use_model_func(
query, query,
system_prompt=sys_prompt, system_prompt=sys_prompt,
mode=query_param.mode,
) )
if len(response) > len(sys_prompt): if len(response) > len(sys_prompt):