Skip to content

Commit 65e5473

Browse files
hassiebphassiebp
committed
push
1 parent 54a2e43 commit 65e5473

1 file changed

Lines changed: 41 additions & 35 deletions

File tree

‎tests/test_openai.py‎

Lines changed: 41 additions & 35 deletions
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ def test_openai_chat_completion(openai):
3939

4040
sleep(1)
4141

42-
generation = get_api().observations.get_many(
42+
generation = get_api().legacy.observations_v1.get_many(
4343
name=generation_name, type="GENERATION"
4444
)
4545

@@ -96,7 +96,7 @@ def test_openai_chat_completion_stream(openai):
9696
langfuse.flush()
9797
sleep(3)
9898

99-
generation = get_api().observations.get_many(
99+
generation = get_api().legacy.observations_v1.get_many(
100100
name=generation_name, type="GENERATION"
101101
)
102102

@@ -156,7 +156,7 @@ def test_openai_chat_completion_stream_with_next_iteration(openai):
156156

157157
langfuse.flush()
158158

159-
generation = get_api().observations.get_many(
159+
generation = get_api().legacy.observations_v1.get_many(
160160
name=generation_name, type="GENERATION"
161161
)
162162

@@ -205,7 +205,7 @@ def test_openai_chat_completion_stream_fail(openai):
205205

206206
langfuse.flush()
207207

208-
generation = get_api().observations.get_many(
208+
generation = get_api().legacy.observations_v1.get_many(
209209
name=generation_name, type="GENERATION"
210210
)
211211

@@ -255,7 +255,7 @@ def test_openai_chat_completion_with_langfuse_prompt(openai):
255255

256256
langfuse.flush()
257257

258-
generation = get_api().observations.get_many(
258+
generation = get_api().legacy.observations_v1.get_many(
259259
name=generation_name, type="GENERATION"
260260
)
261261

@@ -278,7 +278,7 @@ def test_openai_chat_completion_fail(openai):
278278

279279
langfuse.flush()
280280

281-
generation = get_api().observations.get_many(
281+
generation = get_api().legacy.observations_v1.get_many(
282282
name=generation_name, type="GENERATION"
283283
)
284284

@@ -338,7 +338,7 @@ def test_openai_chat_completion_two_calls(openai):
338338

339339
langfuse.flush()
340340

341-
generation = get_api().observations.get_many(
341+
generation = get_api().legacy.observations_v1.get_many(
342342
name=generation_name, type="GENERATION"
343343
)
344344

@@ -348,7 +348,7 @@ def test_openai_chat_completion_two_calls(openai):
348348

349349
assert generation.data[0].input == [{"content": "1 + 1 = ", "role": "user"}]
350350

351-
generation_2 = get_api().observations.get_many(
351+
generation_2 = get_api().legacy.observations_v1.get_many(
352352
name=generation_name_2, type="GENERATION"
353353
)
354354

@@ -372,7 +372,7 @@ def test_openai_chat_completion_with_seed(openai):
372372

373373
langfuse.flush()
374374

375-
generation = get_api().observations.get_many(
375+
generation = get_api().legacy.observations_v1.get_many(
376376
name=generation_name, type="GENERATION"
377377
)
378378

@@ -399,7 +399,7 @@ def test_openai_completion(openai):
399399

400400
langfuse.flush()
401401

402-
generation = get_api().observations.get_many(
402+
generation = get_api().legacy.observations_v1.get_many(
403403
name=generation_name, type="GENERATION"
404404
)
405405

@@ -447,7 +447,7 @@ def test_openai_completion_stream(openai):
447447

448448
assert len(content) > 0
449449

450-
generation = get_api().observations.get_many(
450+
generation = get_api().legacy.observations_v1.get_many(
451451
name=generation_name, type="GENERATION"
452452
)
453453

@@ -496,7 +496,7 @@ def test_openai_completion_fail(openai):
496496

497497
langfuse.flush()
498498

499-
generation = get_api().observations.get_many(
499+
generation = get_api().legacy.observations_v1.get_many(
500500
name=generation_name, type="GENERATION"
501501
)
502502

@@ -539,7 +539,7 @@ def test_openai_completion_stream_fail(openai):
539539

540540
langfuse.flush()
541541

542-
generation = get_api().observations.get_many(
542+
generation = get_api().legacy.observations_v1.get_many(
543543
name=generation_name, type="GENERATION"
544544
)
545545

@@ -588,7 +588,7 @@ def test_openai_completion_with_langfuse_prompt(openai):
588588

589589
langfuse.flush()
590590

591-
generation = get_api().observations.get_many(
591+
generation = get_api().legacy.observations_v1.get_many(
592592
name=generation_name, type="GENERATION"
593593
)
594594

@@ -630,7 +630,7 @@ async def test_async_chat(openai):
630630

631631
langfuse.flush()
632632

633-
generation = get_api().observations.get_many(
633+
generation = get_api().legacy.observations_v1.get_many(
634634
name=generation_name, type="GENERATION"
635635
)
636636

@@ -676,7 +676,7 @@ async def test_async_chat_stream(openai):
676676

677677
langfuse.flush()
678678

679-
generation = get_api().observations.get_many(
679+
generation = get_api().legacy.observations_v1.get_many(
680680
name=generation_name, type="GENERATION"
681681
)
682682

@@ -734,7 +734,7 @@ async def test_async_chat_stream_with_anext(openai):
734734

735735
print(result)
736736

737-
generation = get_api().observations.get_many(
737+
generation = get_api().legacy.observations_v1.get_many(
738738
name=generation_name, type="GENERATION"
739739
)
740740

@@ -796,7 +796,7 @@ class StepByStepAIResponse(BaseModel):
796796

797797
langfuse.flush()
798798

799-
generation = get_api().observations.get_many(
799+
generation = get_api().legacy.observations_v1.get_many(
800800
name=generation_name, type="GENERATION"
801801
)
802802

@@ -840,7 +840,7 @@ class StepByStepAIResponse(BaseModel):
840840

841841
langfuse.flush()
842842

843-
generation = get_api().observations.get_many(
843+
generation = get_api().legacy.observations_v1.get_many(
844844
name=generation_name, type="GENERATION"
845845
)
846846

@@ -884,7 +884,7 @@ def test_openai_tool_call(openai):
884884

885885
langfuse.flush()
886886

887-
generation = get_api().observations.get_many(
887+
generation = get_api().legacy.observations_v1.get_many(
888888
name=generation_name, type="GENERATION"
889889
)
890890

@@ -940,7 +940,7 @@ def test_openai_tool_call_streamed(openai):
940940

941941
langfuse.flush()
942942

943-
generation = get_api().observations.get_many(
943+
generation = get_api().legacy.observations_v1.get_many(
944944
name=generation_name, type="GENERATION"
945945
)
946946

@@ -1018,7 +1018,7 @@ def test_structured_output_response_format_kwarg(openai):
10181018

10191019
langfuse.flush()
10201020

1021-
generation = get_api().observations.get_many(
1021+
generation = get_api().legacy.observations_v1.get_many(
10221022
name=generation_name, type="GENERATION"
10231023
)
10241024

@@ -1087,7 +1087,7 @@ class CalendarEvent(BaseModel):
10871087

10881088
if Version(openai.__version__) >= Version("1.50.0"):
10891089
# Check the trace and observation properties
1090-
generation = get_api().observations.get_many(
1090+
generation = get_api().legacy.observations_v1.get_many(
10911091
name=generation_name, type="GENERATION"
10921092
)
10931093

@@ -1133,7 +1133,7 @@ async def test_close_async_stream(openai):
11331133

11341134
langfuse.flush()
11351135

1136-
generation = get_api().observations.get_many(
1136+
generation = get_api().legacy.observations_v1.get_many(
11371137
name=generation_name, type="GENERATION"
11381138
)
11391139

@@ -1194,7 +1194,7 @@ def test_base_64_image_input(openai):
11941194

11951195
langfuse.flush()
11961196

1197-
generation = get_api().observations.get_many(
1197+
generation = get_api().legacy.observations_v1.get_many(
11981198
name=generation_name, type="GENERATION"
11991199
)
12001200

@@ -1245,7 +1245,7 @@ def test_audio_input_and_output(openai):
12451245

12461246
langfuse.flush()
12471247

1248-
generation = get_api().observations.get_many(
1248+
generation = get_api().legacy.observations_v1.get_many(
12491249
name=generation_name, type="GENERATION"
12501250
)
12511251

@@ -1285,7 +1285,7 @@ def test_response_api_text_input(openai):
12851285
)
12861286

12871287
langfuse.flush()
1288-
generation = get_api().observations.get_many(
1288+
generation = get_api().legacy.observations_v1.get_many(
12891289
name=generation_name, type="GENERATION"
12901290
)
12911291

@@ -1331,7 +1331,7 @@ def test_response_api_image_input(openai):
13311331

13321332
langfuse.flush()
13331333

1334-
generation = get_api().observations.get_many(
1334+
generation = get_api().legacy.observations_v1.get_many(
13351335
name=generation_name, type="GENERATION"
13361336
)
13371337

@@ -1363,7 +1363,7 @@ def test_response_api_web_search(openai):
13631363

13641364
langfuse.flush()
13651365

1366-
generation = get_api().observations.get_many(
1366+
generation = get_api().legacy.observations_v1.get_many(
13671367
name=generation_name, type="GENERATION"
13681368
)
13691369

@@ -1400,7 +1400,7 @@ def test_response_api_streaming(openai):
14001400

14011401
langfuse.flush()
14021402

1403-
generation = get_api().observations.get_many(
1403+
generation = get_api().legacy.observations_v1.get_many(
14041404
name=generation_name, type="GENERATION"
14051405
)
14061406

@@ -1454,7 +1454,7 @@ def test_response_api_functions(openai):
14541454

14551455
langfuse.flush()
14561456

1457-
generation = get_api().observations.get_many(
1457+
generation = get_api().legacy.observations_v1.get_many(
14581458
name=generation_name, type="GENERATION"
14591459
)
14601460

@@ -1486,7 +1486,7 @@ def test_response_api_reasoning(openai):
14861486
)
14871487
langfuse.flush()
14881488

1489-
generation = get_api().observations.get_many(
1489+
generation = get_api().legacy.observations_v1.get_many(
14901490
name=generation_name, type="GENERATION"
14911491
)
14921492

@@ -1518,7 +1518,9 @@ def test_openai_embeddings(openai):
15181518
langfuse.flush()
15191519
sleep(1)
15201520

1521-
embedding = get_api().observations.get_many(name=embedding_name, type="EMBEDDING")
1521+
embedding = get_api().legacy.observations_v1.get_many(
1522+
name=embedding_name, type="EMBEDDING"
1523+
)
15221524

15231525
assert len(embedding.data) != 0
15241526
embedding_data = embedding.data[0]
@@ -1552,7 +1554,9 @@ def test_openai_embeddings_multiple_inputs(openai):
15521554
langfuse.flush()
15531555
sleep(1)
15541556

1555-
embedding = get_api().observations.get_many(name=embedding_name, type="EMBEDDING")
1557+
embedding = get_api().legacy.observations_v1.get_many(
1558+
name=embedding_name, type="EMBEDDING"
1559+
)
15561560

15571561
assert len(embedding.data) != 0
15581562
embedding_data = embedding.data[0]
@@ -1583,7 +1587,9 @@ async def test_async_openai_embeddings(openai):
15831587
langfuse.flush()
15841588
sleep(1)
15851589

1586-
embedding = get_api().observations.get_many(name=embedding_name, type="EMBEDDING")
1590+
embedding = get_api().legacy.observations_v1.get_many(
1591+
name=embedding_name, type="EMBEDDING"
1592+
)
15871593

15881594
assert len(embedding.data) != 0
15891595
embedding_data = embedding.data[0]

0 commit comments

Comments
 (0)