// src/vastai/client.ts
import { Vast.aiClient } from '@vastai/sdk';

let instance: Vast.aiClient | null = null;

export function getVast.aiClient(): Vast.aiClient {
  if (!instance) {
    instance = new Vast.aiClient({
      apiKey: process.env.VASTAI_API_KEY!,
      // Additional options
    });
  }
  return instance;
}

import { Vast.aiError } from '@vastai/sdk';

async function safeVast.aiCall<T>(
  operation: () => Promise<T>
): Promise<{ data: T | null; error: Error | null }> {
  try {
    const data = await operation();
    return { data, error: null };
  } catch (err) {
    if (err instanceof Vast.aiError) {
      console.error({
        code: err.code,
        message: err.message,
      });
    }
    return { data: null, error: err as Error };
  }
}

async function withRetry<T>(
  operation: () => Promise<T>,
  maxRetries = 3,
  backoffMs = 1000
): Promise<T> {
  for (let attempt = 1; attempt <= maxRetries; attempt++) {
    try {
      return await operation();
    } catch (err) {
      if (attempt === maxRetries) throw err;
      const delay = backoffMs * Math.pow(2, attempt - 1);
      await new Promise(r => setTimeout(r, delay));
    }
  }
  throw new Error('Unreachable');
}

const clients = new Map<string, Vast.aiClient>();

export function getClientForTenant(tenantId: string): Vast.aiClient {
  if (!clients.has(tenantId)) {
    const apiKey = getTenantApiKey(tenantId);
    clients.set(tenantId, new Vast.aiClient({ apiKey }));
  }
  return clients.get(tenantId)!;
}

from contextlib import asynccontextmanager
from vastai import Vast.aiClient

@asynccontextmanager
async def get_vastai_client():
    client = Vast.aiClient()
    try:
        yield client
    finally:
        await client.close()

import { z } from 'zod';

const vastaiResponseSchema = z.object({
  id: z.string(),
  status: z.enum(['active', 'inactive']),
  createdAt: z.string().datetime(),
});