Skip to content

Commit d87c76a

Browse files
Update nano_pipelines.py
1 parent bc6968c commit d87c76a

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

libs/ktem/ktem/index/file/graph/nano_pipelines.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -237,12 +237,13 @@ def build_graphrag(working_dir, llm_func, embedding_func):
237237
class NanoGraphRAGIndexingPipeline(GraphRAGIndexingPipeline):
238238
"""GraphRAG specific indexing pipeline"""
239239

240-
prompts: dict[str, str] = {}
240+
prompts: dict[str, list] = {}
241241

242242
@classmethod
243243
def get_user_settings(cls) -> dict:
244244
try:
245245
from nano_graphrag.prompt import PROMPTS
246+
content: list
246247

247248
blacklist_keywords = ["default", "response", "process"]
248249
return {
@@ -262,6 +263,7 @@ def get_user_settings(cls) -> dict:
262263

263264
def call_graphrag_index(self, graph_id: str, docs: list[Document]):
264265
from nano_graphrag.prompt import PROMPTS
266+
content: list
265267

266268
# modify the prompt if it is set in the settings
267269
for prompt_name, content in self.prompts.items():

0 commit comments

Comments
 (0)