feat: extract base properties and extend coded

Author: cotovanu-cristian

src/uipath/agent/models/agent.py

@@ -502,24 +502,12 @@ class AgentSettings(BaseModel):
 class BaseAgentDefinition(BaseModel):
     """Agent definition model."""
 
-    id: str = Field(..., description="Agent id or project name")
-    name: str = Field(..., description="Agent name or project name")
-    metadata: Optional[AgentMetadata] = Field(None, description="Agent metadata")
-    messages: List[AgentMessage] = Field(
-        ..., description="List of system and user messages"
-    )
     input_schema: Dict[str, Any] = Field(
         ..., alias="inputSchema", description="JSON schema for input arguments"
     )
     output_schema: Dict[str, Any] = Field(
         ..., alias="outputSchema", description="JSON schema for output arguments"
     )
-    version: str = Field("1.0.0", description="Agent version")
-    resources: List[AgentResourceConfig] = Field(
-        ..., description="List of tools, context, mcp and escalation resources"
-    )
-    features: List[Any] = Field(default_factory=list, description="Agent feature list")
-    settings: AgentSettings = Field(..., description="Agent settings")
 
     model_config = ConfigDict(
         validate_by_name=True, validate_by_alias=True, extra="allow"
@@ -530,18 +518,41 @@ class AgentType(str, Enum):
     """Agent type."""
 
     LOW_CODE = "lowCode"
+    CODED = "coded"
 
 
 class LowCodeAgentDefinition(BaseAgentDefinition):
     """Low code agent definition."""
 
     type: Literal[AgentType.LOW_CODE] = AgentType.LOW_CODE
 
+    id: str = Field(..., description="Agent id or project name")
+    name: str = Field(..., description="Agent name or project name")
+    metadata: Optional[AgentMetadata] = Field(None, description="Agent metadata")
+    messages: List[AgentMessage] = Field(
+        ..., description="List of system and user messages"
+    )
+
+    version: str = Field("1.0.0", description="Agent version")
+    resources: List[AgentResourceConfig] = Field(
+        ..., description="List of tools, context, mcp and escalation resources"
+    )
+    features: List[Any] = Field(default_factory=list, description="Agent feature list")
+    settings: AgentSettings = Field(..., description="Agent settings")
 
-class CodedAgentDefinition(BaseModel):
+    model_config = ConfigDict(
+        validate_by_name=True, validate_by_alias=True, extra="allow"
+    )
+
+
+class CodedAgentDefinition(BaseAgentDefinition):
     """Coded agent definition."""
 
-    pass
+    type: Literal[AgentType.CODED] = AgentType.CODED
+
+    model_config = ConfigDict(
+        validate_by_name=True, validate_by_alias=True, extra="allow"
+    )
 
 
 KnownAgentDefinition = Annotated[

tests/agent/models/test_agent.py

@@ -49,12 +49,13 @@ def test_agent_config_loads_unknown_agent_type(self):
         assert isinstance(config, UnknownAgentDefinition), (
             "AgentDefinition should be an unknown type."
         )
-        assert config.id == "b2564199-e479-4b6f-9336-dc50f457afda"
-        assert config.name == "Agent"
-        assert config.version == "1.0.0"
+        config_data = config.model_dump()
+        assert config_data["id"] == "b2564199-e479-4b6f-9336-dc50f457afda"
+        assert config_data["name"] == "Agent"
+        assert config_data["version"] == "1.0.0"
 
         # Validate resources
-        assert len(config.resources) == 0
+        assert len(config_data["resources"]) == 0
 
     def test_agent_with_all_tool_types_loads(self):
         """Test that AgentDefinition can load a complete agent package with all tool types"""