知微
fa45d8aa5f
- health_checklist.json: 192.168.1.122→node122
- ocr_client.py: docstring IP→node122
- docs/market-data-requirements.md: IP→node122
- 所有API调用通过ProxyHandler({})绕过系统代理
Privoxy对node122:18003返回500，直连正常
42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
from fastapi import Request
|
|
from fastapi_sso.sso.base import OpenID
|
|
|
|
from litellm.integrations.custom_logger import CustomLogger
|
|
|
|
|
|
class CustomSSOLoginHandler(CustomLogger):
|
|
"""
|
|
Custom logger for the UI SSO sign in
|
|
|
|
Use this to parse the request headers and return a OpenID object
|
|
|
|
Useful when you have an OAuth proxy in front of LiteLLM
|
|
and you want to use the headers from the proxy to sign in the user
|
|
"""
|
|
|
|
async def handle_custom_ui_sso_sign_in(
|
|
self,
|
|
request: Request,
|
|
) -> OpenID:
|
|
from litellm.proxy.auth.trusted_proxy_utils import (
|
|
require_trusted_proxy_request,
|
|
)
|
|
from litellm.proxy.proxy_server import general_settings
|
|
|
|
require_trusted_proxy_request(
|
|
request=request,
|
|
general_settings=general_settings,
|
|
feature_name="Custom UI SSO",
|
|
)
|
|
|
|
request_headers_dict = dict(request.headers)
|
|
return OpenID(
|
|
id=request_headers_dict.get("x-litellm-user-id"),
|
|
email=request_headers_dict.get("x-litellm-user-email"),
|
|
first_name="Test",
|
|
last_name="Test",
|
|
display_name="Test",
|
|
picture="https://test.com/test.png",
|
|
provider="test",
|
|
)
|