diff --git a/pilot/base_modules/agent/controller.py b/pilot/base_modules/agent/controller.py index ef699a047..e0c0ffaf8 100644 --- a/pilot/base_modules/agent/controller.py +++ b/pilot/base_modules/agent/controller.py @@ -83,7 +83,7 @@ async def agent_hub_update(update_param: PluginHubParam = Body()): return Result.succ(None) except Exception as e: logger.error("Agent Hub Update Error!", e) - return Result.faild(code="E0020", msg=f"Agent Hub Update Error! {e}") + return Result.failed(code="E0020", msg=f"Agent Hub Update Error! {e}") @router.post("/v1/agent/query", response_model=Result[str]) @@ -133,7 +133,7 @@ async def agent_install(plugin_name: str, user: str = None): return Result.succ(None) except Exception as e: logger.error("Plugin Install Error!", e) - return Result.faild(code="E0021", msg=f"Plugin Install Error {e}") + return Result.failed(code="E0021", msg=f"Plugin Install Error {e}") @router.post("/v1/agent/uninstall", response_model=Result[str]) @@ -147,7 +147,7 @@ async def agent_uninstall(plugin_name: str, user: str = None): return Result.succ(None) except Exception as e: logger.error("Plugin Uninstall Error!", e) - return Result.faild(code="E0022", msg=f"Plugin Uninstall Error {e}") + return Result.failed(code="E0022", msg=f"Plugin Uninstall Error {e}") @router.post("/v1/personal/agent/upload", response_model=Result[str]) @@ -160,4 +160,4 @@ async def personal_agent_upload(doc_file: UploadFile = File(...), user: str = No return Result.succ(None) except Exception as e: logger.error("Upload Personal Plugin Error!", e) - return Result.faild(code="E0023", msg=f"Upload Personal Plugin Error {e}") + return Result.failed(code="E0023", msg=f"Upload Personal Plugin Error {e}") diff --git a/pilot/scene/chat_knowledge/refine_summary/chat.py b/pilot/scene/chat_knowledge/refine_summary/chat.py index d0b1e9471..f5418bf7a 100644 --- a/pilot/scene/chat_knowledge/refine_summary/chat.py +++ b/pilot/scene/chat_knowledge/refine_summary/chat.py @@ -32,6 +32,19 @@ class ExtractRefineSummary(BaseChat): } return input_values + # async def stream_call(self): + # last_output = None + # async for output in super().stream_call(): + # last_output = output + # yield output + # + # last_output = f"{last_output}" + # yield last_output + + def stream_call_reinforce_fn(self, text): + """return reference""" + return f"{text}" + @property def chat_type(self) -> str: return ChatScene.ExtractRefineSummary.value diff --git a/pilot/server/knowledge/api.py b/pilot/server/knowledge/api.py index 7e778b860..2fce49eb4 100644 --- a/pilot/server/knowledge/api.py +++ b/pilot/server/knowledge/api.py @@ -245,15 +245,8 @@ async def document_summary(request: DocumentSummaryRequest): headers=headers, media_type="text/plain", ) - - # return Result.succ( - # knowledge_space_service.create_knowledge_document( - # space=space_name, request=request - # ) - # ) - # return Result.succ([]) except Exception as e: - return Result.faild(code="E000X", msg=f"document summary error {e}") + return Result.failed(code="E000X", msg=f"document summary error {e}") @router.post("/knowledge/entity/extract") @@ -276,4 +269,4 @@ async def entity_extract(request: EntityExtractRequest): ) return Result.succ(res) except Exception as e: - return Result.faild(code="E000X", msg=f"entity extract error {e}") + return Result.failed(code="E000X", msg=f"entity extract error {e}") diff --git a/pilot/server/knowledge/service.py b/pilot/server/knowledge/service.py index efe096e40..046192d43 100644 --- a/pilot/server/knowledge/service.py +++ b/pilot/server/knowledge/service.py @@ -576,6 +576,9 @@ class KnowledgeService: reduce -> merge the summaries by map process Args: docs:List[str] + model_name:model name str + max_iteration:max iteration will call llm to summary + concurrency_limit:the max concurrency threads to call llm """ from pilot.scene.base import ChatScene from pilot.common.chat_util import llm_chat_response_nostream