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

2024-12-06 15:35:09 +08:00
parent a1c4a036fd
commit 6a010abb62
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"],