chore: format

This commit is contained in:
Timothy Jaeryang Baek
2026-02-12 16:13:48 -06:00
parent 3238d94a0e
commit 2a11175f22
65 changed files with 204 additions and 92 deletions
-1
View File
@@ -355,4 +355,3 @@ async def chat_completed(request: Request, form_data: dict, user: Any):
return result
except Exception as e:
raise Exception(f"Error: {e}")
+2 -4
View File
@@ -4125,8 +4125,7 @@ async def streaming_chat_response_handler(response, ctx):
code = sanitize_code(code)
if CODE_INTERPRETER_BLOCKED_MODULES:
blocking_code = textwrap.dedent(
f"""
blocking_code = textwrap.dedent(f"""
import builtins
BLOCKED_MODULES = {CODE_INTERPRETER_BLOCKED_MODULES}
@@ -4142,8 +4141,7 @@ async def streaming_chat_response_handler(response, ctx):
return _real_import(name, globals, locals, fromlist, level)
builtins.__import__ = restricted_import
"""
)
""")
code = blocking_code + "\n" + code
if (
+3 -2
View File
@@ -782,7 +782,6 @@ def extract_urls(text: str) -> list[str]:
return url_pattern.findall(text)
async def cleanup_response(
response: Optional[aiohttp.ClientResponse],
session: Optional[aiohttp.ClientSession],
@@ -799,7 +798,9 @@ async def stream_wrapper(response, session, content_handler=None):
This is more reliable than BackgroundTask which may not run if client disconnects.
"""
try:
stream = content_handler(response.content) if content_handler else response.content
stream = (
content_handler(response.content) if content_handler else response.content
)
async for chunk in stream:
yield chunk
finally:
+4 -4
View File
@@ -353,12 +353,12 @@ async def get_tools(
headers = include_user_info_headers(headers, user)
metadata = extra_params.get("__metadata__", {})
if metadata and metadata.get("chat_id"):
headers[FORWARD_SESSION_INFO_HEADER_CHAT_ID] = metadata.get(
"chat_id"
headers[FORWARD_SESSION_INFO_HEADER_CHAT_ID] = (
metadata.get("chat_id")
)
if metadata and metadata.get("message_id"):
headers[FORWARD_SESSION_INFO_HEADER_MESSAGE_ID] = metadata.get(
"message_id"
headers[FORWARD_SESSION_INFO_HEADER_MESSAGE_ID] = (
metadata.get("message_id")
)
def make_tool_function(