Get a proxy hook from the proxy_hook_mapping
(self, hook: str)
| 555 | self.proxy_hook_mapping[hook] = proxy_hook_obj |
| 556 | |
| 557 | def get_proxy_hook(self, hook: str) -> Optional[CustomLogger]: |
| 558 | """ |
| 559 | Get a proxy hook from the proxy_hook_mapping |
| 560 | """ |
| 561 | return self.proxy_hook_mapping.get(hook) |
| 562 | |
| 563 | def _init_litellm_callbacks(self, llm_router: Optional[Router] = None): |
| 564 | self._add_proxy_hooks(llm_router) |