diff --git a/src/opencode_ai/types/session_init_params.py b/src/opencode_ai/types/session_init_params.py
index 8d6d0d8..9e0b554 100644
--- a/src/opencode_ai/types/session_init_params.py
+++ b/src/opencode_ai/types/session_init_params.py
@@ -5,6 +5,7 @@ from __future__ import annotations
 from typing_extensions import Required, Annotated, TypedDict
 
 from .._utils import PropertyInfo
+from .session_chat_params import ModelParam
 
 __all__ = ["SessionInitParams"]
 
@@ -12,6 +13,4 @@ __all__ = ["SessionInitParams"]
 class SessionInitParams(TypedDict, total=False):
     message_id: Required[Annotated[str, PropertyInfo(alias="messageID")]]
 
-    model_id: Required[Annotated[str, PropertyInfo(alias="modelID")]]
-
-    provider_id: Required[Annotated[str, PropertyInfo(alias="providerID")]]
+    model: Required[ModelParam]