add observationid to link calls

Author: hassiebp

langfuse/_client/client.py

@@ -101,6 +101,7 @@
     ChatPromptClient,
     CreateDatasetItemRequest,
     CreateDatasetRequest,
+    CreateDatasetRunItemRequest,
     Dataset,
     DatasetItem,
     DatasetStatus,
@@ -2806,15 +2807,14 @@ async def _process_experiment_item(
                 # Link to dataset run if this is a dataset item
                 if hasattr(item, "id") and hasattr(item, "dataset_id"):
                     try:
-                        from langfuse.model import CreateDatasetRunItemRequest
-
                         dataset_run_item = self.api.dataset_run_items.create(
                             request=CreateDatasetRunItemRequest(
                                 runName=experiment_run_name,
                                 runDescription=experiment_description,
                                 metadata=experiment_metadata,
                                 datasetItemId=item.id,  # type: ignore
                                 traceId=trace_id,
+                                observationId=span.id,
                             )
                         )