error is throw when invalid model type is provided

Author: srebrek

src/MaIN.Core/Hub/Contexts/AgentContext.cs

@@ -210,11 +210,7 @@ public async Task<IAgentContextExecutor> CreateAsync(bool flow = false, bool int
     {
         if (_ensureModelDownloaded && !string.IsNullOrWhiteSpace(_agent.Model))
         {
-            var model = ModelRegistry.GetById(_agent.Model);
-            if (model is LocalModel)
-            {
-                await AIHub.Model().EnsureDownloadedAsync(_agent.Model);
-            }
+            await AIHub.Model().EnsureDownloadedAsync(_agent.Model);
         }
 
         await _agentService.CreateAgent(_agent, flow, interactiveResponse, _inferenceParams, _memoryParams, _disableCache);

src/MaIN.Core/Hub/Contexts/ChatContext.cs

@@ -216,7 +216,7 @@ public async Task<ChatResult> CompleteAsync(
             throw new EmptyChatException(_chat.Id);
         }
 
-        if (_ensureModelDownloaded && _chat.ModelInstance is LocalModel)
+        if (_ensureModelDownloaded)
         {
             await AIHub.Model().EnsureDownloadedAsync(_chat.ModelId);
         }

src/MaIN.Core/Hub/Contexts/ModelContext.cs

@@ -62,7 +62,12 @@ public async Task<IModelContext> EnsureDownloadedAsync(string modelId, Cancellat
 
         var model = ModelRegistry.GetById(modelId);
 
-        if (model is not LocalModel localModel || localModel.IsDownloaded(_defaultModelsPath))
+        if (model is not LocalModel localModel)
+        {
+            throw new InvalidModelTypeException(nameof(LocalModel));
+        }
+
+        if (localModel.IsDownloaded(_defaultModelsPath))
         {
             return this;
         }