ragflow / web_server /hook /api /client_authentication.py
KevinHuSh
build python version rag-flow (#21)
3079197
raw
history blame
1.33 kB
import requests
from web_server.db.service_registry import ServiceRegistry
from web_server.settings import RegistryServiceName
from web_server.hook import HookManager
from web_server.hook.common.parameters import ClientAuthenticationParameters, ClientAuthenticationReturn
from web_server.settings import HOOK_SERVER_NAME
@HookManager.register_client_authentication_hook
def authentication(parm: ClientAuthenticationParameters) -> ClientAuthenticationReturn:
service_list = ServiceRegistry.load_service(
server_name=HOOK_SERVER_NAME,
service_name=RegistryServiceName.CLIENT_AUTHENTICATION.value
)
if not service_list:
raise Exception(f"client authentication error: no found server"
f" {HOOK_SERVER_NAME} service client_authentication")
service = service_list[0]
response = getattr(requests, service.f_method.lower(), None)(
url=service.f_url,
json=parm.to_dict()
)
if response.status_code != 200:
raise Exception(
f"client authentication error: request authentication url failed, status code {response.status_code}")
elif response.json().get("code") != 0:
return ClientAuthenticationReturn(code=response.json().get("code"), message=response.json().get("msg"))
return ClientAuthenticationReturn()