fix: minor fix

This commit is contained in:
Junyan Qin
2025-08-04 23:00:54 +08:00
parent 15e524c6e6
commit 7f25d61531
2 changed files with 2 additions and 2 deletions

View File

@@ -55,7 +55,6 @@ class OpenAIChatCompletions(requester.ProviderAPIRequester):
) -> llm_entities.Message: ) -> llm_entities.Message:
chatcmpl_message = chat_completion.choices[0].message.model_dump() chatcmpl_message = chat_completion.choices[0].message.model_dump()
# print(chatcmpl_message.keys(),chatcmpl_message.values()) # print(chatcmpl_message.keys(),chatcmpl_message.values())
# 确保 role 字段存在且不为 None # 确保 role 字段存在且不为 None
if 'role' not in chatcmpl_message or chatcmpl_message['role'] is None: if 'role' not in chatcmpl_message or chatcmpl_message['role'] is None:
chatcmpl_message['role'] = 'assistant' chatcmpl_message['role'] = 'assistant'
@@ -241,6 +240,7 @@ class OpenAIChatCompletions(requester.ProviderAPIRequester):
messages: typing.List[llm_entities.Message], messages: typing.List[llm_entities.Message],
funcs: typing.List[tools_entities.LLMFunction] = None, funcs: typing.List[tools_entities.LLMFunction] = None,
extra_args: dict[str, typing.Any] = {}, extra_args: dict[str, typing.Any] = {},
remove_think: bool = False,
) -> llm_entities.Message: ) -> llm_entities.Message:
req_messages = [] # req_messages 仅用于类内,外部同步由 query.messages 进行 req_messages = [] # req_messages 仅用于类内,外部同步由 query.messages 进行
for m in messages: for m in messages:
@@ -260,6 +260,7 @@ class OpenAIChatCompletions(requester.ProviderAPIRequester):
use_model=model, use_model=model,
use_funcs=funcs, use_funcs=funcs,
extra_args=extra_args, extra_args=extra_args,
remove_think=remove_think,
) )
return msg return msg
except asyncio.TimeoutError: except asyncio.TimeoutError:

View File

@@ -89,7 +89,6 @@ class LocalAgentRunner(runner.RequestRunner):
req_messages = query.prompt.messages.copy() + query.messages.copy() + [user_message] req_messages = query.prompt.messages.copy() + query.messages.copy() + [user_message]
is_stream = query.adapter.is_stream_output_supported()
try: try:
is_stream = await query.adapter.is_stream_output_supported() is_stream = await query.adapter.is_stream_output_supported()
except AttributeError: except AttributeError: