新增标题提取
This commit is contained in:
@@ -35,7 +35,6 @@ async def chat_stream(request: DeepAgentChatRequest):
|
|||||||
* `checkpoint_id`: (可选) 历史快照 ID。
|
* `checkpoint_id`: (可选) 历史快照 ID。
|
||||||
* `config_params`: (可选) 对话配置参数
|
* `config_params`: (可选) 对话配置参数
|
||||||
* `need_suggestion`: (可选) 是否需要建议按钮,需要建议的频率,0-1的浮点数
|
* `need_suggestion`: (可选) 是否需要建议按钮,需要建议的频率,0-1的浮点数
|
||||||
* `need_title`: (可选) 是否需要标题
|
|
||||||
* `use_report`: (可选) 是否需要使用report功能 true/false
|
* `use_report`: (可选) 是否需要使用report功能 true/false
|
||||||
|
|
||||||
|
|
||||||
@@ -71,6 +70,10 @@ async def chat_stream(request: DeepAgentChatRequest):
|
|||||||
|
|
||||||
"""
|
"""
|
||||||
logger.info(f"chat request data: {request}")
|
logger.info(f"chat request data: {request}")
|
||||||
|
if request.thread_id:
|
||||||
|
need_title = False
|
||||||
|
else:
|
||||||
|
need_title = True
|
||||||
source_thread_id = request.thread_id
|
source_thread_id = request.thread_id
|
||||||
checkpoint_id = request.checkpoint_id
|
checkpoint_id = request.checkpoint_id
|
||||||
|
|
||||||
@@ -259,7 +262,7 @@ async def chat_stream(request: DeepAgentChatRequest):
|
|||||||
yield f"data: {json.dumps({'suggested_questions': suggested_questions}, ensure_ascii=False)}\n\n"
|
yield f"data: {json.dumps({'suggested_questions': suggested_questions}, ensure_ascii=False)}\n\n"
|
||||||
|
|
||||||
# 获取标题
|
# 获取标题
|
||||||
if request.need_title:
|
if need_title:
|
||||||
title = await conversation_title(agent=main_agent, config=current_config)
|
title = await conversation_title(agent=main_agent, config=current_config)
|
||||||
yield f"data: {json.dumps({'title': title}, ensure_ascii=False)}\n\n"
|
yield f"data: {json.dumps({'title': title}, ensure_ascii=False)}\n\n"
|
||||||
|
|
||||||
|
|||||||
@@ -15,7 +15,6 @@ class DeepAgentChatRequest(BaseModel):
|
|||||||
checkpoint_id: Optional[str] = Field(None, description="回溯点的ID,用于从历史点开启新对话")
|
checkpoint_id: Optional[str] = Field(None, description="回溯点的ID,用于从历史点开启新对话")
|
||||||
config_params: Optional[AgentConfig] = None
|
config_params: Optional[AgentConfig] = None
|
||||||
need_suggestion: float = 0
|
need_suggestion: float = 0
|
||||||
need_title: bool = False
|
|
||||||
use_report: bool = False # ← 新增:是否使用深度报告
|
use_report: bool = False # ← 新增:是否使用深度报告
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user