diff --git a/eventual-consistency-service/app/graphrag/graph_rag.py b/eventual-consistency-service/app/graphrag/graph_rag.py index 4f8ccc61..1d8f6084 100644 --- a/eventual-consistency-service/app/graphrag/graph_rag.py +++ b/eventual-consistency-service/app/graphrag/graph_rag.py @@ -88,12 +88,6 @@ async def chunk_docs( doc_tasks = [] async with asyncio.TaskGroup() as grp: async for content in docs_chan: - # v_id = content["v_id"] - # txt = content["attributes"]["text"] - # send the document to be embedded - logger.info("chunk writes to extract") - # await embed_chan.put((v_id, txt, "Document")) - task = grp.create_task( workers.chunk_doc(conn, content, upsert_chan, embed_chan, extract_chan) ) diff --git a/eventual-consistency-service/app/graphrag/workers.py b/eventual-consistency-service/app/graphrag/workers.py index d696df8b..98b3e69c 100644 --- a/eventual-consistency-service/app/graphrag/workers.py +++ b/eventual-consistency-service/app/graphrag/workers.py @@ -377,7 +377,7 @@ async def process_community( summarizer = community_summarizer.CommunitySummarizer(llm) summary = await summarizer.summarize(comm_id, children) - logger.debug(f"*******>{comm_id}: {children}, {summary}") + logger.debug(f"Community {comm_id}: {children}, {summary}") await upsert_chan.put( ( util.upsert_vertex, # func to call