tweak repr param typing

CHANGELOG.md

@@ -10,6 +10,7 @@ All notable changes to this project will be documented in this file, following t
 - Breaking: Renamed geometrical primitive `line` to `tube`
 - Breaking: Change multiple geometrical primitive parameters
 - Multi-state data model tweaks
+- Support for representation-specific parameters (customize presentation visuals)
 
 
 ## [v1.1.0] - 2024-12-09

molviewspec/app/api/examples.py

@@ -736,7 +736,7 @@ async def repr_params_example() -> MVSResponse:
     """
     builder = create_builder()
     component = (
-        builder.download(url=_url_for_mmcif("4hhb")).parse(format="mmcif").model_structure(ref="structure").component()
+        builder.download(url=_url_for_mmcif("1a23")).parse(format="mmcif").model_structure(ref="structure").component()
     )
     component.representation(type="cartoon", size_factor=1.5, tubular_helices=True)
     component.representation(type="surface", ignore_hydrogens=True).opacity(opacity=0.8)

molviewspec/molviewspec/nodes.py

@@ -5,7 +5,7 @@
 from __future__ import annotations
 
 from datetime import datetime, timezone
-from typing import Any, Dict, Literal, Mapping, Optional, Tuple, Type, TypeVar, Union
+from typing import Any, Literal, Mapping, Optional, Tuple, TypeVar, Union
 from uuid import uuid4
 
 from pydantic import BaseModel, Field
@@ -439,26 +439,25 @@ class RepresentationParams(BaseModel):
 
 
 class CartoonParams(RepresentationParams):
+    type: Literal["cartoon"] = "cartoon"
     size_factor: Optional[float] = Field(description="Scales the corresponding visuals.")
     tubular_helices: Optional[bool] = Field(description="Simplify corkscrew helices to tubes.")
     # TODO support for variable size, e.g. based on b-factors?
 
 
 class BallAndStickParams(RepresentationParams):
+    type: Literal["ball_and_stick"] = "ball_and_stick"
     ignore_hydrogens: Optional[bool] = Field(descripton="Controls whether hydrogen atoms are drawn.")
     size_factor: Optional[float] = Field(description="Scales the corresponding visuals.")
 
 
 class SurfaceParams(RepresentationParams):
+    type: Literal["surface"] = "surface"
     ignore_hydrogens: Optional[bool] = Field(descripton="Controls whether hydrogen atoms are drawn.")
     size_factor: Optional[float] = Field(description="Scales the corresponding visuals.")
 
 
-RepresentationTypeParams: Dict[RepresentationTypeT, Type[Union[CartoonParams, BallAndStickParams, SurfaceParams]]] = {
-    "cartoon": CartoonParams,
-    "ball_and_stick": BallAndStickParams,
-    "surface": SurfaceParams,
-}
+RepresentationTypeParams = {t.__fields__["type"].default: t for t in (CartoonParams, BallAndStickParams, SurfaceParams)}
 
 
 SchemaT = Literal[