perf:del dify stream in ai.yaml config.and enbale stream in lark.yaml.

fix:localagent remove_think bug
This commit is contained in:
Dong_master
2025-07-14 01:42:42 +08:00
committed by Junyan Qin
parent f5a0cb9175
commit f84a79bf74
5 changed files with 20 additions and 43 deletions
+3 -5
View File
@@ -64,7 +64,6 @@ class OpenAIChatCompletions(requester.ProviderAPIRequester):
reasoning_content = chatcmpl_message['reasoning_content'] if 'reasoning_content' in chatcmpl_message else None
# deepseek的reasoner模型
print(pipeline_config['trigger'].get('misc', '').get('remove_think'))
if pipeline_config['trigger'].get('misc', '').get('remove_think'):
pass
else:
@@ -79,6 +78,7 @@ class OpenAIChatCompletions(requester.ProviderAPIRequester):
self,
pipeline_config: dict[str, typing.Any],
chat_completion: chat_completion.ChatCompletion,
idx: int,
) -> llm_entities.MessageChunk:
# 处理流式chunk和完整响应的差异
@@ -106,7 +106,7 @@ class OpenAIChatCompletions(requester.ProviderAPIRequester):
else:
delta['content'] = delta['content']
else:
if reasoning_content is not None:
if reasoning_content is not None and idx == 0:
delta['content'] += f'<think>\n{reasoning_content}'
elif reasoning_content is None:
if self.is_content:
@@ -165,11 +165,10 @@ class OpenAIChatCompletions(requester.ProviderAPIRequester):
pipeline_config = query.pipeline_config
async for chunk in self._req_stream(args, extra_body=extra_args):
# 处理流式消息
delta_message = await self._make_msg_chunk(pipeline_config,chunk)
delta_message = await self._make_msg_chunk(pipeline_config,chunk,chunk_idx)
if delta_message.content:
current_content += delta_message.content
delta_message.content = current_content
print(current_content)
# delta_message.all_content = current_content
if delta_message.tool_calls:
for tool_call in delta_message.tool_calls:
@@ -352,7 +351,6 @@ class OpenAIChatCompletions(requester.ProviderAPIRequester):
extra_args=extra_args,
):
yield item
print(item)
except asyncio.TimeoutError:
raise errors.RequesterError('请求超时')