Skip to content

Commit 2fced79

Browse files
committed
Update test to send required model param
1 parent 0fc6b7b commit 2fced79

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/test_logger_batch.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1046,6 +1046,7 @@ def test_get_last_output() -> None:
10461046
input="input",
10471047
output="llm output",
10481048
name="test-span",
1049+
model="test-model",
10491050
created_at=datetime.datetime.now(),
10501051
duration_ns=1_000_000,
10511052
status_code=200,
@@ -1280,7 +1281,7 @@ def test_set_session_id(mock_traces_client: Mock, mock_projects_client: Mock, mo
12801281

12811282
# Log a trace
12821283
logger.start_trace(input="input", name="test-trace", created_at=datetime.datetime.now())
1283-
logger.add_llm_span(input="input", output="output")
1284+
logger.add_llm_span(input="input", output="output", model="test-model")
12841285
logger.conclude("output", status_code=200)
12851286
logger.flush()
12861287

@@ -1352,7 +1353,7 @@ def test_start_session_with_external_id(
13521353

13531354
# Log a trace
13541355
logger.start_trace(input="input", name="test-trace", created_at=datetime.datetime.now())
1355-
logger.add_llm_span(input="input", output="output")
1356+
logger.add_llm_span(input="input", output="output", model="test-model")
13561357
logger.conclude("output", status_code=200)
13571358
logger.flush()
13581359

0 commit comments

Comments
 (0)