移除kwargs中的hashing_kv参数取为变量

2024-12-06 15:35:09 +08:00 · 2024-12-06 15:35:09 +08:00 · 4da7dd1865
commit 4da7dd1865
parent efdd4b8b8e
1 changed files with 2 additions and 2 deletions
--- a/lightrag/llm.py
+++ b/lightrag/llm.py
@ -97,7 +97,7 @@ async def openai_complete_if_cache(

    # Save to cache
    await save_to_cache(
-        kwargs.get("hashing_kv"),
+        hashing_kv,
        CacheData(
            args_hash=args_hash,
            content=content,
@ -271,7 +271,7 @@ async def bedrock_complete_if_cache(

    # Save to cache
    await save_to_cache(
-        kwargs.get("hashing_kv"),
+        hashing_kv,
        CacheData(
            args_hash=args_hash,
            content=response["output"]["message"]["content"][0]["text"],