feat!: migrate api client to latest fern

Author: hassiebp

langfuse/_client/client.py

@@ -78,15 +78,15 @@
 from langfuse._utils import _get_timestamp
 from langfuse._utils.parse_error import handle_fern_exception
 from langfuse._utils.prompt_cache import PromptCache
-from langfuse.api.resources.commons.errors.error import Error
-from langfuse.api.resources.commons.errors.not_found_error import NotFoundError
-from langfuse.api.resources.ingestion.types.score_body import ScoreBody
-from langfuse.api.resources.prompts.types import (
+from langfuse.api.datasets import Dataset, DatasetItem, DatasetStatus
+from langfuse.api.ingestion import ScoreBody
+from langfuse.api.prompts import (
     CreatePromptRequest_Chat,
     CreatePromptRequest_Text,
     Prompt_Chat,
     Prompt_Text,
 )
+from langfuse.api.utils import Error, NotFoundError
 from langfuse.batch_evaluation import (
     BatchEvaluationResult,
     BatchEvaluationResumeToken,
@@ -112,12 +112,7 @@
     ChatMessageDict,
     ChatMessageWithPlaceholdersDict,
     ChatPromptClient,
-    CreateDatasetItemRequest,
-    CreateDatasetRequest,
     CreateDatasetRunItemRequest,
-    Dataset,
-    DatasetItem,
-    DatasetStatus,
     MapValue,
     PromptClient,
     TextPromptClient,
@@ -3271,16 +3266,15 @@ def create_dataset(
             Dataset: The created dataset as returned by the Langfuse API.
         """
         try:
-            body = CreateDatasetRequest(
+            langfuse_logger.debug(f"Creating datasets {name}")
+
+            return self.api.datasets.create(
                 name=name,
                 description=description,
                 metadata=metadata,
-                inputSchema=input_schema,
-                expectedOutputSchema=expected_output_schema,
+                input_schema=input_schema,
+                expected_output_schema=expected_output_schema,
             )
-            langfuse_logger.debug(f"Creating datasets {body}")
-
-            return self.api.datasets.create(request=body)
 
         except Error as e:
             handle_fern_exception(e)
@@ -3331,18 +3325,18 @@ def create_dataset_item(
             ```
         """
         try:
-            body = CreateDatasetItemRequest(
-                datasetName=dataset_name,
+            langfuse_logger.debug(f"Creating dataset item for dataset {dataset_name}")
+
+            return self.api.dataset_items.create(
+                dataset_name=dataset_name,
                 input=input,
-                expectedOutput=expected_output,
+                expected_output=expected_output,
                 metadata=metadata,
-                sourceTraceId=source_trace_id,
-                sourceObservationId=source_observation_id,
+                source_trace_id=source_trace_id,
+                source_observation_id=source_observation_id,
                 status=status,
                 id=id,
             )
-            langfuse_logger.debug(f"Creating dataset item {body}")
-            return self.api.dataset_items.create(request=body)
         except Error as e:
             handle_fern_exception(e)
             raise e
@@ -3711,7 +3705,7 @@ def create_prompt(
                         labels=labels,
                         tags=tags,
                         config=config or {},
-                        commitMessage=commit_message,
+                        commit_message=commit_message,
                         type="chat",
                     )
                 )
@@ -3731,7 +3725,7 @@ def create_prompt(
                 labels=labels,
                 tags=tags,
                 config=config or {},
-                commitMessage=commit_message,
+                commit_message=commit_message,
                 type="text",
             )
 

langfuse/_client/datasets.py

@@ -4,19 +4,19 @@
 
 from opentelemetry.util._decorator import _agnosticcontextmanager
 
+from langfuse.api.datasets import (
+    CreateDatasetRunItemRequest,
+    Dataset,
+    DatasetItem,
+    DatasetStatus,
+)
 from langfuse.batch_evaluation import CompositeEvaluatorFunction
 from langfuse.experiment import (
     EvaluatorFunction,
     ExperimentResult,
     RunEvaluatorFunction,
     TaskFunction,
 )
-from langfuse.model import (
-    CreateDatasetRunItemRequest,
-    Dataset,
-    DatasetItem,
-    DatasetStatus,
-)
 
 from .span import LangfuseSpan
 

langfuse/_client/resource_manager.py

@@ -42,7 +42,7 @@
 from langfuse._utils.environment import get_common_release_envs
 from langfuse._utils.prompt_cache import PromptCache
 from langfuse._utils.request import LangfuseClient
-from langfuse.api.client import AsyncFernLangfuse, FernLangfuse
+from langfuse.api.client import AsyncLangfuseAPI, LangfuseAPI
 from langfuse.logger import langfuse_logger
 from langfuse.types import MaskFunction
 
@@ -213,7 +213,7 @@ def _initialize_instance(
             client_headers = additional_headers if additional_headers else {}
             self.httpx_client = httpx.Client(timeout=timeout, headers=client_headers)
 
-        self.api = FernLangfuse(
+        self.api = LangfuseAPI(
             base_url=base_url,
             username=self.public_key,
             password=secret_key,
@@ -223,7 +223,7 @@ def _initialize_instance(
             httpx_client=self.httpx_client,
             timeout=timeout,
         )
-        self.async_api = AsyncFernLangfuse(
+        self.async_api = AsyncLangfuseAPI(
             base_url=base_url,
             username=self.public_key,
             password=secret_key,

langfuse/_task_manager/media_manager.py

@@ -10,10 +10,9 @@
 
 from langfuse._client.environment_variables import LANGFUSE_MEDIA_UPLOAD_ENABLED
 from langfuse._utils import _get_timestamp
-from langfuse.api import GetMediaUploadUrlRequest, PatchMediaBody
-from langfuse.api.client import FernLangfuse
+from langfuse.api.client import LangfuseAPI
 from langfuse.api.core import ApiError
-from langfuse.api.resources.media.types.media_content_type import MediaContentType
+from langfuse.api.media import MediaContentType
 from langfuse.media import LangfuseMedia
 
 from .media_upload_queue import UploadMediaJob
@@ -28,7 +27,7 @@ class MediaManager:
     def __init__(
         self,
         *,
-        api_client: FernLangfuse,
+        api_client: LangfuseAPI,
         media_upload_queue: Queue,
         max_retries: Optional[int] = 3,
     ):
@@ -219,14 +218,12 @@ def _process_upload_media_job(
     ) -> None:
         upload_url_response = self._request_with_backoff(
             self._api_client.media.get_upload_url,
-            request=GetMediaUploadUrlRequest(
-                contentLength=data["content_length"],
-                contentType=cast(MediaContentType, data["content_type"]),
-                sha256Hash=data["content_sha256_hash"],
-                field=data["field"],
-                traceId=data["trace_id"],
-                observationId=data["observation_id"],
-            ),
+            content_length=data["content_length"],
+            content_type=cast(MediaContentType, data["content_type"]),
+            sha256hash=data["content_sha256_hash"],
+            field=data["field"],
+            trace_id=data["trace_id"],
+            observation_id=data["observation_id"],
         )
 
         upload_url = upload_url_response.upload_url
@@ -266,12 +263,10 @@ def _process_upload_media_job(
         self._request_with_backoff(
             self._api_client.media.patch,
             media_id=data["media_id"],
-            request=PatchMediaBody(
-                uploadedAt=_get_timestamp(),
-                uploadHttpStatus=upload_response.status_code,
-                uploadHttpError=upload_response.text,
-                uploadTimeMs=upload_time_ms,
-            ),
+            uploaded_at=_get_timestamp(),
+            upload_http_status=upload_response.status_code,
+            upload_http_error=upload_response.text,
+            upload_time_ms=upload_time_ms,
         )
 
         self._log.debug(

langfuse/_utils/parse_error.py

@@ -4,16 +4,16 @@
 # our own api errors
 from langfuse._utils.request import APIError, APIErrors
 from langfuse.api.core import ApiError
+from langfuse.api.health import ServiceUnavailableError
 
 # fern api errors
-from langfuse.api.resources.commons.errors import (
+from langfuse.api.utils import (
     AccessDeniedError,
     Error,
     MethodNotAllowedError,
     NotFoundError,
     UnauthorizedError,
 )
-from langfuse.api.resources.health.errors import ServiceUnavailableError
 
 SUPPORT_URL = "https://langfuse.com/support"
 API_DOCS_URL = "https://api.reference.langfuse.com"

langfuse/api/.fern/metadata.json

@@ -0,0 +1,10 @@
+{
+  "cliVersion": "3.24.3",
+  "generatorName": "fernapi/fern-python-sdk",
+  "generatorVersion": "4.46.2",
+  "generatorConfig": {
+    "client": {
+      "class_name": "LangfuseAPI"
+    }
+  }
+}