refac
This commit is contained in:
@@ -2607,12 +2607,17 @@ async def process_chat_payload(request, form_data, user, metadata, model):
|
|||||||
# so system terminals work even when no other tools are selected)
|
# so system terminals work even when no other tools are selected)
|
||||||
if terminal_id:
|
if terminal_id:
|
||||||
try:
|
try:
|
||||||
terminal_tools, system_prompt = await get_terminal_tools(
|
terminal_result = await get_terminal_tools(
|
||||||
request,
|
request,
|
||||||
terminal_id,
|
terminal_id,
|
||||||
user,
|
user,
|
||||||
extra_params,
|
extra_params,
|
||||||
)
|
)
|
||||||
|
if isinstance(terminal_result, tuple):
|
||||||
|
terminal_tools, system_prompt = terminal_result
|
||||||
|
else:
|
||||||
|
terminal_tools = terminal_result
|
||||||
|
system_prompt = None
|
||||||
if terminal_tools:
|
if terminal_tools:
|
||||||
tools_dict = {**tools_dict, **terminal_tools}
|
tools_dict = {**tools_dict, **terminal_tools}
|
||||||
if system_prompt:
|
if system_prompt:
|
||||||
|
|||||||
@@ -956,7 +956,7 @@ async def get_terminal_tools(
|
|||||||
terminal_id: str,
|
terminal_id: str,
|
||||||
user: UserModel,
|
user: UserModel,
|
||||||
extra_params: dict,
|
extra_params: dict,
|
||||||
) -> tuple[dict[str, dict], Optional[str]]:
|
) -> dict[str, dict] | tuple[dict[str, dict], Optional[str]]:
|
||||||
"""Resolve tools for a terminal server identified by terminal_id.
|
"""Resolve tools for a terminal server identified by terminal_id.
|
||||||
|
|
||||||
- Finds the connection in TERMINAL_SERVER_CONNECTIONS
|
- Finds the connection in TERMINAL_SERVER_CONNECTIONS
|
||||||
|
|||||||
Reference in New Issue
Block a user