diff --git a/docsgpt/logging.py b/docsgpt/logging.py
index 2ece4e1f..be1d3659 100644
--- a/docsgpt/logging.py
+++ b/docsgpt/logging.py
@@ -97,7 +97,11 @@ def _agent_log_keys(agent: Any, data: Optional[Dict] = None) -> Dict[str, Any]:
         "agent_id": getattr(agent, "agent_id", None),
         "conversation_id": getattr(agent, "conversation_id", None),
         "endpoint": data.get("endpoint", ""),
-        "model": getattr(agent, "gpt_model", None) or getattr(agent, "model", None),
+        "model": (
+            getattr(agent, "gpt_model", None)
+            or getattr(agent, "model", None)
+            or getattr(agent, "model_id", None)
+        ),
     }
 
 
diff --git a/tests/agents/test_continuation_log_context.py b/tests/agents/test_continuation_log_context.py
index 28a3d21a..6392e154 100644
--- a/tests/agents/test_continuation_log_context.py
+++ b/tests/agents/test_continuation_log_context.py
@@ -74,6 +74,7 @@ def test_model_calls_in_a_continuation_carry_the_turns_identity():
     assert seen[0]["user_id"] == "user-cont"
     assert seen[0]["agent_id"] == "aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa"
     assert seen[0]["endpoint"] == "stream"
+    assert seen[0]["model"] == "gpt-4"
     assert "activity_id" not in seen[0], "a continuation is not an activity of its own"