resolve #28

Author: Masterain98

main.py

@@ -93,13 +93,14 @@ def get_commit_hash_str():
 def identify_user(request: Request) -> None:
     # Extract headers
     device_id = request.headers.get("x-hutao-device-id", "unknown-device")
+    reqable_id = request.headers.get("Reqable-Id", None)
     user_agent = request.headers.get("User-Agent", "unknown-group")
 
     # Assign to Apitally consumer
     request.state.apitally_consumer = ApitallyConsumer(
-        identifier=device_id,
+        identifier=device_id if reqable_id is None else reqable_id,
         name=device_id,
-        group=user_agent
+        group=user_agent if reqable_id is None else "Reqable",
     )
 
 

utils/dgp_utils.py

@@ -84,6 +84,9 @@ async def validate_client_is_updated(request: Request, user_agent: Annotated[str
     if user_agent.startswith("PaimonsNotebook/"):
         logger.info("Client is Paimon's Notebook, allowed.")
         return True
+    if user_agent.startswith("Reqable/"):
+        logger.info("Client is Reqable, allowed.")
+        return True
 
     allowed_user_agents = await redis_client.get("allowed_user_agents")
     if allowed_user_agents: