From 8be43c5e2469f17db7b519359497e70186ff277c Mon Sep 17 00:00:00 2001 From: aries_ckt <916701291@qq.com> Date: Wed, 15 Nov 2023 14:10:10 +0800 Subject: [PATCH] feat:add summary --- pilot/server/knowledge/service.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pilot/server/knowledge/service.py b/pilot/server/knowledge/service.py index 046192d43..7fa8c09e8 100644 --- a/pilot/server/knowledge/service.py +++ b/pilot/server/knowledge/service.py @@ -461,8 +461,8 @@ class KnowledgeService: summary = await self._mapreduce_extract_summary( docs=texts, model_name=model_name, - max_iteration=space_context["summary"]["max_iteration"], - concurrency_limit=space_context["summary"]["concurrency_limit"], + max_iteration=int(space_context["summary"]["max_iteration"]), + concurrency_limit=int(space_context["summary"]["concurrency_limit"]), ) else: summary = await self._mapreduce_extract_summary(