track the org jobs

ui/src/app/api/hf-jobs/route.ts

@@ -16,7 +16,8 @@ export async function POST(request: NextRequest) {
             return NextResponse.json({ error: 'Token and job ID required' }, { status: 400 });
           }
 
-          const jobStatus = await checkHFJobStatus(token, jobConfig.hf_job_id);
+          const jobNamespaceOverride = jobConfig?.hf_job_namespace;
+          const jobStatus = await checkHFJobStatus(token, jobConfig.hf_job_id, jobNamespaceOverride);
           return NextResponse.json({ status: jobStatus });
         } catch (error: any) {
           console.error('Job status check error:', error);
@@ -904,13 +905,19 @@ async function submitHFJobUV(token: string, hardware: string, scriptPath: string
   });
 }
 
-async function checkHFJobStatus(token: string, jobId: string): Promise<any> {
+async function checkHFJobStatus(token: string, jobId: string, jobNamespace?: string): Promise<any> {
   return new Promise((resolve, reject) => {
     console.log(`Checking HF Job status for: ${jobId}`);
-    
-    const childProcess = spawn('hf', [
-      'jobs', 'inspect', jobId
-    ], {
+    const args = ['jobs', 'inspect'];
+
+    if (jobNamespace) {
+      console.log(`Using namespace override for status check: ${jobNamespace}`);
+      args.push('--namespace', jobNamespace);
+    }
+
+    args.push(jobId);
+
+    const childProcess = spawn('hf', args, {
       env: { 
         ...process.env, 
         HF_TOKEN: token 

ui/src/components/HFJobStatus.tsx

@@ -7,10 +7,25 @@ import { Button } from '@headlessui/react';
 interface HFJobStatusProps {
   hfJobId: string;
   hfJobUrl?: string;
+  hfJobNamespace?: string;
 }
 
-export default function HFJobStatus({ hfJobId, hfJobUrl }: HFJobStatusProps) {
-  const { status, loading, error } = useHFJobStatus(hfJobId);
+const extractNamespaceFromUrl = (url?: string) => {
+  if (!url) {
+    return undefined;
+  }
+  try {
+    const match = url.match(/\/jobs\/([^\/]+)\//);
+    return match?.[1];
+  } catch (error) {
+    console.warn('Failed to derive HF job namespace from URL:', error);
+    return undefined;
+  }
+};
+
+export default function HFJobStatus({ hfJobId, hfJobUrl, hfJobNamespace }: HFJobStatusProps) {
+  const derivedNamespace = hfJobNamespace || extractNamespaceFromUrl(hfJobUrl);
+  const { status, loading, error } = useHFJobStatus(hfJobId, derivedNamespace);
 
   if (error) {
     return (
@@ -122,4 +137,4 @@ export default function HFJobStatus({ hfJobId, hfJobUrl }: HFJobStatusProps) {
       )}
     </div>
   );
-}
+}

ui/src/components/HFJobsWorkflow.tsx

@@ -409,6 +409,7 @@ export default function HFJobsWorkflow({ jobConfig, onComplete, hackathonEligibl
               is_hf_job: true,
               training_backend: 'hf-jobs',
               hf_job_submitted: true,
+              hf_job_namespace: jobNamespace,
             },
             info: response.data.message || 'HF Job submitted',
             status: 'submitted',
@@ -437,6 +438,7 @@ export default function HFJobsWorkflow({ jobConfig, onComplete, hackathonEligibl
                   is_hf_job: true,
                   training_backend: 'hf-jobs',
                   hf_job_submitted: true,
+                  hf_job_namespace: jobNamespace,
                 },
                 info: response.data.message || 'HF Job submitted',
                 status: 'submitted',

ui/src/components/JobOverview.tsx

@@ -109,6 +109,7 @@ export default function JobOverview({ job }: JobOverviewProps) {
             <HFJobStatus 
               hfJobId={jobConfig.hf_job_id} 
               hfJobUrl={jobConfig.hf_job_url} 
+              hfJobNamespace={jobConfig.hf_job_namespace}
             />
           ) : isHFJob && !hfJobSubmitted ? (
             <span className="px-3 py-1 rounded-full text-sm bg-yellow-500/10 text-yellow-500">

ui/src/components/JobsTable.tsx

@@ -99,6 +99,7 @@ export default function JobsTable({ onlyActive = false }: JobsTableProps) {
               <HFJobStatus 
                 hfJobId={jobConfig.hf_job_id} 
                 hfJobUrl={jobConfig.hf_job_url} 
+                hfJobNamespace={jobConfig.hf_job_namespace}
               />
             );
           } else {

ui/src/hooks/useHFJobStatus.tsx

@@ -12,7 +12,7 @@ interface HFJobStatus {
   url: string;
 }
 
-export function useHFJobStatus(hfJobId: string | null, refreshInterval = 30000) {
+export function useHFJobStatus(hfJobId: string | null, hfJobNamespace?: string, refreshInterval = 30000) {
   const [status, setStatus] = useState<HFJobStatus | null>(null);
   const [loading, setLoading] = useState(false);
   const [error, setError] = useState<string | null>(null);
@@ -31,7 +31,10 @@ export function useHFJobStatus(hfJobId: string | null, refreshInterval = 30000)
         const response = await apiClient.post('/api/hf-jobs', {
           action: 'checkStatus',
           token,
-          jobConfig: { hf_job_id: hfJobId },
+          jobConfig: {
+            hf_job_id: hfJobId,
+            hf_job_namespace: hfJobNamespace,
+          },
         });
 
         if (response.data.status) {
@@ -52,7 +55,7 @@ export function useHFJobStatus(hfJobId: string | null, refreshInterval = 30000)
     const interval = setInterval(fetchStatus, refreshInterval);
 
     return () => clearInterval(interval);
-  }, [hfJobId, token, refreshInterval]);
+  }, [hfJobId, hfJobNamespace, token, refreshInterval]);
 
   return { status, loading, error, refetch: () => {
     if (hfJobId && token) {

ui/src/types.ts

@@ -199,6 +199,14 @@ export interface JobConfig {
   job: string;
   config: ConfigObject;
   meta: MetaConfig;
+  is_hf_job?: boolean;
+  hf_job_id?: string;
+  hf_job_url?: string | null;
+  hf_job_namespace?: string;
+  dataset_repo?: string;
+  hardware?: string;
+  training_backend?: string;
+  hf_job_submitted?: boolean;
 }
 
 export interface ConfigDoc {