mirror of
https://github.com/langbot-app/LangBot.git
synced 2025-11-25 19:37:36 +08:00
fix: minor fix
This commit is contained in:
@@ -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:
|
||||||
|
|||||||
@@ -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:
|
||||||
|
|||||||
Reference in New Issue
Block a user