Merge branch 'HKUDS:main' into main

lightrag/kg/postgres_impl.py

@@ -1766,7 +1766,7 @@ SQL_TEMPLATES = {
                 FROM LIGHTRAG_DOC_CHUNKS
                 where workspace=$1
                 AND id IN (SELECT chunk_id FROM relevant_chunks)
-            )
+            ) as chunk_distances
             WHERE distance>$2
             ORDER BY distance DESC
             LIMIT $3

lightrag/operate.py

@@ -404,7 +404,7 @@ async def extract_entities(
         language=language,
     )
 
-    continue_prompt = PROMPTS["entity_continue_extraction"]
+    continue_prompt = PROMPTS["entity_continue_extraction"].format(**context_base)
     if_loop_prompt = PROMPTS["entity_if_loop_extraction"]
 
     processed_chunks = 0