chore: initial public snapshot for github upload

llm-gateway-competitors/litellm-wheel-src/litellm/integrations/focus/serializers/__init__.py

@@ -0,0 +1,6 @@
+"""Serializer package exports for Focus integration."""
+
+from .base import FocusSerializer
+from .parquet import FocusParquetSerializer
+
+__all__ = ["FocusSerializer", "FocusParquetSerializer"]

llm-gateway-competitors/litellm-wheel-src/litellm/integrations/focus/serializers/base.py

@@ -0,0 +1,18 @@
+"""Serializer abstractions for Focus export."""
+
+from __future__ import annotations
+
+from abc import ABC, abstractmethod
+
+import polars as pl
+
+
+class FocusSerializer(ABC):
+    """Base serializer turning Focus frames into bytes."""
+
+    extension: str = ""
+
+    @abstractmethod
+    def serialize(self, frame: pl.DataFrame) -> bytes:
+        """Convert the normalized Focus frame into the chosen format."""
+        raise NotImplementedError

llm-gateway-competitors/litellm-wheel-src/litellm/integrations/focus/serializers/parquet.py

@@ -0,0 +1,22 @@
+"""Parquet serializer for Focus export."""
+
+from __future__ import annotations
+
+import io
+
+import polars as pl
+
+from .base import FocusSerializer
+
+
+class FocusParquetSerializer(FocusSerializer):
+    """Serialize normalized Focus frames to Parquet bytes."""
+
+    extension = "parquet"
+
+    def serialize(self, frame: pl.DataFrame) -> bytes:
+        """Encode the provided frame as a parquet payload."""
+        target = frame if not frame.is_empty() else pl.DataFrame(schema=frame.schema)
+        buffer = io.BytesIO()
+        target.write_parquet(buffer, compression="snappy")
+        return buffer.getvalue()