Update factory.py

Second PR for issue #1538
microsoft · Dec 19, 2024 · 3a88ab3 · 3a88ab3
1 parent c1c09ba
commit 3a88ab3
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/graphrag/cache/factory.py b/graphrag/cache/factory.py
@@ -49,7 +49,13 @@ def create_cache(
                     FilePipelineStorage(root_dir=root_dir).child(kwargs["base_dir"])
                 )
             case CacheType.blob:
-                return JsonPipelineCache(BlobPipelineStorage(**kwargs))
+                return JsonPipelineCache(
+                    BlobPipelineStorage(
+                        connection_string=kwargs['connection_string'],
+                        container_name=kwargs['container_name'],
+                        storage_account_blob_url=kwargs['storage_account_blob_url']
+                    ).child(kwargs["base_dir"])
+                )
             case _:
                 if cache_type in cls.cache_types:
                     return cls.cache_types[cache_type](**kwargs)