back to typealias

Author: xzrderek

eval_protocol/proxy/proxy_core/models.py

@@ -3,26 +3,20 @@
 """
 
 from pydantic import BaseModel
-from typing import Optional, List, Any, Dict, Protocol
+from typing import Optional, List, Any, Dict, TypeAlias, Callable
 from fastapi import Request
 
 
+ChatRequestHook: TypeAlias = Callable[[Dict[str, Any], Request, "ChatParams"], tuple[Dict[str, Any], "ChatParams"]]
+TracesRequestHook: TypeAlias = Callable[[Request, "TracesParams"], "TracesParams"]
+
+
 class AccountInfo(BaseModel):
     """Account information returned from authentication."""
 
     account_id: str
 
 
-class ChatRequestHook(Protocol):
-    def __call__(
-        self, data: Dict[str, Any], request: Request, params: "ChatParams"
-    ) -> tuple[Dict[str, Any], "ChatParams"]: ...
-
-
-class TracesRequestHook(Protocol):
-    def __call__(self, request: Request, params: "TracesParams") -> "TracesParams": ...
-
-
 class ChatParams(BaseModel):
     """Typed container for chat completion URL path parameters."""