feat(api): update api client

Author: hassiebp

langfuse/_client/client.py

@@ -2941,7 +2941,7 @@ def create_dataset(
             )
             langfuse_logger.debug(f"Creating datasets {body}")
 
-            return self.api.datasets.create(request=body)
+            return cast(Dataset, self.api.datasets.create(request=body))
 
         except Error as e:
             handle_fern_exception(e)
@@ -2993,17 +2993,18 @@ def create_dataset_item(
         """
         try:
             body = CreateDatasetItemRequest(
-                datasetName=dataset_name,
+                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)
+
+            return cast(DatasetItem, self.api.dataset_items.create(request=body))
         except Error as e:
             handle_fern_exception(e)
             raise e
@@ -3364,7 +3365,7 @@ def create_prompt(
                         labels=labels,
                         tags=tags,
                         config=config or {},
-                        commitMessage=commit_message,
+                        commit_message=commit_message,
                         type="chat",
                     )
                 )
@@ -3384,7 +3385,7 @@ def create_prompt(
                 labels=labels,
                 tags=tags,
                 config=config or {},
-                commitMessage=commit_message,
+                commit_message=commit_message,
                 type="text",
             )
 

langfuse/_client/datasets.py

@@ -131,11 +131,11 @@ def run(
 
             self.langfuse.api.dataset_run_items.create(
                 request=CreateDatasetRunItemRequest(
-                    runName=run_name,
-                    datasetItemId=self.id,
-                    traceId=span.trace_id,
+                    run_name=run_name,
+                    dataset_item_id=self.id,
+                    trace_id=span.trace_id,
                     metadata=run_metadata,
-                    runDescription=run_description,
+                    run_description=run_description,
                 )
             )
 

langfuse/_task_manager/media_manager.py

@@ -220,12 +220,12 @@ def _process_upload_media_job(
         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"],
+                content_length=data["content_length"],
+                content_type=cast(MediaContentType, data["content_type"]),
+                sha_256_hash=data["content_sha256_hash"],
                 field=data["field"],
-                traceId=data["trace_id"],
-                observationId=data["observation_id"],
+                trace_id=data["trace_id"],
+                observation_id=data["observation_id"],
             ),
         )
 
@@ -267,10 +267,10 @@ def _process_upload_media_job(
             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,
             ),
         )