Assigning model version (#104)

thiago-aixplain · web-flow · commit 92d187066b52 · 2023-12-22T14:01:43.000-03:00
diff --git a/aixplain/factories/model_factory.py b/aixplain/factories/model_factory.py
@@ -69,6 +69,7 @@ def _create_model_from_response(cls, response: Dict) -> Model:
             function=Function(response["function"]["id"]),
             parameters=parameters,
             is_subscribed=True if "subscription" in response else False,
+            version=response["version"]["id"],
         )
 
     @classmethod
diff --git a/aixplain/modules/model.py b/aixplain/modules/model.py
@@ -59,7 +59,7 @@ def __init__(
         description: Text = "",
         api_key: Optional[Text] = None,
         supplier: Text = "aiXplain",
-        version: Text = "1.0",
+        version: Optional[Text] = None,
         function: Optional[Text] = None,
         is_subscribed: bool = False,
         **additional_info,

Original file line number	Diff line number	Diff line change
`@@ -69,6 +69,7 @@ def _create_model_from_response(cls, response: Dict) -> Model:`
`69`	`69`	`function=Function(response["function"]["id"]),`
`70`	`70`	`parameters=parameters,`
`71`	`71`	`is_subscribed=True if "subscription" in response else False,`
	`72`	`+ version=response["version"]["id"],`
`72`	`73`	`)`
`73`	`74`
`74`	`75`	`@classmethod`