model injection working

Author: Jovonni

api/src/models/dataset.rs

@@ -6,7 +6,7 @@ use uuid::Uuid;
 #[derive(Debug, Clone, Serialize, Deserialize, FromRow)]
 pub struct Dataset {
     pub id: Uuid,
-    pub project_id: Uuid,
+    pub project_id: Option<Uuid>,
     pub name: String,
     pub description: Option<String>,
     pub format: String,
@@ -45,7 +45,7 @@ pub struct UploadUrlResponse {
 #[derive(Debug, Clone, Serialize, Deserialize, FromRow)]
 pub struct DataSource {
     pub id: Uuid,
-    pub project_id: Uuid,
+    pub project_id: Option<Uuid>,
     pub name: String,
     pub source_type: String,
     pub connection_string: Option<String>,

api/src/models/model.rs

@@ -6,7 +6,7 @@ use uuid::Uuid;
 #[derive(Debug, Clone, Serialize, Deserialize, FromRow)]
 pub struct Model {
     pub id: Uuid,
-    pub project_id: Uuid,
+    pub project_id: Option<Uuid>,
     pub name: String,
     pub description: Option<String>,
     pub framework: String,

api/src/models/pipeline.rs

@@ -6,7 +6,7 @@ use uuid::Uuid;
 #[derive(Debug, Clone, Serialize, Deserialize, FromRow)]
 pub struct Pipeline {
     pub id: Uuid,
-    pub project_id: Uuid,
+    pub project_id: Option<Uuid>,
     pub name: String,
     pub description: Option<String>,
     pub config: serde_json::Value,

api/src/routes/sdk.rs

@@ -1374,7 +1374,7 @@ pub async fn create_sweep(
     Json(req): Json<CreateSweepRequest>,
 ) -> AppResult<Json<serde_json::Value>> {
     let model = resolve_model_id(&state.db, &req.model_id).await?;
-    let project_id = req.project_id.unwrap_or(model.project_id);
+    let project_id = req.project_id.or(model.project_id);
 
     let dataset_id = if let Some(ref ds) = req.dataset_id {
         Some(resolve_dataset_id(&state.db, ds).await?)

sdk/python/openmodelstudio/registry.py

@@ -154,6 +154,14 @@ def registry_install(name: str, registry_url: str = None, models_dir: str = None
     _api_url = api_url or os.environ.get("OPENMODELSTUDIO_API_URL") or _load_api_url()
     _token = token or os.environ.get("OPENMODELSTUDIO_TOKEN")
 
+    # Auto-detect local platform if no api_url is configured
+    if not _api_url:
+        _api_url = _auto_detect_api_url()
+
+    # Auto-login if we have an api_url but no token
+    if _api_url and not _token:
+        _token = _auto_login(_api_url)
+
     if _api_url:
         try:
             main_file = files[0]
@@ -196,6 +204,32 @@ def _load_api_url() -> str:
         return ""
 
 
+def _auto_detect_api_url() -> str:
+    """Auto-detect local platform API (K8s NodePort at localhost:31001)."""
+    try:
+        resp = requests.get("http://localhost:31001/healthz", timeout=2)
+        if resp.ok:
+            return "http://localhost:31001"
+    except Exception:
+        pass
+    return ""
+
+
+def _auto_login(api_url: str) -> str:
+    """Auto-login with default credentials to get a token for registration."""
+    try:
+        resp = requests.post(
+            f"{api_url}/auth/login",
+            json={"email": "test@openmodel.studio", "password": "Test1234"},
+            timeout=10,
+        )
+        if resp.ok:
+            return resp.json().get("access_token", "")
+    except Exception:
+        pass
+    return ""
+
+
 def registry_uninstall(name: str, models_dir: str = None,
                        api_url: str = None, token: str = None) -> bool:
     """Uninstall a locally installed model.
@@ -222,6 +256,10 @@ def registry_uninstall(name: str, models_dir: str = None,
     # Also unregister from platform API
     _api_url = api_url or os.environ.get("OPENMODELSTUDIO_API_URL") or _load_api_url()
     _token = token or os.environ.get("OPENMODELSTUDIO_TOKEN")
+    if not _api_url:
+        _api_url = _auto_detect_api_url()
+    if _api_url and not _token:
+        _token = _auto_login(_api_url)
     if _api_url:
         try:
             headers = {"Content-Type": "application/json"}

web/src/app/registry/[id]/page.tsx

@@ -14,6 +14,7 @@ import {
   Package,
   ArrowLeft,
   Download,
+  Trash2,
   User,
   Tag,
   ExternalLink,
@@ -219,6 +220,16 @@ export default function RegistryModelDetailPage() {
     }
   };
 
+  const handleUninstall = async () => {
+    try {
+      await api.post("/models/registry-uninstall", { name: modelName });
+      toast.success(`Uninstalled ${modelName}`);
+      setIsInstalled(false);
+    } catch (err) {
+      toast.error(err instanceof Error ? err.message : "Failed to uninstall");
+    }
+  };
+
   const handleCopyInstall = () => {
     navigator.clipboard.writeText(`openmodelstudio install ${modelName}`);
     setCopied(true);
@@ -294,10 +305,14 @@ export default function RegistryModelDetailPage() {
               <Badge variant="secondary" className="bg-muted text-[10px]">
                 v{model.version}
               </Badge>
-              {isInstalled && (
+              {isInstalled ? (
                 <Badge className="bg-emerald-500/10 text-emerald-400 border-emerald-500/20 text-[10px]">
                   Installed
                 </Badge>
+              ) : (
+                <Badge variant="outline" className="text-muted-foreground/50 text-[10px]">
+                  Not Installed
+                </Badge>
               )}
             </div>
             <div className="flex items-center gap-3 mt-1">
@@ -324,12 +339,23 @@ export default function RegistryModelDetailPage() {
                 </Button>
               </motion.div>
             )}
+            {isInstalled && (
+              <motion.div whileHover={{ scale: 1.05 }} whileTap={{ scale: 0.97 }}>
+                <Button
+                  variant="outline"
+                  className="gap-2 border-red-500/30 text-red-400 hover:bg-red-500/10 hover:text-red-300"
+                  onClick={handleUninstall}
+                >
+                  <Trash2 className="h-4 w-4" /> Uninstall
+                </Button>
+              </motion.div>
+            )}
             <motion.div whileHover={{ scale: 1.05 }} whileTap={{ scale: 0.97 }}>
               <Button
                 className="gap-2 bg-white text-black hover:bg-white/90 shadow-lg shadow-white/10"
                 onClick={() => setInstallOpen(true)}
               >
-                <Download className="h-4 w-4" /> {isInstalled ? "Install to Project" : "Install"}
+                <Download className="h-4 w-4" /> {isInstalled ? "Reinstall" : "Install"}
               </Button>
             </motion.div>
           </div>