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

Add cache type "extract" for entity extraction

Browse Source
This commit is contained in:
yangdx
2025-02-01 23:05:02 +08:00
parent 2d387fa6de
commit c9481c81b9
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lightrag/operate.py
View File

@@ -391,7 +391,7 @@ async def extract_entities(
arg_hash,
_prompt,
"default",
cache_type="default",
cache_type="extract",
llm=custom_llm,
)
if cached_return:
@@ -407,7 +407,7 @@ async def extract_entities(
res: str = await use_llm_func(input_text)
await save_to_cache(
llm_response_cache,
CacheData(args_hash=arg_hash, content=res, prompt=_prompt),
CacheData(args_hash=arg_hash, content=res, prompt=_prompt, cache_type="extract"),
)
return res

2
lightrag/utils.py
View File

@@ -105,7 +105,7 @@ def compute_args_hash(*args, cache_type: str = None) -> str:
"""Compute a hash for the given arguments.
Args:
*args: Arguments to hash
cache_type: Type of cache (e.g., 'keywords', 'query')
cache_type: Type of cache (e.g., 'keywords', 'query', 'extract')
Returns:
str: Hash string
"""