Skip to content

Commit 3f2b5c5

Browse files
committed
make request and response models differ
1 parent 40a0008 commit 3f2b5c5

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

tests/models/test_instrumented.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@ async def request(
8484
{}, # test unexpected parts # type: ignore
8585
],
8686
usage=RequestUsage(input_tokens=100, output_tokens=200),
87-
model_name='gpt-4o',
87+
model_name='gpt-4o-2024-11-20',
8888
provider_details=dict(finish_reason='stop', foo='bar'),
8989
)
9090

@@ -111,7 +111,7 @@ async def _get_event_iterator(self) -> AsyncIterator[ModelResponseStreamEvent]:
111111

112112
@property
113113
def model_name(self) -> str:
114-
return 'gpt-4o'
114+
return 'gpt-4o-2024-11-20'
115115

116116
@property
117117
def provider_name(self) -> str:
@@ -182,7 +182,7 @@ async def test_instrumented_model(capfire: CaptureLogfire):
182182
'gen_ai.request.temperature': 1,
183183
'logfire.msg': 'chat gpt-4o',
184184
'logfire.span_type': 'span',
185-
'gen_ai.response.model': 'gpt-4o',
185+
'gen_ai.response.model': 'gpt-4o-2024-11-20',
186186
'gen_ai.usage.input_tokens': 100,
187187
'gen_ai.usage.output_tokens': 200,
188188
'operation.cost': 0.00225,
@@ -412,7 +412,7 @@ async def test_instrumented_model_stream(capfire: CaptureLogfire):
412412
'gen_ai.request.temperature': 1,
413413
'logfire.msg': 'chat gpt-4o',
414414
'logfire.span_type': 'span',
415-
'gen_ai.response.model': 'gpt-4o',
415+
'gen_ai.response.model': 'gpt-4o-2024-11-20',
416416
'gen_ai.usage.input_tokens': 300,
417417
'gen_ai.usage.output_tokens': 400,
418418
'operation.cost': 0.00475,
@@ -510,7 +510,7 @@ async def test_instrumented_model_stream_break(capfire: CaptureLogfire):
510510
'gen_ai.request.temperature': 1,
511511
'logfire.msg': 'chat gpt-4o',
512512
'logfire.span_type': 'span',
513-
'gen_ai.response.model': 'gpt-4o',
513+
'gen_ai.response.model': 'gpt-4o-2024-11-20',
514514
'gen_ai.usage.input_tokens': 300,
515515
'gen_ai.usage.output_tokens': 400,
516516
'operation.cost': 0.00475,
@@ -624,7 +624,7 @@ async def test_instrumented_model_attributes_mode(capfire: CaptureLogfire, instr
624624
'gen_ai.request.temperature': 1,
625625
'logfire.msg': 'chat gpt-4o',
626626
'logfire.span_type': 'span',
627-
'gen_ai.response.model': 'gpt-4o',
627+
'gen_ai.response.model': 'gpt-4o-2024-11-20',
628628
'gen_ai.usage.input_tokens': 100,
629629
'gen_ai.usage.output_tokens': 200,
630630
'events': [
@@ -806,7 +806,7 @@ async def test_instrumented_model_attributes_mode(capfire: CaptureLogfire, instr
806806
'finish_reason': 'stop',
807807
}
808808
],
809-
'gen_ai.response.model': 'gpt-4o',
809+
'gen_ai.response.model': 'gpt-4o-2024-11-20',
810810
'gen_ai.system_instructions': [{'type': 'text', 'content': 'instructions'}],
811811
'gen_ai.usage.input_tokens': 100,
812812
'gen_ai.usage.output_tokens': 200,

0 commit comments

Comments
 (0)