push

Author: hassiebp

langfuse/_client/datasets.py

@@ -5,7 +5,6 @@
 from opentelemetry.util._decorator import _agnosticcontextmanager
 
 from langfuse.api.datasets import (
-    CreateDatasetRunItemRequest,
     Dataset,
     DatasetItem,
     DatasetStatus,
@@ -131,13 +130,11 @@ def run(
             )
 
             self.langfuse.api.dataset_run_items.create(
-                request=CreateDatasetRunItemRequest(
-                    runName=run_name,
-                    datasetItemId=self.id,
-                    traceId=span.trace_id,
-                    metadata=run_metadata,
-                    runDescription=run_description,
-                )
+                run_name=run_name,
+                dataset_item_id=self.id,
+                trace_id=span.trace_id,
+                metadata=run_metadata,
+                run_description=run_description,
             )
 
             yield span

pyproject.toml

@@ -94,8 +94,5 @@ module = [
 ignore_missing_imports = true
 
 [[tool.mypy.overrides]]
-module = [
-    "langfuse.api.*",
-]
-ignore_errors = true
-
+module = "langfuse.api.*"
+disable_error_code = ["redundant-cast", "no-untyped-def"]