fix list of models

Author: deep1401

api/transformerlab/routers/model.py

@@ -38,6 +38,45 @@ async def model_local_list():
     return models
 
 
+@router.get("/model/registry_versions")
+@cached(key="models:registry_versions", ttl="1h", tags=["models", "models:registry_versions", "asset_versions"])
+async def model_registry_versions():
+    """List model registry versions as selectable model IDs."""
+    from transformerlab.services import asset_version_service
+
+    groups = await asset_version_service.list_groups("model")
+    versions: list[dict] = []
+
+    for group in groups:
+        group_id = group.get("group_id")
+        group_name = group.get("group_name") or group_id
+        if not group_id:
+            continue
+
+        group_versions = await asset_version_service.list_versions("model", group_id)
+        for version in group_versions:
+            model_id = version.get("asset_id")
+            version_label = version.get("version_label")
+            if not model_id or not version_label:
+                continue
+
+            tag = version.get("tag")
+            tag_suffix = f" ({tag})" if tag else ""
+            versions.append(
+                {
+                    "model_id": model_id,
+                    "name": f"{group_name}/{version_label}{tag_suffix}",
+                    "group_id": group_id,
+                    "group_name": group_name,
+                    "version_label": version_label,
+                    "tag": tag,
+                    "created_at": version.get("created_at"),
+                }
+            )
+
+    return versions
+
+
 @router.get("/model/create")
 async def model_local_create(id: str, name: str, json_data={}):
     # Use filesystem instead of database

src/renderer/components/Experiment/Tasks/EditInteractiveTaskModal.tsx

@@ -38,7 +38,7 @@ type ProviderOption = {
 type ConfigField = {
   field_name: string;
   env_var: string;
-  field_type: 'str' | 'integer';
+  field_type: 'str' | 'integer' | 'model';
   required?: boolean;
   placeholder?: string;
   help_text?: string;
@@ -399,11 +399,18 @@ export default function EditInteractiveTaskModal({
         body.provider_name = provider.name;
       }
 
-      // Persist model name to history before saving
-      const modelName = configFieldValues['MODEL_NAME'];
-      if (modelName?.trim() && (interactiveType || galleryId)) {
-        const taskTypeOrId = interactiveType || galleryId;
-        saveModelToHistory(getModelHistoryKey(taskTypeOrId), modelName.trim());
+      // Persist all model-type fields to history before saving
+      const taskTypeOrId = interactiveType || galleryId;
+      if (taskTypeOrId) {
+        const historyKey = getModelHistoryKey(taskTypeOrId);
+        templateConfigFields
+          .filter((field) => field.field_type === 'model')
+          .forEach((field) => {
+            const modelValue = configFieldValues[field.env_var];
+            if (modelValue?.trim()) {
+              saveModelToHistory(historyKey, modelValue.trim());
+            }
+          });
       }
 
       // The caller is responsible for actually persisting the changes via API.
@@ -542,20 +549,20 @@ export default function EditInteractiveTaskModal({
                 <>
                   {templateConfigFields.map((field) => {
                     const isNgrokField = field.env_var === 'NGROK_AUTH_TOKEN';
-                    const isModelNameField = field.env_var === 'MODEL_NAME';
+                    const isModelField = field.field_type === 'model';
                     return (
                       <FormControl
                         key={field.env_var}
                         required={field.required && !isNgrokField}
                       >
                         <FormLabel>{field.field_name}</FormLabel>
-                        {isModelNameField ? (
+                        {isModelField ? (
                           <ModelNameInput
                             value={configFieldValues[field.env_var] || ''}
                             onChange={(v) =>
                               handleConfigFieldChange(field.env_var, v)
                             }
-                            taskTypeOrId={interactiveType}
+                            taskTypeOrId={interactiveType || galleryId}
                             placeholder={field.placeholder}
                             disabled={false}
                             required={!!field.required}

src/renderer/components/Experiment/Tasks/NewInteractiveTaskModal.tsx

@@ -234,12 +234,23 @@ export default function NewInteractiveTaskModal({
     open && step === 'config' ? chatAPI.Endpoints.Models.LocalList() : null,
     fetcher,
   );
+  const { data: registryModelsData } = useSWR(
+    open && step === 'config'
+      ? chatAPI.Endpoints.Models.RegistryVersionList()
+      : null,
+    fetcher,
+  );
 
   const installedModels = React.useMemo(() => {
     if (!installedModelsData || !Array.isArray(installedModelsData)) return [];
     return installedModelsData as Array<{ model_id: string; name?: string }>;
   }, [installedModelsData]);
 
+  const registryModels = React.useMemo(() => {
+    if (!registryModelsData || !Array.isArray(registryModelsData)) return [];
+    return registryModelsData as Array<{ model_id: string; name?: string }>;
+  }, [registryModelsData]);
+
   React.useEffect(() => {
     if (!open) {
       setStep('provider');
@@ -737,7 +748,10 @@ export default function NewInteractiveTaskModal({
                                       listbox: { sx: { maxHeight: 240 } },
                                     }}
                                   >
-                                    {installedModels.map((m) => (
+                                    {(registryModels.length > 0
+                                      ? registryModels
+                                      : installedModels
+                                    ).map((m) => (
                                       <Option
                                         key={m.model_id}
                                         value={m.model_id}

src/renderer/lib/api-client/endpoints.ts

@@ -179,6 +179,7 @@ Endpoints.Dataset = {
 
 Endpoints.Models = {
   LocalList: () => `${API_URL()}model/list`,
+  RegistryVersionList: () => `${API_URL()}model/registry_versions`,
   ModelGroups: () => `${API_URL()}model/model_groups_list`,
   GetPeftsForModel: () => `${API_URL()}model/pefts`,
   DeletePeft: (modelId: string, peft: string) =>