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