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, arg_hash,
_prompt, _prompt,
"default", "default",
cache_type="default", cache_type="extract",
llm=custom_llm, llm=custom_llm,
) )
if cached_return: if cached_return:
@@ -407,7 +407,7 @@ async def extract_entities(
res: str = await use_llm_func(input_text) res: str = await use_llm_func(input_text)
await save_to_cache( await save_to_cache(
llm_response_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 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. """Compute a hash for the given arguments.
Args: Args:
*args: Arguments to hash *args: Arguments to hash
cache_type: Type of cache (e.g., 'keywords', 'query') cache_type: Type of cache (e.g., 'keywords', 'query', 'extract')
Returns: Returns:
str: Hash string str: Hash string
""" """