Add model context capabilities to Python (#564)

Python SDK: add contextLength, inputModalities, outputModalities,
capabilities; also added tests for these fields

---------

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

Author: bmehta001

sdk/python/README.md

@@ -142,6 +142,31 @@ cached = catalog.get_cached_models()
 loaded = catalog.get_loaded_models()
 ```
 
+### Inspecting Model Metadata
+
+`Model` exposes metadata properties from the catalog:
+
+```python
+model = catalog.get_model("phi-3.5-mini")
+
+# Identity
+print(model.id)             # e.g. "phi-3.5-mini-instruct-generic-gpu:3"
+print(model.alias)          # e.g. "phi-3.5-mini"
+
+# Context and token limits
+print(model.context_length) # e.g. 131072 (tokens), or None if unknown
+
+# Modalities and capabilities
+print(model.input_modalities)   # e.g. "text" or "text,image"
+print(model.output_modalities)  # e.g. "text"
+print(model.capabilities)       # e.g. "chat,completion"
+print(model.supports_tool_calling)  # True, False, or None
+
+# Cache / load state
+print(model.is_cached)
+print(model.is_loaded)
+```
+
 ### Loading and Running a Model
 
 ```python

sdk/python/src/detail/model_data_types.py

@@ -74,3 +74,7 @@ class ModelInfo(BaseModel):
     max_output_tokens: Optional[int] = Field(alias="maxOutputTokens")
     min_fl_version: Optional[str] = Field(alias="minFLVersion")
     created_at_unix: int = Field(alias="createdAt")
+    context_length: Optional[int] = Field(alias="contextLength")
+    input_modalities: Optional[str] = Field(alias="inputModalities")
+    output_modalities: Optional[str] = Field(alias="outputModalities")
+    capabilities: Optional[str] = Field(alias="capabilities")

sdk/python/src/imodel.py

@@ -37,6 +37,36 @@ def is_loaded(self) -> bool:
         """True if the model is loaded into memory."""
         pass
 
+    @property
+    @abstractmethod
+    def context_length(self) -> Optional[int]:
+        """Maximum context length (in tokens) supported by the model, or ``None`` if unknown."""
+        pass
+
+    @property
+    @abstractmethod
+    def input_modalities(self) -> Optional[str]:
+        """Comma-separated input modalities (e.g. ``"text,image"``), or ``None`` if unknown."""
+        pass
+
+    @property
+    @abstractmethod
+    def output_modalities(self) -> Optional[str]:
+        """Comma-separated output modalities (e.g. ``"text"``), or ``None`` if unknown."""
+        pass
+
+    @property
+    @abstractmethod
+    def capabilities(self) -> Optional[str]:
+        """Comma-separated capability tags (e.g. ``"chat,completion"``), or ``None`` if unknown."""
+        pass
+
+    @property
+    @abstractmethod
+    def supports_tool_calling(self) -> Optional[bool]:
+        """Whether the model supports tool/function calling, or ``None`` if unknown."""
+        pass
+
     @abstractmethod
     def download(self, progress_callback: Callable[[float], None] = None) -> None:
         """

sdk/python/src/model.py

@@ -94,6 +94,31 @@ def alias(self) -> str:
         """Alias of this model."""
         return self._alias
 
+    @property
+    def context_length(self) -> Optional[int]:
+        """Maximum context length (in tokens) of the currently selected variant."""
+        return self._selected_variant.context_length
+
+    @property
+    def input_modalities(self) -> Optional[str]:
+        """Comma-separated input modalities of the currently selected variant."""
+        return self._selected_variant.input_modalities
+
+    @property
+    def output_modalities(self) -> Optional[str]:
+        """Comma-separated output modalities of the currently selected variant."""
+        return self._selected_variant.output_modalities
+
+    @property
+    def capabilities(self) -> Optional[str]:
+        """Comma-separated capability tags of the currently selected variant."""
+        return self._selected_variant.capabilities
+
+    @property
+    def supports_tool_calling(self) -> Optional[bool]:
+        """Whether the currently selected variant supports tool/function calling."""
+        return self._selected_variant.supports_tool_calling
+
     @property
     def is_cached(self) -> bool:
         """Is the currently selected variant cached locally?"""

sdk/python/src/model_variant.py

@@ -57,6 +57,31 @@ def info(self) -> ModelInfo:
         """Full catalog metadata for this variant."""
         return self._model_info
 
+    @property
+    def context_length(self) -> Optional[int]:
+        """Maximum context length (in tokens) supported by this variant, or ``None`` if unknown."""
+        return self._model_info.context_length
+
+    @property
+    def input_modalities(self) -> Optional[str]:
+        """Comma-separated input modalities (e.g. ``"text,image"``), or ``None`` if unknown."""
+        return self._model_info.input_modalities
+
+    @property
+    def output_modalities(self) -> Optional[str]:
+        """Comma-separated output modalities (e.g. ``"text"``), or ``None`` if unknown."""
+        return self._model_info.output_modalities
+
+    @property
+    def capabilities(self) -> Optional[str]:
+        """Comma-separated capability tags (e.g. ``"chat,completion"``), or ``None`` if unknown."""
+        return self._model_info.capabilities
+
+    @property
+    def supports_tool_calling(self) -> Optional[bool]:
+        """Whether this variant supports tool/function calling, or ``None`` if unknown."""
+        return self._model_info.supports_tool_calling
+
     @property
     def is_cached(self) -> bool:
         """``True`` if this variant is present in the local model cache."""

sdk/python/test/test_model.py

@@ -56,3 +56,33 @@ def test_should_load_and_unload_model(self, catalog):
             # Safety cleanup
             if model.is_loaded:
                 model.unload()
+
+    def test_should_expose_context_length(self, catalog):
+        """Model should expose context_length from ModelInfo metadata."""
+        model = catalog.get_model(TEST_MODEL_ALIAS)
+        assert model is not None
+        # context_length should be None or a positive integer
+        ctx = model.context_length
+        assert ctx is None or (isinstance(ctx, int) and ctx > 0)
+
+    def test_should_expose_modalities(self, catalog):
+        """Model should expose input_modalities and output_modalities."""
+        model = catalog.get_model(TEST_MODEL_ALIAS)
+        assert model is not None
+        # Modalities should be None or non-empty strings
+        for val in (model.input_modalities, model.output_modalities):
+            assert val is None or (isinstance(val, str) and len(val) > 0)
+
+    def test_should_expose_capabilities(self, catalog):
+        """Model should expose capabilities metadata."""
+        model = catalog.get_model(TEST_MODEL_ALIAS)
+        assert model is not None
+        caps = model.capabilities
+        assert caps is None or (isinstance(caps, str) and len(caps) > 0)
+
+    def test_should_expose_supports_tool_calling(self, catalog):
+        """Model should expose supports_tool_calling metadata."""
+        model = catalog.get_model(TEST_MODEL_ALIAS)
+        assert model is not None
+        stc = model.supports_tool_calling
+        assert stc is None or isinstance(stc, bool)