truefoundry 0.5.9__py3-none-any.whl → 0.5.10__py3-none-any.whl

truefoundry/ml/autogen/client/models/{agent.py → agent_manifest.py}

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -22,57 +22,65 @@ from truefoundry.pydantic_v1 import (
     BaseModel,
     Field,
     StrictStr,
+    conint,
     conlist,
     constr,
     validator,
 )
 
 
-class Agent(BaseModel):
+class AgentManifest(BaseModel):
     """
-    Agent artifact.  # noqa: E501
+    Agent manifest.  # noqa: E501
     """
 
+    name: Optional[constr(strict=True, max_length=256)] = Field(
+        default=None,
+        description="+sort=1 +label=Name +message=The name should start with lowercase alphabets  and can contain alphanumeric and can include '-' in between +usage=Name of the entity",
+    )
     description: Optional[constr(strict=True, max_length=512)] = Field(
         default=None, description="+sort=2 +label=Description"
     )
     metadata: Dict[str, Any] = Field(
         default=...,
-        description="+label=Metadata +docs=Key value pairs to store additional metadata +usage=Key value pairs to store additional metadata +uiType=JsonInput",
+        description='+label=Metadata +usage=Key value metadata. Should be valid JSON. For e.g. `{"business-unit": "sales", "quality": "good", "rating": 4.5}` +uiType=JsonInput +uiProps={"descriptionInline":true}',
     )
     version_alias: Optional[constr(strict=True, max_length=128)] = Field(
         default=None,
-        description="+label=Version Alias +usage=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc) +docs=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc) +message=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc)",
+        description='+label=Version Alias +usage=Version alias is alternate, ideally human readable, version string to reference an artifact version. It should start with `v` followed by alphanumeric and it can include `.` and `-` in between (e.g. `v1.0.0`, `v1-prod`, `v3-dev`, etc) +message=The version alias should start with `v` followed by alphanumeric and can include `.` and `-` in between (e.g. `v1.0.0`, `v1-prod`, `v3-dev`, etc) +uiProps={"descriptionInline":true} +placeholder=v1.0.0',
     )
-    name: Optional[constr(strict=True, max_length=256)] = Field(
+    ml_repo: Optional[constr(strict=True)] = Field(
         default=None,
-        description="+sort=1 +label=Name +message=The name should start with lowercase alphabets  and can contain alphanumeric and can include '-' in between",
+        description="+label=ML Repo +usage=Name of the ML Repo +uiType=Hidden",
     )
-    ml_repo: Optional[constr(strict=True)] = Field(
+    version: Optional[conint(strict=True, ge=1)] = Field(
         default=None,
-        description="+label=ML Repo +usage=name of the ML Repo +docs=name of the ML Repo +uiType=Hidden",
+        description="+label=Version of the entity +usage=Version of the entity +uiType=Hidden",
     )
     type: Optional[StrictStr] = "agent"
     available_tools: conlist(constr(strict=True, min_length=1)) = Field(
-        default=..., description="+sort=30 +uiType=Hidden"
+        default=...,
+        description='+sort=30 +uiType=Hidden +label=Tools +usage=Tools available to the agent +uiProps={"descriptionInline":true}',
     )
     goal: constr(strict=True, max_length=128, min_length=1) = Field(
-        default=..., description="+sort=10 +uiType=TextArea"
+        default=...,
+        description='+sort=10 +uiType=TextArea +label=Goal +usage=Short form description. Will be used as `description` when this agent is used as a tool. +uiProps={"descriptionInline":true}',
     )
     instruction: constr(strict=True, max_length=2620, min_length=1) = Field(
         default=...,
-        description='`instruction` is the system prompt for now. (2.5 * 1024) +sort=20 +uiType=AgentInstructions +uiProps={"helpText":"Use the syntax ${Tool FQN} to reference a tool, and ${AGENT FQN} to reference another agent"}',
+        description='`instruction` is the system prompt for now. (2.5 * 1024) +sort=20 +uiType=AgentInstructions +uiProps={"helpText":"Use the syntax ${Tool FQN} to reference a tool, and ${AGENT FQN} to reference another agent"} +label=Instructions +usage=Instructions for the agent to follow to achieve the goal +uiProps={"descriptionInline":true}',
     )
     model_id: constr(strict=True, min_length=1) = Field(
         default=...,
-        description='+sort=40 +uiType=EnabledModelSelector +uiProps={"searchable":true,"modelType":"chat","providerType":"openai"}',
+        description='+sort=40 +uiType=EnabledModelSelector +uiProps={"searchable":true,"modelType":"chat","providerType":"openai"} +label=Model +usage=Model to use when running the agent',
     )
     __properties = [
+        "name",
         "description",
         "metadata",
         "version_alias",
-        "name",
         "ml_repo",
+        "version",
         "type",
         "available_tools",
         "goal",
@@ -80,27 +88,27 @@ class Agent(BaseModel):
         "model_id",
     ]
 
-    @validator("version_alias")
-    def version_alias_validate_regular_expression(cls, value):
+    @validator("name")
+    def name_validate_regular_expression(cls, value):
         """Validates the regular expression"""
         if value is None:
             return value
 
-        if not re.match(r"^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$", value):
+        if not re.match(r"^[A-Za-z0-9_\-]+$", value):
             raise ValueError(
-                r"must validate the regular expression /^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$/"
+                r"must validate the regular expression /^[A-Za-z0-9_\-]+$/"
             )
         return value
 
-    @validator("name")
-    def name_validate_regular_expression(cls, value):
+    @validator("version_alias")
+    def version_alias_validate_regular_expression(cls, value):
         """Validates the regular expression"""
         if value is None:
             return value
 
-        if not re.match(r"^[A-Za-z0-9_\-]+$", value):
+        if not re.match(r"^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$", value):
             raise ValueError(
-                r"must validate the regular expression /^[A-Za-z0-9_\-]+$/"
+                r"must validate the regular expression /^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$/"
             )
         return value
 
@@ -141,8 +149,8 @@ class Agent(BaseModel):
         return json.dumps(self.to_dict())
 
     @classmethod
-    def from_json(cls, json_str: str) -> Agent:
-        """Create an instance of Agent from a JSON string"""
+    def from_json(cls, json_str: str) -> AgentManifest:
+        """Create an instance of AgentManifest from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self):
@@ -151,21 +159,22 @@ class Agent(BaseModel):
         return _dict
 
     @classmethod
-    def from_dict(cls, obj: dict) -> Agent:
-        """Create an instance of Agent from a dict"""
+    def from_dict(cls, obj: dict) -> AgentManifest:
+        """Create an instance of AgentManifest from a dict"""
         if obj is None:
             return None
 
         if not isinstance(obj, dict):
-            return Agent.parse_obj(obj)
+            return AgentManifest.parse_obj(obj)
 
-        _obj = Agent.parse_obj(
+        _obj = AgentManifest.parse_obj(
             {
+                "name": obj.get("name"),
                 "description": obj.get("description"),
                 "metadata": obj.get("metadata"),
                 "version_alias": obj.get("version_alias"),
-                "name": obj.get("name"),
                 "ml_repo": obj.get("ml_repo"),
+                "version": obj.get("version"),
                 "type": obj.get("type") if obj.get("type") is not None else "agent",
                 "available_tools": obj.get("available_tools"),
                 "goal": obj.get("goal"),

truefoundry/ml/autogen/client/models/{agent_open_api_tool.py → agent_open_api_tool_manifest.py}

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -20,51 +20,69 @@ from typing import Any, Dict, Optional
 
 from truefoundry.ml.autogen.client.models.method import Method
 from truefoundry.ml.autogen.client.models.openapi_spec import OpenapiSpec
-from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, constr, validator
+from truefoundry.pydantic_v1 import (
+    BaseModel,
+    Field,
+    StrictStr,
+    conint,
+    constr,
+    validator,
+)
 
 
-class AgentOpenAPITool(BaseModel):
+class AgentOpenAPIToolManifest(BaseModel):
     """
-    OpenAPI Tool artifact.  # noqa: E501
+    Agent OpenAPI Tool manifest.  # noqa: E501
     """
 
+    name: Optional[constr(strict=True, max_length=256)] = Field(
+        default=None,
+        description="+sort=1 +label=Name +message=The name should start with lowercase alphabets  and can contain alphanumeric and can include '-' in between +usage=Name of the entity",
+    )
     description: Optional[constr(strict=True, max_length=512)] = Field(
         default=None, description="+sort=2 +label=Description"
     )
     metadata: Dict[str, Any] = Field(
         default=...,
-        description="+label=Metadata +docs=Key value pairs to store additional metadata +usage=Key value pairs to store additional metadata +uiType=JsonInput",
+        description='+label=Metadata +usage=Key value metadata. Should be valid JSON. For e.g. `{"business-unit": "sales", "quality": "good", "rating": 4.5}` +uiType=JsonInput +uiProps={"descriptionInline":true}',
     )
     version_alias: Optional[constr(strict=True, max_length=128)] = Field(
         default=None,
-        description="+label=Version Alias +usage=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc) +docs=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc) +message=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc)",
+        description='+label=Version Alias +usage=Version alias is alternate, ideally human readable, version string to reference an artifact version. It should start with `v` followed by alphanumeric and it can include `.` and `-` in between (e.g. `v1.0.0`, `v1-prod`, `v3-dev`, etc) +message=The version alias should start with `v` followed by alphanumeric and can include `.` and `-` in between (e.g. `v1.0.0`, `v1-prod`, `v3-dev`, etc) +uiProps={"descriptionInline":true} +placeholder=v1.0.0',
     )
-    name: Optional[constr(strict=True, max_length=256)] = Field(
+    ml_repo: Optional[constr(strict=True)] = Field(
         default=None,
-        description="+sort=1 +label=Name +message=The name should start with lowercase alphabets  and can contain alphanumeric and can include '-' in between",
+        description="+label=ML Repo +usage=Name of the ML Repo +uiType=Hidden",
     )
-    ml_repo: Optional[constr(strict=True)] = Field(
+    version: Optional[conint(strict=True, ge=1)] = Field(
         default=None,
-        description="+label=ML Repo +usage=name of the ML Repo +docs=name of the ML Repo +uiType=Hidden",
+        description="+label=Version of the entity +usage=Version of the entity +uiType=Hidden",
     )
     type: Optional[StrictStr] = "openapi-tool"
     openapi_spec: OpenapiSpec = Field(...)
-    base_url: constr(strict=True) = Field(default=..., description="+sort=30")
+    base_url: constr(strict=True) = Field(
+        default=...,
+        description='+sort=30 +label=Base URL +usage=HTTP endpoint where the API is hosted for the tools. E.g. `https://api.example.com/v1` +uiProps={"descriptionInline":true} +placeholder=https://api.example.com/v1',
+    )
     path: constr(strict=True, min_length=1) = Field(
         default=...,
-        description="+sort=40 +uiType=MethodPathSelector +label=Method + Path +usage=Select one Method and Path for this tool",
+        description='+sort=40 +uiType=MethodPathSelector +label=API Route Path +usage=API Route Path for the tool call HTTP request. E.g. `GET /weather` +uiProps={"descriptionInline":true}',
+    )
+    method: Method = Field(
+        default=...,
+        description="+sort=50 +uiType=Hidden +label=API HTTP Method +usage=HTTP Method for the tool call HTTP request",
     )
-    method: Method = Field(default=..., description="+sort=50 +uiType=Hidden")
     headers: Optional[Dict[str, StrictStr]] = Field(
         default=None,
-        description='+sort=60 +uiType=KV +uiProps={"allowSecrets":true,"secretConfig":{"enableNew":true,"hideOptions":true}}',
+        description='+sort=60 +uiType=KV +uiProps={"allowSecrets":true,"secretConfig":{"enableNew":true,"hideOptions":true}} +label=Headers +usage=HTTP Headers for the tool call HTTP request. E.g. `Authorization: Bearer <token>` +uiProps={"descriptionInline":true}',
     )
     __properties = [
+        "name",
         "description",
         "metadata",
         "version_alias",
-        "name",
         "ml_repo",
+        "version",
         "type",
         "openapi_spec",
         "base_url",
@@ -73,27 +91,27 @@ class AgentOpenAPITool(BaseModel):
         "headers",
     ]
 
-    @validator("version_alias")
-    def version_alias_validate_regular_expression(cls, value):
+    @validator("name")
+    def name_validate_regular_expression(cls, value):
         """Validates the regular expression"""
         if value is None:
             return value
 
-        if not re.match(r"^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$", value):
+        if not re.match(r"^[A-Za-z0-9_\-]+$", value):
             raise ValueError(
-                r"must validate the regular expression /^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$/"
+                r"must validate the regular expression /^[A-Za-z0-9_\-]+$/"
             )
         return value
 
-    @validator("name")
-    def name_validate_regular_expression(cls, value):
+    @validator("version_alias")
+    def version_alias_validate_regular_expression(cls, value):
         """Validates the regular expression"""
         if value is None:
             return value
 
-        if not re.match(r"^[A-Za-z0-9_\-]+$", value):
+        if not re.match(r"^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$", value):
             raise ValueError(
-                r"must validate the regular expression /^[A-Za-z0-9_\-]+$/"
+                r"must validate the regular expression /^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$/"
             )
         return value
 
@@ -146,8 +164,8 @@ class AgentOpenAPITool(BaseModel):
         return json.dumps(self.to_dict())
 
     @classmethod
-    def from_json(cls, json_str: str) -> AgentOpenAPITool:
-        """Create an instance of AgentOpenAPITool from a JSON string"""
+    def from_json(cls, json_str: str) -> AgentOpenAPIToolManifest:
+        """Create an instance of AgentOpenAPIToolManifest from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self):
@@ -159,21 +177,22 @@ class AgentOpenAPITool(BaseModel):
         return _dict
 
     @classmethod
-    def from_dict(cls, obj: dict) -> AgentOpenAPITool:
-        """Create an instance of AgentOpenAPITool from a dict"""
+    def from_dict(cls, obj: dict) -> AgentOpenAPIToolManifest:
+        """Create an instance of AgentOpenAPIToolManifest from a dict"""
         if obj is None:
             return None
 
         if not isinstance(obj, dict):
-            return AgentOpenAPITool.parse_obj(obj)
+            return AgentOpenAPIToolManifest.parse_obj(obj)
 
-        _obj = AgentOpenAPITool.parse_obj(
+        _obj = AgentOpenAPIToolManifest.parse_obj(
             {
+                "name": obj.get("name"),
                 "description": obj.get("description"),
                 "metadata": obj.get("metadata"),
                 "version_alias": obj.get("version_alias"),
-                "name": obj.get("name"),
                 "ml_repo": obj.get("ml_repo"),
+                "version": obj.get("version"),
                 "type": obj.get("type")
                 if obj.get("type") is not None
                 else "openapi-tool",

truefoundry/ml/autogen/client/models/agent_open_api_tool_with_fqn.py

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -20,7 +20,14 @@ from typing import Any, Dict, Optional
 
 from truefoundry.ml.autogen.client.models.method import Method
 from truefoundry.ml.autogen.client.models.openapi_spec import OpenapiSpec
-from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, constr, validator
+from truefoundry.pydantic_v1 import (
+    BaseModel,
+    Field,
+    StrictStr,
+    conint,
+    constr,
+    validator,
+)
 
 
 class AgentOpenAPIToolWithFQN(BaseModel):
@@ -28,45 +35,58 @@ class AgentOpenAPIToolWithFQN(BaseModel):
     AgentOpenAPIToolWithFQN
     """
 
+    name: Optional[constr(strict=True, max_length=256)] = Field(
+        default=None,
+        description="+sort=1 +label=Name +message=The name should start with lowercase alphabets  and can contain alphanumeric and can include '-' in between +usage=Name of the entity",
+    )
     description: Optional[constr(strict=True, max_length=512)] = Field(
         default=None, description="+sort=2 +label=Description"
     )
     metadata: Dict[str, Any] = Field(
         default=...,
-        description="+label=Metadata +docs=Key value pairs to store additional metadata +usage=Key value pairs to store additional metadata +uiType=JsonInput",
+        description='+label=Metadata +usage=Key value metadata. Should be valid JSON. For e.g. `{"business-unit": "sales", "quality": "good", "rating": 4.5}` +uiType=JsonInput +uiProps={"descriptionInline":true}',
     )
     version_alias: Optional[constr(strict=True, max_length=128)] = Field(
         default=None,
-        description="+label=Version Alias +usage=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc) +docs=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc) +message=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc)",
+        description='+label=Version Alias +usage=Version alias is alternate, ideally human readable, version string to reference an artifact version. It should start with `v` followed by alphanumeric and it can include `.` and `-` in between (e.g. `v1.0.0`, `v1-prod`, `v3-dev`, etc) +message=The version alias should start with `v` followed by alphanumeric and can include `.` and `-` in between (e.g. `v1.0.0`, `v1-prod`, `v3-dev`, etc) +uiProps={"descriptionInline":true} +placeholder=v1.0.0',
     )
-    name: Optional[constr(strict=True, max_length=256)] = Field(
+    ml_repo: Optional[constr(strict=True)] = Field(
         default=None,
-        description="+sort=1 +label=Name +message=The name should start with lowercase alphabets  and can contain alphanumeric and can include '-' in between",
+        description="+label=ML Repo +usage=Name of the ML Repo +uiType=Hidden",
     )
-    ml_repo: Optional[constr(strict=True)] = Field(
+    version: Optional[conint(strict=True, ge=1)] = Field(
         default=None,
-        description="+label=ML Repo +usage=name of the ML Repo +docs=name of the ML Repo +uiType=Hidden",
+        description="+label=Version of the entity +usage=Version of the entity +uiType=Hidden",
+    )
+    type: StrictStr = Field(
+        default=..., description="+value=openapi-tool +usage=Type of the agent"
     )
-    type: StrictStr = Field(default=..., description="+value=openapi-tool")
     openapi_spec: OpenapiSpec = Field(...)
-    base_url: constr(strict=True) = Field(default=..., description="+sort=30")
+    base_url: constr(strict=True) = Field(
+        default=...,
+        description='+sort=30 +label=Base URL +usage=HTTP endpoint where the API is hosted for the tools. E.g. `https://api.example.com/v1` +uiProps={"descriptionInline":true} +placeholder=https://api.example.com/v1',
+    )
     path: constr(strict=True, min_length=1) = Field(
         default=...,
-        description="+sort=40 +uiType=MethodPathSelector +label=Method + Path +usage=Select one Method and Path for this tool",
+        description='+sort=40 +uiType=MethodPathSelector +label=API Route Path +usage=API Route Path for the tool call HTTP request. E.g. `GET /weather` +uiProps={"descriptionInline":true}',
+    )
+    method: Method = Field(
+        default=...,
+        description="+sort=50 +uiType=Hidden +label=API HTTP Method +usage=HTTP Method for the tool call HTTP request",
     )
-    method: Method = Field(default=..., description="+sort=50 +uiType=Hidden")
     headers: Optional[Dict[str, StrictStr]] = Field(
         default=None,
-        description='+sort=60 +uiType=KV +uiProps={"allowSecrets":true,"secretConfig":{"enableNew":true,"hideOptions":true}}',
+        description='+sort=60 +uiType=KV +uiProps={"allowSecrets":true,"secretConfig":{"enableNew":true,"hideOptions":true}} +label=Headers +usage=HTTP Headers for the tool call HTTP request. E.g. `Authorization: Bearer <token>` +uiProps={"descriptionInline":true}',
     )
-    id: StrictStr = Field(...)
-    fqn: StrictStr = Field(...)
+    id: StrictStr = Field(default=..., description="+usage=ID of the agent tool")
+    fqn: StrictStr = Field(default=..., description="+usage=FQN of the agent tool")
     __properties = [
+        "name",
         "description",
         "metadata",
         "version_alias",
-        "name",
         "ml_repo",
+        "version",
         "type",
         "openapi_spec",
         "base_url",
@@ -77,27 +97,27 @@ class AgentOpenAPIToolWithFQN(BaseModel):
         "fqn",
     ]
 
-    @validator("version_alias")
-    def version_alias_validate_regular_expression(cls, value):
+    @validator("name")
+    def name_validate_regular_expression(cls, value):
         """Validates the regular expression"""
         if value is None:
             return value
 
-        if not re.match(r"^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$", value):
+        if not re.match(r"^[A-Za-z0-9_\-]+$", value):
             raise ValueError(
-                r"must validate the regular expression /^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$/"
+                r"must validate the regular expression /^[A-Za-z0-9_\-]+$/"
             )
         return value
 
-    @validator("name")
-    def name_validate_regular_expression(cls, value):
+    @validator("version_alias")
+    def version_alias_validate_regular_expression(cls, value):
         """Validates the regular expression"""
         if value is None:
             return value
 
-        if not re.match(r"^[A-Za-z0-9_\-]+$", value):
+        if not re.match(r"^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$", value):
             raise ValueError(
-                r"must validate the regular expression /^[A-Za-z0-9_\-]+$/"
+                r"must validate the regular expression /^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$/"
             )
         return value
 
@@ -170,11 +190,12 @@ class AgentOpenAPIToolWithFQN(BaseModel):
 
         _obj = AgentOpenAPIToolWithFQN.parse_obj(
             {
+                "name": obj.get("name"),
                 "description": obj.get("description"),
                 "metadata": obj.get("metadata"),
                 "version_alias": obj.get("version_alias"),
-                "name": obj.get("name"),
                 "ml_repo": obj.get("ml_repo"),
+                "version": obj.get("version"),
                 "type": obj.get("type"),
                 "openapi_spec": OpenapiSpec.from_dict(obj.get("openapi_spec"))
                 if obj.get("openapi_spec") is not None