@fastino-ai/pioneer-cli 0.2.1 → 0.2.3

package/src/config.ts

@@ -7,82 +7,40 @@ import fs from "fs";
 import os from "os";
 import path from "path";
 
-export interface AgentProviderConfig {
-  provider: "anthropic" | "openai" | "local";
-  model: string;
-  apiKey?: string;
-  baseUrl?: string;
-}
-
-export interface BudgetConfig {
-  maxTokens?: number;
-  maxCost?: number; // in USD
-  maxTime?: number; // in seconds
-  maxIterations?: number;
-}
-
-export interface SandboxConfig {
-  useDocker?: boolean;
-  dockerImage?: string;
-  timeout?: number;
-  memoryLimit?: string;
-  cpuLimit?: number;
-}
-
-export interface MLConfig {
-  modal?: {
-    tokenId?: string;
-    tokenSecret?: string;
-  };
-  wandb?: {
-    apiKey?: string;
-    entity?: string;
-    project?: string;
-  };
-}
-
-export interface EvolutionConfigOptions {
+export interface TelemetryConfig {
   enabled?: boolean;
-  targetScore?: number;
-  maxIterations?: number;
-  budgetPerIteration?: BudgetConfig;
-  trainingProvider?: "openai" | "modal" | "local";
-  trainingBaseModel?: string;
+  anonymousId?: string;
 }
 
 export interface Config {
-  // Existing Pioneer config
+  // Pioneer API configuration
   apiKey?: string;
   baseUrl?: string;
 
-  // Agent config
-  agent?: AgentProviderConfig;
-  budget?: BudgetConfig;
-  sandbox?: SandboxConfig;
-  ml?: MLConfig;
-  evolution?: EvolutionConfigOptions;
+  // MLE Agent model selection (for Pioneer server)
+  mleModel?: string;
 
-  // System prompt customization
-  systemPrompt?: string;
+  // Hugging Face token for pushing datasets/models
+  hfToken?: string;
+
+  // Telemetry (opt-in analytics)
+  telemetry?: TelemetryConfig;
 }
 
 const CONFIG_DIR = path.join(os.homedir(), ".pioneer");
 const CONFIG_FILE = path.join(CONFIG_DIR, "config.json");
 
 export const DEFAULT_BASE_URL =
-  process.env.PIONEER_API_URL ?? "http://localhost:5001";
+  process.env.PIONEER_API_URL ?? "https://api.fastino.ai";
 
-export const DEFAULT_AGENT_CONFIG: AgentProviderConfig = {
-  provider: "anthropic",
-  model: "claude-sonnet-4-5-20250929",
-};
+// Available MLE models for server-side agent
+export const AVAILABLE_MLE_MODELS = [
+  { id: "claude-sonnet-4-5-20250929", name: "Claude Sonnet 4.5", description: "Balanced (default)" },
+  { id: "claude-haiku-4-5-20251001", name: "Claude Haiku 4.5", description: "Fast & cheap" },
+  { id: "claude-opus-4-5-20251101", name: "Claude Opus 4.5", description: "Most capable" },
+] as const;
 
-export const DEFAULT_BUDGET: BudgetConfig = {
-  maxTokens: 500000,  // 500k tokens
-  maxCost: 5.0,       // $5 USD
-  maxTime: 7200,      // 2 hours
-  maxIterations: 100,
-};
+export const DEFAULT_MLE_MODEL = "claude-sonnet-4-5-20250929";
 
 function ensureConfigDir(): void {
   if (!fs.existsSync(CONFIG_DIR)) {
@@ -149,98 +107,50 @@ export function getApiKey(): string | undefined {
 }
 
 export function getBaseUrl(): string {
+  // Environment variable takes precedence
+  if (process.env.PIONEER_API_URL) {
+    return process.env.PIONEER_API_URL;
+  }
   const config = loadConfig();
   return config.baseUrl ?? DEFAULT_BASE_URL;
 }
 
-// Agent configuration helpers
-export function getAgentConfig(): AgentProviderConfig {
-  const config = loadConfig();
-
-  // Check environment variables for API keys
-  const anthropicKey = process.env.ANTHROPIC_API_KEY;
-  const openaiKey = process.env.OPENAI_API_KEY;
-
-  if (config.agent) {
-    return {
-      ...config.agent,
-      apiKey: config.agent.apiKey ||
-        (config.agent.provider === "anthropic" ? anthropicKey : openaiKey),
-    };
-  }
-
-  // Default to Anthropic if key is available
-  if (anthropicKey) {
-    return {
-      ...DEFAULT_AGENT_CONFIG,
-      apiKey: anthropicKey,
-    };
-  }
-
-  // Fall back to OpenAI
-  if (openaiKey) {
-    return {
-      provider: "openai",
-      model: "gpt-4o",
-      apiKey: openaiKey,
-    };
-  }
-
-  return DEFAULT_AGENT_CONFIG;
-}
-
-export function getBudgetConfig(): BudgetConfig {
-  const config = loadConfig();
-  return {
-    ...DEFAULT_BUDGET,
-    ...config.budget,
-  };
+export function getMleModel(): string | undefined {
+  return loadConfig().mleModel;
 }
 
-export function getSandboxConfig(): SandboxConfig {
-  const config = loadConfig();
-  return {
-    useDocker: false,
-    dockerImage: "python:3.11-slim",
-    timeout: 30000,
-    memoryLimit: "512m",
-    cpuLimit: 1,
-    ...config.sandbox,
-  };
+export function setMleModel(model: string): void {
+  saveConfig({ mleModel: model });
 }
 
-export function getMLConfig(): MLConfig {
-  const config = loadConfig();
-  return {
-    modal: {
-      tokenId: process.env.MODAL_TOKEN_ID,
-      tokenSecret: process.env.MODAL_TOKEN_SECRET,
-      ...config.ml?.modal,
-    },
-    wandb: {
-      apiKey: process.env.WANDB_API_KEY,
-      ...config.ml?.wandb,
-    },
-  };
+export function getConfigDir(): string {
+  ensureConfigDir();
+  return CONFIG_DIR;
 }
 
-export function getEvolutionConfig(): EvolutionConfigOptions {
-  const config = loadConfig();
-  return {
-    enabled: false,
-    targetScore: 0.9,
-    maxIterations: 10,
-    trainingProvider: "local",
-    ...config.evolution,
-  };
+/**
+ * Get Hugging Face token with priority:
+ * 1. Explicit token passed as parameter (from --hf-token flag)
+ * 2. HF_TOKEN environment variable
+ * 3. Token stored in config file
+ */
+export function getHfToken(explicitToken?: string): string | undefined {
+  if (explicitToken) {
+    return explicitToken;
+  }
+  if (process.env.HF_TOKEN) {
+    return process.env.HF_TOKEN;
+  }
+  return loadConfig().hfToken;
 }
 
-export function getSystemPrompt(): string | undefined {
-  return loadConfig().systemPrompt;
+export function setHfToken(token: string): void {
+  saveConfig({ hfToken: token });
 }
 
-// Get config directory path
-export function getConfigDir(): string {
+export function clearHfToken(): void {
+  const config = loadConfig();
+  delete config.hfToken;
   ensureConfigDir();
-  return CONFIG_DIR;
+  fs.writeFileSync(CONFIG_FILE, JSON.stringify(config, null, 2));
 }