Skip to content

Commit 039b8f5

Browse files
committed
fix: update return type of latencyMs metric for ollama model provider
1 parent 6e208a8 commit 039b8f5

2 files changed

Lines changed: 4 additions & 4 deletions

File tree

src/strands/models/ollama.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -280,7 +280,7 @@ def format_chunk(self, event: dict[str, Any]) -> StreamEvent:
280280
"totalTokens": event["data"].eval_count + event["data"].prompt_eval_count,
281281
},
282282
"metrics": {
283-
"latencyMs": event["data"].total_duration / 1e6,
283+
"latencyMs": int(event["data"].total_duration / 1e6),
284284
},
285285
},
286286
}

tests/strands/models/test_ollama.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -407,7 +407,7 @@ def test_format_chunk_metadata(model):
407407
"totalTokens": 150,
408408
},
409409
"metrics": {
410-
"latencyMs": 1.0,
410+
"latencyMs": 1,
411411
},
412412
},
413413
}
@@ -447,7 +447,7 @@ async def test_stream(ollama_client, model, agenerator, alist, captured_warnings
447447
{
448448
"metadata": {
449449
"usage": {"inputTokens": 5, "outputTokens": 10, "totalTokens": 15},
450-
"metrics": {"latencyMs": 1.0},
450+
"metrics": {"latencyMs": 1},
451451
}
452452
},
453453
]
@@ -525,7 +525,7 @@ async def test_stream_with_tool_calls(ollama_client, model, agenerator, alist):
525525
assert tru_events[8] == {
526526
"metadata": {
527527
"usage": {"inputTokens": 8, "outputTokens": 15, "totalTokens": 23},
528-
"metrics": {"latencyMs": 2.0},
528+
"metrics": {"latencyMs": 2},
529529
}
530530
}
531531
expected_request = {

0 commit comments

Comments
 (0)