feat:mapreduce summary

2025-07-31 15:47:05 +00:00 · 2023-10-31 19:33:58 +08:00 · 2023-10-31 19:33:58 +08:00 · 67f41559a8
commit 67f41559a8
parent de90244828
1 changed files with 2 additions and 0 deletions
--- a/pilot/server/knowledge/service.py
+++ b/pilot/server/knowledge/service.py
@ -578,10 +578,12 @@ class KnowledgeService:
        from pilot.common.chat_util import llm_chat_response_nostream
        import uuid
        tasks = []
+        max_iteration = 5
        if len(docs) == 1:
            summary = self._llm_extract_summary(doc=docs[0])
            return summary
        else:
+            max_iteration = max_iteration if len(docs) > max_iteration else len(docs)
            for doc in docs:
                chat_param = {
                    "chat_session_id": uuid.uuid1(),