Commit 31a50c6e authored by Your Name's avatar Your Name

Add image_model property to MultiModelManager

parent 2a5f2bf6
...@@ -381,6 +381,11 @@ class MultiModelManager: ...@@ -381,6 +381,11 @@ class MultiModelManager:
self.tts_model: Optional[str] = None self.tts_model: Optional[str] = None
self.image_models: List[str] = [] self.image_models: List[str] = []
self.vision_models: List[str] = [] self.vision_models: List[str] = []
@property
def image_model(self) -> Optional[str]:
"""Return the first image model or None."""
return self.image_models[0] if self.image_models else None
self.tool_parser = ModelParserAdapter() self.tool_parser = ModelParserAdapter()
self.current_model_key: Optional[str] = None self.current_model_key: Optional[str] = None
self.config: Dict[str, Dict] = {} self.config: Dict[str, Dict] = {}
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment