ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(router): register _aresponses_websocket in Router factory functions

Browse Source 
Also suppress pre-existing pyright type error on model_response.close() call
which is guarded by hasattr at runtime.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
Sameer Kankute 2026-03-02 17:24:12 +05:30
parent 512a438935
commit 0921e26b8c
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
litellm/router.py
View File

@ -882,6 +882,9 @@ class Router:
self._arealtime = self.factory_function(
litellm._arealtime, call_type="_arealtime"
)
self._aresponses_websocket = self.factory_function(
litellm._aresponses_websocket, call_type="_aresponses_websocket"
)
self.acreate_fine_tuning_job = self.factory_function(
litellm.acreate_fine_tuning_job, call_type="acreate_fine_tuning_job"
)
@ -1824,7 +1827,7 @@ class Router:
finally:
if hasattr(model_response, "close"):
try:
model_response.close()
model_response.close() # type: ignore[reportAttributeAccessIssue]
except BaseException as close_err:
verbose_router_logger.debug(
"stream_with_fallbacks: error closing model_response: %s",
@ -4659,6 +4662,7 @@ class Router:
"afile_delete",
"afile_content",
"_arealtime",
"_aresponses_websocket",
"acreate_fine_tuning_job",
"acancel_fine_tuning_job",
"alist_fine_tuning_jobs",
@ -4831,6 +4835,7 @@ class Router:
"anthropic_messages",
"aresponses",
"_arealtime",
"_aresponses_websocket",
"acreate_fine_tuning_job",
"acancel_fine_tuning_job",
"alist_fine_tuning_jobs",