build python version rag-flow (#21)

* clean rust version project * clean rust version project * build python version rag-flow
2024-01-15 08:46:22 +08:00
parent db8cae3f1e
commit 30791976d5
123 changed files with 4985 additions and 4239 deletions
--- a/web_server/hook/api/site_authentication.py
+++ b/web_server/hook/api/site_authentication.py
@@ -0,0 +1,49 @@
+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 SignatureParameters, AuthenticationParameters, AuthenticationReturn,\
+    SignatureReturn
+from web_server.settings import HOOK_SERVER_NAME, PARTY_ID
+
+
+@HookManager.register_site_signature_hook
+def signature(parm: SignatureParameters) -> SignatureReturn:
+    service_list = ServiceRegistry.load_service(server_name=HOOK_SERVER_NAME, service_name=RegistryServiceName.SIGNATURE.value)
+    if not service_list:
+        raise Exception(f"signature error: no found server {HOOK_SERVER_NAME} service signature")
+    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:
+        if response.json().get("code") == 0:
+            return SignatureReturn(site_signature=response.json().get("data"))
+        else:
+            raise Exception(f"signature error: request signature url failed, result: {response.json()}")
+    else:
+        raise Exception(f"signature error: request signature url failed, status code {response.status_code}")
+
+
+@HookManager.register_site_authentication_hook
+def authentication(parm: AuthenticationParameters) -> AuthenticationReturn:
+    if not parm.src_party_id or str(parm.src_party_id) == "0":
+        parm.src_party_id = PARTY_ID
+    service_list = ServiceRegistry.load_service(server_name=HOOK_SERVER_NAME,
+                                                service_name=RegistryServiceName.SITE_AUTHENTICATION.value)
+    if not service_list:
+        raise Exception(
+            f"site authentication error: no found server {HOOK_SERVER_NAME} service site_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"site authentication error: request site_authentication url failed, status code {response.status_code}")
+    elif response.json().get("code") != 0:
+        return AuthenticationReturn(code=response.json().get("code"), message=response.json().get("msg"))
+    return AuthenticationReturn()