add back enums

Author: hassiebp

langfuse/_client/client.py

@@ -80,6 +80,7 @@
 from langfuse._utils.prompt_cache import PromptCache
 from langfuse.api import (
     CreateChatPromptRequest,
+    CreateChatPromptType,
     CreateTextPromptRequest,
     Dataset,
     DatasetItem,
@@ -3710,7 +3711,7 @@ def create_prompt(
                         tags=tags,
                         config=config or {},
                         commit_message=commit_message,
-                        type="chat",
+                        type=CreateChatPromptType.CHAT,
                     )
                 )
                 server_prompt = self.api.prompts.create(request=request)

langfuse/api/.fern/metadata.json

@@ -3,6 +3,9 @@
   "generatorName": "fernapi/fern-python-sdk",
   "generatorVersion": "4.46.2",
   "generatorConfig": {
+    "pydantic_config": {
+      "enum_type": "python_enums"
+    },
     "client": {
       "class_name": "LangfuseAPI"
     }

langfuse/api/annotation_queues/client.py

@@ -313,6 +313,7 @@ def create_queue_item(
         Examples
         --------
         from langfuse import LangfuseAPI
+        from langfuse.annotation_queues import AnnotationQueueObjectType
 
         client = LangfuseAPI(
             x_langfuse_sdk_name="YOUR_X_LANGFUSE_SDK_NAME",
@@ -325,7 +326,7 @@ def create_queue_item(
         client.annotation_queues.create_queue_item(
             queue_id="queueId",
             object_id="objectId",
-            object_type="TRACE",
+            object_type=AnnotationQueueObjectType.TRACE,
         )
         """
         _response = self._raw_client.create_queue_item(
@@ -857,6 +858,7 @@ async def create_queue_item(
         import asyncio
 
         from langfuse import AsyncLangfuseAPI
+        from langfuse.annotation_queues import AnnotationQueueObjectType
 
         client = AsyncLangfuseAPI(
             x_langfuse_sdk_name="YOUR_X_LANGFUSE_SDK_NAME",
@@ -872,7 +874,7 @@ async def main() -> None:
             await client.annotation_queues.create_queue_item(
                 queue_id="queueId",
                 object_id="objectId",
-                object_type="TRACE",
+                object_type=AnnotationQueueObjectType.TRACE,
             )
 
 

langfuse/api/annotation_queues/types/annotation_queue_object_type.py

@@ -2,6 +2,25 @@
 
 import typing
 
-AnnotationQueueObjectType = typing.Union[
-    typing.Literal["TRACE", "OBSERVATION", "SESSION"], typing.Any
-]
+from ...core import enum
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class AnnotationQueueObjectType(enum.StrEnum):
+    TRACE = "TRACE"
+    OBSERVATION = "OBSERVATION"
+    SESSION = "SESSION"
+
+    def visit(
+        self,
+        trace: typing.Callable[[], T_Result],
+        observation: typing.Callable[[], T_Result],
+        session: typing.Callable[[], T_Result],
+    ) -> T_Result:
+        if self is AnnotationQueueObjectType.TRACE:
+            return trace()
+        if self is AnnotationQueueObjectType.OBSERVATION:
+            return observation()
+        if self is AnnotationQueueObjectType.SESSION:
+            return session()

langfuse/api/annotation_queues/types/annotation_queue_status.py

@@ -2,4 +2,21 @@
 
 import typing
 
-AnnotationQueueStatus = typing.Union[typing.Literal["PENDING", "COMPLETED"], typing.Any]
+from ...core import enum
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class AnnotationQueueStatus(enum.StrEnum):
+    PENDING = "PENDING"
+    COMPLETED = "COMPLETED"
+
+    def visit(
+        self,
+        pending: typing.Callable[[], T_Result],
+        completed: typing.Callable[[], T_Result],
+    ) -> T_Result:
+        if self is AnnotationQueueStatus.PENDING:
+            return pending()
+        if self is AnnotationQueueStatus.COMPLETED:
+            return completed()

langfuse/api/blob_storage_integrations/client.py

@@ -146,6 +146,12 @@ def upsert_blob_storage_integration(
         Examples
         --------
         from langfuse import LangfuseAPI
+        from langfuse.blob_storage_integrations import (
+            BlobStorageExportFrequency,
+            BlobStorageExportMode,
+            BlobStorageIntegrationFileType,
+            BlobStorageIntegrationType,
+        )
 
         client = LangfuseAPI(
             x_langfuse_sdk_name="YOUR_X_LANGFUSE_SDK_NAME",
@@ -157,14 +163,14 @@ def upsert_blob_storage_integration(
         )
         client.blob_storage_integrations.upsert_blob_storage_integration(
             project_id="projectId",
-            type="S3",
+            type=BlobStorageIntegrationType.S3,
             bucket_name="bucketName",
             region="region",
-            export_frequency="hourly",
+            export_frequency=BlobStorageExportFrequency.HOURLY,
             enabled=True,
             force_path_style=True,
-            file_type="JSON",
-            export_mode="FULL_HISTORY",
+            file_type=BlobStorageIntegrationFileType.JSON,
+            export_mode=BlobStorageExportMode.FULL_HISTORY,
         )
         """
         _response = self._raw_client.upsert_blob_storage_integration(
@@ -358,6 +364,12 @@ async def upsert_blob_storage_integration(
         import asyncio
 
         from langfuse import AsyncLangfuseAPI
+        from langfuse.blob_storage_integrations import (
+            BlobStorageExportFrequency,
+            BlobStorageExportMode,
+            BlobStorageIntegrationFileType,
+            BlobStorageIntegrationType,
+        )
 
         client = AsyncLangfuseAPI(
             x_langfuse_sdk_name="YOUR_X_LANGFUSE_SDK_NAME",
@@ -372,14 +384,14 @@ async def upsert_blob_storage_integration(
         async def main() -> None:
             await client.blob_storage_integrations.upsert_blob_storage_integration(
                 project_id="projectId",
-                type="S3",
+                type=BlobStorageIntegrationType.S3,
                 bucket_name="bucketName",
                 region="region",
-                export_frequency="hourly",
+                export_frequency=BlobStorageExportFrequency.HOURLY,
                 enabled=True,
                 force_path_style=True,
-                file_type="JSON",
-                export_mode="FULL_HISTORY",
+                file_type=BlobStorageIntegrationFileType.JSON,
+                export_mode=BlobStorageExportMode.FULL_HISTORY,
             )
 
 

langfuse/api/blob_storage_integrations/types/blob_storage_export_frequency.py

@@ -2,6 +2,25 @@
 
 import typing
 
-BlobStorageExportFrequency = typing.Union[
-    typing.Literal["hourly", "daily", "weekly"], typing.Any
-]
+from ...core import enum
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class BlobStorageExportFrequency(enum.StrEnum):
+    HOURLY = "hourly"
+    DAILY = "daily"
+    WEEKLY = "weekly"
+
+    def visit(
+        self,
+        hourly: typing.Callable[[], T_Result],
+        daily: typing.Callable[[], T_Result],
+        weekly: typing.Callable[[], T_Result],
+    ) -> T_Result:
+        if self is BlobStorageExportFrequency.HOURLY:
+            return hourly()
+        if self is BlobStorageExportFrequency.DAILY:
+            return daily()
+        if self is BlobStorageExportFrequency.WEEKLY:
+            return weekly()

langfuse/api/blob_storage_integrations/types/blob_storage_export_mode.py

@@ -2,6 +2,25 @@
 
 import typing
 
-BlobStorageExportMode = typing.Union[
-    typing.Literal["FULL_HISTORY", "FROM_TODAY", "FROM_CUSTOM_DATE"], typing.Any
-]
+from ...core import enum
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class BlobStorageExportMode(enum.StrEnum):
+    FULL_HISTORY = "FULL_HISTORY"
+    FROM_TODAY = "FROM_TODAY"
+    FROM_CUSTOM_DATE = "FROM_CUSTOM_DATE"
+
+    def visit(
+        self,
+        full_history: typing.Callable[[], T_Result],
+        from_today: typing.Callable[[], T_Result],
+        from_custom_date: typing.Callable[[], T_Result],
+    ) -> T_Result:
+        if self is BlobStorageExportMode.FULL_HISTORY:
+            return full_history()
+        if self is BlobStorageExportMode.FROM_TODAY:
+            return from_today()
+        if self is BlobStorageExportMode.FROM_CUSTOM_DATE:
+            return from_custom_date()

langfuse/api/blob_storage_integrations/types/blob_storage_integration_file_type.py

@@ -2,6 +2,25 @@
 
 import typing
 
-BlobStorageIntegrationFileType = typing.Union[
-    typing.Literal["JSON", "CSV", "JSONL"], typing.Any
-]
+from ...core import enum
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class BlobStorageIntegrationFileType(enum.StrEnum):
+    JSON = "JSON"
+    CSV = "CSV"
+    JSONL = "JSONL"
+
+    def visit(
+        self,
+        json: typing.Callable[[], T_Result],
+        csv: typing.Callable[[], T_Result],
+        jsonl: typing.Callable[[], T_Result],
+    ) -> T_Result:
+        if self is BlobStorageIntegrationFileType.JSON:
+            return json()
+        if self is BlobStorageIntegrationFileType.CSV:
+            return csv()
+        if self is BlobStorageIntegrationFileType.JSONL:
+            return jsonl()

langfuse/api/blob_storage_integrations/types/blob_storage_integration_type.py

@@ -2,6 +2,25 @@
 
 import typing
 
-BlobStorageIntegrationType = typing.Union[
-    typing.Literal["S3", "S3_COMPATIBLE", "AZURE_BLOB_STORAGE"], typing.Any
-]
+from ...core import enum
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class BlobStorageIntegrationType(enum.StrEnum):
+    S3 = "S3"
+    S3COMPATIBLE = "S3_COMPATIBLE"
+    AZURE_BLOB_STORAGE = "AZURE_BLOB_STORAGE"
+
+    def visit(
+        self,
+        s3: typing.Callable[[], T_Result],
+        s3compatible: typing.Callable[[], T_Result],
+        azure_blob_storage: typing.Callable[[], T_Result],
+    ) -> T_Result:
+        if self is BlobStorageIntegrationType.S3:
+            return s3()
+        if self is BlobStorageIntegrationType.S3COMPATIBLE:
+            return s3compatible()
+        if self is BlobStorageIntegrationType.AZURE_BLOB_STORAGE:
+            return azure_blob_storage()