fastlib-py 0.3.1__tar.gz → 0.3.2__tar.gz

{fastlib_py-0.3.1 → fastlib_py-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fastlib-py
-Version: 0.3.1
+Version: 0.3.2
 Summary: Fastlib, high performance, easy to use, fast to code, ready for production
 Author-email: tyvekZhang <tyvekzhang@gmail.com>
 Maintainer-email: tyvekZhang <tyvekzhang@gmail.com>

{fastlib_py-0.3.1 → fastlib_py-0.3.2}/fastlib/config/_server_config.py

@@ -19,6 +19,7 @@ class ServerConfig(BaseConfig):
         name: The server name.
         version: The server version.
         app_desc: The server description.
+        enable_api_prefix: Enable the API prefix for routes.
         api_prefix: The API prefix for routes.
         debug: Whether to enable debug mode.
         workers: The number of worker processes.
@@ -35,6 +36,7 @@ class ServerConfig(BaseConfig):
     app_desc: str = field(
         default_factory=lambda: ServerConfig._project_config.description
     )
+    enable_api_prefix: bool = False
     api_prefix: str = "/api"
     debug: bool = False
     workers: int = 1

{fastlib_py-0.3.1 → fastlib_py-0.3.2}/fastlib/middleware/i18n.py

@@ -1,40 +1,40 @@
-# SPDX-License-Identifier: MIT
-from starlette.middleware.base import BaseHTTPMiddleware
-
-from fastlib.i18n.manager import reset_language, set_language
-from fastlib.i18n.types import Language
-
-
-class I18nMiddleware(BaseHTTPMiddleware):
-    """
-    Middleware for automatic language detection and setting based on HTTP headers.
-    """
-
-    async def dispatch(self, request, call_next):
-        """
-        Process the request and set the language based on Accept-Language header.
-
-        Args:
-            request: The incoming HTTP request
-            call_next: The next middleware or route handler in the chain
-
-        Returns:
-            The HTTP response from the next handler
-        """
-        # Get language preference from request headers
-        accept_lang = request.headers.get("Accept-Language", "en")
-
-        # Set the language for current request
-        if accept_lang.startswith("en"):
-            language = Language.CHINESE
-        elif accept_lang.startswith("zh"):
-            language = Language.CHINESE
-
-        set_language(language)
-        try:
-            response = await call_next(request)
-        finally:
-            # Always reset to prevent leaking context between requests
-            reset_language()
-
-        return response
+# SPDX-License-Identifier: MIT
+from starlette.middleware.base import BaseHTTPMiddleware
+
+from fastlib.i18n.manager import reset_language, set_language
+from fastlib.i18n.types import Language
+
+
+class I18nMiddleware(BaseHTTPMiddleware):
+    """
+    Middleware for automatic language detection and setting based on HTTP headers.
+    """
+
+    async def dispatch(self, request, call_next):
+        """
+        Process the request and set the language based on Accept-Language header.
+
+        Args:
+            request: The incoming HTTP request
+            call_next: The next middleware or route handler in the chain
+
+        Returns:
+            The HTTP response from the next handler
+        """
+        # Get language preference from request headers
+        accept_lang = request.headers.get("Accept-Language", "en")
+
+        # Set the language for current request
+        if accept_lang.startswith("en"):
+            language = Language.CHINESE
+        elif accept_lang.startswith("zh"):
+            language = Language.CHINESE
+
+        set_language(language)
+        try:
+            response = await call_next(request)
+        finally:
+            # Always reset to prevent leaking context between requests
+            reset_language()
+
+        return response

{fastlib_py-0.3.1 → fastlib_py-0.3.2}/fastlib/stream/enums.py

@@ -1,36 +1,36 @@
-# SPDX-License-Identifier: MIT
-"""
-Enums for stream module
-"""
-
-from enum import Enum
-
-
-class MessageRole(str, Enum):
-    """Message role enumeration"""
-
-    SYSTEM = "system"
-    USER = "user"
-    ASSISTANT = "assistant"
-    FUNCTION = "function"
-
-
-class MessageStatus(str, Enum):
-    """Message status enumeration"""
-
-    CREATED = "created"
-    RUNNING = "running"
-    COMPLETED = "completed"
-    FAILED = "failed"
-    CANCELLED = "cancelled"
-
-
-class StreamEventType(str, Enum):
-    """Stream event type"""
-
-    START = "start"
-    CONTENT = "content"
-    FUNCTION_CALL = "function_call"
-    ERROR = "error"
-    END = "end"
-    METADATA = "metadata"
+# SPDX-License-Identifier: MIT
+"""
+Enums for stream module
+"""
+
+from enum import Enum
+
+
+class MessageRole(str, Enum):
+    """Message role enumeration"""
+
+    SYSTEM = "system"
+    USER = "user"
+    ASSISTANT = "assistant"
+    FUNCTION = "function"
+
+
+class MessageStatus(str, Enum):
+    """Message status enumeration"""
+
+    CREATED = "created"
+    RUNNING = "running"
+    COMPLETED = "completed"
+    FAILED = "failed"
+    CANCELLED = "cancelled"
+
+
+class StreamEventType(str, Enum):
+    """Stream event type"""
+
+    START = "start"
+    CONTENT = "content"
+    FUNCTION_CALL = "function_call"
+    ERROR = "error"
+    END = "end"
+    METADATA = "metadata"

{fastlib_py-0.3.1 → fastlib_py-0.3.2}/fastlib/stream/event.py

@@ -1,101 +1,101 @@
-# SPDX-License-Identifier: MIT
-"""
-Copy from https://github.com/sysid/sse-starlette/blob/main/sse_starlette/event.py
-"""
-
-import io
-import json
-import re
-from typing import Any, Optional, Union
-
-
-class ServerSentEvent:
-    """
-    Helper class to format data for Server-Sent Events (SSE).
-    """
-
-    _LINE_SEP_EXPR = re.compile(r"\r\n|\r|\n")
-    DEFAULT_SEPARATOR = "\r\n"
-
-    def __init__(
-        self,
-        data: Optional[Any] = None,
-        *,
-        event: Optional[str] = None,
-        id: Optional[str] = None,
-        retry: Optional[int] = None,
-        comment: Optional[str] = None,
-        sep: Optional[str] = None,
-    ) -> None:
-        self.data = data
-        self.event = event
-        self.id = id
-        self.retry = retry
-        self.comment = comment
-        self._sep = sep if sep is not None else self.DEFAULT_SEPARATOR
-
-    def encode(self) -> bytes:
-        buffer = io.StringIO()
-        if self.comment is not None:
-            for chunk in self._LINE_SEP_EXPR.split(str(self.comment)):
-                buffer.write(f": {chunk}{self._sep}")
-
-        if self.id is not None:
-            # Clean newlines in the event id
-            buffer.write("id: " + self._LINE_SEP_EXPR.sub("", self.id) + self._sep)
-
-        if self.event is not None:
-            # Clean newlines in the event name
-            buffer.write(
-                "event: " + self._LINE_SEP_EXPR.sub("", self.event) + self._sep
-            )
-
-        if self.data is not None:
-            # Break multi-line data into multiple data: lines
-            for chunk in self._LINE_SEP_EXPR.split(str(self.data)):
-                buffer.write(f"data: {chunk}{self._sep}")
-
-        if self.retry is not None:
-            if not isinstance(self.retry, int):
-                raise TypeError("retry argument must be int")
-            buffer.write(f"retry: {self.retry}{self._sep}")
-
-        buffer.write(self._sep)
-        return buffer.getvalue().encode("utf-8")
-
-
-class JSONServerSentEvent(ServerSentEvent):
-    """
-    Helper class to format JSON data for Server-Sent Events (SSE).
-    """
-
-    def __init__(
-        self,
-        data: Optional[Any] = None,
-        *args,
-        **kwargs,
-    ) -> None:
-        super().__init__(
-            json.dumps(
-                data,
-                ensure_ascii=False,
-                allow_nan=False,
-                indent=None,
-                separators=(",", ":"),
-            )
-            if data is not None
-            else None,
-            *args,
-            **kwargs,
-        )
-
-
-def ensure_bytes(data: Union[bytes, dict, ServerSentEvent, Any], sep: str) -> bytes:
-    if isinstance(data, bytes):
-        return data
-    if isinstance(data, ServerSentEvent):
-        return data.encode()
-    if isinstance(data, dict):
-        data["sep"] = sep
-        return ServerSentEvent(**data).encode()
-    return ServerSentEvent(str(data), sep=sep).encode()
+# SPDX-License-Identifier: MIT
+"""
+Copy from https://github.com/sysid/sse-starlette/blob/main/sse_starlette/event.py
+"""
+
+import io
+import json
+import re
+from typing import Any, Optional, Union
+
+
+class ServerSentEvent:
+    """
+    Helper class to format data for Server-Sent Events (SSE).
+    """
+
+    _LINE_SEP_EXPR = re.compile(r"\r\n|\r|\n")
+    DEFAULT_SEPARATOR = "\r\n"
+
+    def __init__(
+        self,
+        data: Optional[Any] = None,
+        *,
+        event: Optional[str] = None,
+        id: Optional[str] = None,
+        retry: Optional[int] = None,
+        comment: Optional[str] = None,
+        sep: Optional[str] = None,
+    ) -> None:
+        self.data = data
+        self.event = event
+        self.id = id
+        self.retry = retry
+        self.comment = comment
+        self._sep = sep if sep is not None else self.DEFAULT_SEPARATOR
+
+    def encode(self) -> bytes:
+        buffer = io.StringIO()
+        if self.comment is not None:
+            for chunk in self._LINE_SEP_EXPR.split(str(self.comment)):
+                buffer.write(f": {chunk}{self._sep}")
+
+        if self.id is not None:
+            # Clean newlines in the event id
+            buffer.write("id: " + self._LINE_SEP_EXPR.sub("", self.id) + self._sep)
+
+        if self.event is not None:
+            # Clean newlines in the event name
+            buffer.write(
+                "event: " + self._LINE_SEP_EXPR.sub("", self.event) + self._sep
+            )
+
+        if self.data is not None:
+            # Break multi-line data into multiple data: lines
+            for chunk in self._LINE_SEP_EXPR.split(str(self.data)):
+                buffer.write(f"data: {chunk}{self._sep}")
+
+        if self.retry is not None:
+            if not isinstance(self.retry, int):
+                raise TypeError("retry argument must be int")
+            buffer.write(f"retry: {self.retry}{self._sep}")
+
+        buffer.write(self._sep)
+        return buffer.getvalue().encode("utf-8")
+
+
+class JSONServerSentEvent(ServerSentEvent):
+    """
+    Helper class to format JSON data for Server-Sent Events (SSE).
+    """
+
+    def __init__(
+        self,
+        data: Optional[Any] = None,
+        *args,
+        **kwargs,
+    ) -> None:
+        super().__init__(
+            json.dumps(
+                data,
+                ensure_ascii=False,
+                allow_nan=False,
+                indent=None,
+                separators=(",", ":"),
+            )
+            if data is not None
+            else None,
+            *args,
+            **kwargs,
+        )
+
+
+def ensure_bytes(data: Union[bytes, dict, ServerSentEvent, Any], sep: str) -> bytes:
+    if isinstance(data, bytes):
+        return data
+    if isinstance(data, ServerSentEvent):
+        return data.encode()
+    if isinstance(data, dict):
+        data["sep"] = sep
+        return ServerSentEvent(**data).encode()
+    return ServerSentEvent(str(data), sep=sep).encode()