Skip to content

Commit c5e076e

Browse files
feat(llm_inference): add model_id field in Deployment message (#548)
Co-authored-by: Laure-di <62625835+Laure-di@users.noreply.github.com>
1 parent 01ca415 commit c5e076e

File tree

4 files changed

+18
-0
lines changed

4 files changed

+18
-0
lines changed

scaleway-async/scaleway_async/inference/v1beta1/marshalling.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -153,6 +153,10 @@ def unmarshal_Deployment(data: Any) -> Deployment:
153153
if field is not None:
154154
args["model_name"] = field
155155

156+
field = data.get("model_id", None)
157+
if field is not None:
158+
args["model_id"] = field
159+
156160
field = data.get("region", None)
157161
if field is not None:
158162
args["region"] = field

scaleway-async/scaleway_async/inference/v1beta1/types.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -220,6 +220,11 @@ class Deployment:
220220
The inference model used for the deployment.
221221
"""
222222

223+
model_id: str
224+
"""
225+
ID of the model used for the deployment.
226+
"""
227+
223228
region: Region
224229
"""
225230
Region of the deployment.

scaleway/scaleway/inference/v1beta1/marshalling.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -153,6 +153,10 @@ def unmarshal_Deployment(data: Any) -> Deployment:
153153
if field is not None:
154154
args["model_name"] = field
155155

156+
field = data.get("model_id", None)
157+
if field is not None:
158+
args["model_id"] = field
159+
156160
field = data.get("region", None)
157161
if field is not None:
158162
args["region"] = field

scaleway/scaleway/inference/v1beta1/types.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -220,6 +220,11 @@ class Deployment:
220220
The inference model used for the deployment.
221221
"""
222222

223+
model_id: str
224+
"""
225+
ID of the model used for the deployment.
226+
"""
227+
223228
region: Region
224229
"""
225230
Region of the deployment.

0 commit comments

Comments
 (0)