push

Author: hassiebp

langfuse/batch_evaluation.py

@@ -1173,7 +1173,7 @@ async def _fetch_batch_with_retry(
             )  # type: ignore
             return list(response.data)  # type: ignore
         elif scope == "observations":
-            response = self.client.api.observations.get_many(
+            response = self.client.api.legacy.observations_v1.get_many(
                 page=page,
                 limit=limit,
                 filter=filter,

tests/test_core_sdk.py

@@ -50,7 +50,7 @@ async def update_generation(i, langfuse: Langfuse):
     api = get_api()
     for i in range(100):
         # Find the observations with the expected name
-        observations = api.observations.get_many(name=str(i)).data
+        observations = api.legacy.observations_v1.get_many(name=str(i)).data
 
         # Find generation observations (there should be at least one)
         generation_obs = [obs for obs in observations if obs.type == "GENERATION"]

tests/test_prompt.py

@@ -676,7 +676,7 @@ def test_prompt_end_to_end():
     generation = trace.observations[0]
     assert generation.prompt_id is not None
 
-    observation = api.observations.get(generation.id)
+    observation = api.legacy.observations_v1.get(generation.id)
 
     assert observation.prompt_id is not None