@fastino-ai/pioneer-cli 0.2.7 → 0.2.9

package/src/api.ts

@@ -248,7 +248,8 @@ export interface UploadUrlRequest {
 }
 
 export interface UploadUrlResponse {
-  upload_url: string;
+  upload_url?: string;
+  presigned_url?: string;
   dataset_id: string;
   key: string;
   expires_in: number;
@@ -351,7 +352,7 @@ export async function uploadDataset(
       dataset_name: options.dataset_name,
       dataset_type: options.dataset_type,
       format,
-      content_type: format === "parquet" ? "application/octet-stream" : "text/plain",
+      content_type: "application/octet-stream",
     });
 
     if (!urlResult.ok || !urlResult.data) {
@@ -362,13 +363,21 @@ export async function uploadDataset(
       };
     }
 
-    const { upload_url, dataset_id } = urlResult.data;
+    const { dataset_id } = urlResult.data;
+    const uploadUrl = urlResult.data.upload_url || urlResult.data.presigned_url;
+    if (!uploadUrl) {
+      return {
+        ok: false,
+        status: 500,
+        error: "Upload URL not provided by backend",
+      };
+    }
 
     // Step 2: Upload file directly to S3 via presigned URL
-    const uploadRes = await fetch(upload_url, {
+    const uploadRes = await fetch(uploadUrl, {
       method: "PUT",
       headers: {
-        "Content-Type": format === "parquet" ? "application/octet-stream" : "text/plain",
+        "Content-Type": "application/octet-stream",
       },
       body: fileContent,
     });
@@ -587,6 +596,21 @@ export async function analyzeDataset(
   });
 }
 
+export interface DatasetLLMAnalysisRequest {
+  task_type: "ner" | "classification" | "generative";
+  task_description?: string;
+  labels?: string[];
+  dataset?: Record<string, unknown>[];
+  dataset_name?: string;
+  dataset_version?: string;
+}
+
+export async function analyzeDatasetLLM(
+  req: DatasetLLMAnalysisRequest
+): Promise<ApiResult<DiversityLLMAnalysis>> {
+  return request<DiversityLLMAnalysis>("POST", "/felix/dataset/analyze_llm", req);
+}
+
 // ─────────────────────────────────────────────────────────────────────────────
 // Dataset Generation
 // ─────────────────────────────────────────────────────────────────────────────
@@ -650,6 +674,7 @@ export interface GenerateRequest {
   domain_description?: string;
   temperature?: number;
   quality?: "light" | "medium" | "heavy";
+  generation_profile?: "auto" | "fast" | "balanced" | "quality";
   session_id?: string;
   config_num_examples?: number;
   seed?: number;
@@ -668,6 +693,8 @@ export interface GenerateRequest {
   instruction?: string;
   constraints?: ConstraintRequest[];
   multiplicator?: Record<string, unknown>;
+  include_reasoning_trace?: boolean;
+  reasoning_effort?: "low" | "medium" | "high";
   use_meta_felix?: boolean;
   min_criteria?: number;
   target_choices?: number;
@@ -791,10 +818,19 @@ export interface GenerateNERRequest {
   save_dataset?: boolean;
   dataset_name?: string;
   project_id?: string;
+  generation_profile?: "auto" | "fast" | "balanced" | "quality";
+  include_reasoning_trace?: boolean;
+  reasoning_effort?: "low" | "medium" | "high";
   quality?: "light" | "medium" | "heavy";
   type?: "training" | "evaluation" | "split";
   visibility?: "private" | "public";
   negative_ratio?: number;
+  split_ratio?: Record<string, number>;
+  multiplicator?: Record<string, unknown>;
+  use_meta_felix?: boolean;
+  min_criteria?: number;
+  target_choices?: number;
+  classified_examples?: Record<string, unknown>[];
 }
 
 export async function generateNER(
@@ -812,10 +848,19 @@ export async function generateNER(
     temperature: req.temperature,
     constraints: req.constraints,
     project_id: req.project_id,
+    generation_profile: req.generation_profile,
+    include_reasoning_trace: req.include_reasoning_trace,
+    reasoning_effort: req.reasoning_effort,
     quality: req.quality,
+    min_criteria: req.min_criteria,
+    target_choices: req.target_choices,
     type: req.type,
     visibility: req.visibility,
     negative_ratio: req.negative_ratio,
+    split_ratio: req.split_ratio,
+    multiplicator: req.multiplicator,
+    use_meta_felix: req.use_meta_felix,
+    classified_examples: req.classified_examples,
   });
 }
 
@@ -828,6 +873,7 @@ export interface LabelExistingNERRequest {
   constraints?: ConstraintRequest[];
   save_dataset?: boolean;
   dataset_name?: string;
+  project_id?: string;
 }
 
 export async function labelExistingNER(
@@ -862,10 +908,18 @@ export interface GenerateClassificationRequest {
   save_dataset?: boolean;
   dataset_name?: string;
   project_id?: string;
+  generation_profile?: "auto" | "fast" | "balanced" | "quality";
+  include_reasoning_trace?: boolean;
+  reasoning_effort?: "low" | "medium" | "high";
   quality?: "light" | "medium" | "heavy";
+  min_criteria?: number;
+  target_choices?: number;
+  split_ratio?: Record<string, number>;
+  negative_ratio?: number;
+  multiplicator?: Record<string, unknown>;
+  use_meta_felix?: boolean;
   type?: "training" | "evaluation" | "split";
   visibility?: "private" | "public";
-  negative_ratio?: number;
 }
 
 export async function generateClassification(
@@ -886,7 +940,15 @@ export async function generateClassification(
     constraints: req.constraints,
     multi_label: req.multi_label,
     project_id: req.project_id,
+    generation_profile: req.generation_profile,
+    include_reasoning_trace: req.include_reasoning_trace,
+    reasoning_effort: req.reasoning_effort,
     quality: req.quality,
+    min_criteria: req.min_criteria,
+    target_choices: req.target_choices,
+    split_ratio: req.split_ratio,
+    multiplicator: req.multiplicator,
+    use_meta_felix: req.use_meta_felix,
     type: req.type,
     visibility: req.visibility,
     negative_ratio: req.negative_ratio,
@@ -905,6 +967,7 @@ export interface LabelExistingClassificationRequest {
   multi_label?: boolean;
   save_dataset?: boolean;
   dataset_name?: string;
+  project_id?: string;
 }
 
 export async function labelExistingClassification(
@@ -931,13 +994,21 @@ export interface GenerateCustomRequest {
   num_examples?: number;
   seed?: number;
   session_id?: string;
-  min_criteria?: number;
   temperature?: number;
   constraints?: ConstraintRequest[];
   save_dataset?: boolean;
   dataset_name?: string;
   project_id?: string;
+  generation_profile?: "auto" | "fast" | "balanced" | "quality";
+  include_reasoning_trace?: boolean;
+  reasoning_effort?: "low" | "medium" | "high";
   quality?: "light" | "medium" | "heavy";
+  min_criteria?: number;
+  target_choices?: number;
+  split_ratio?: Record<string, number>;
+  negative_ratio?: number;
+  multiplicator?: Record<string, unknown>;
+  use_meta_felix?: boolean;
   type?: "training" | "evaluation" | "split";
   visibility?: "private" | "public";
 }
@@ -958,7 +1029,15 @@ export async function generateCustom(
     temperature: req.temperature,
     constraints: req.constraints,
     project_id: req.project_id,
+    generation_profile: req.generation_profile,
+    include_reasoning_trace: req.include_reasoning_trace,
+    reasoning_effort: req.reasoning_effort,
+    target_choices: req.target_choices,
     quality: req.quality,
+    split_ratio: req.split_ratio,
+    negative_ratio: req.negative_ratio,
+    multiplicator: req.multiplicator,
+    use_meta_felix: req.use_meta_felix,
     type: req.type,
     visibility: req.visibility,
   });
@@ -973,6 +1052,20 @@ export interface GenerateRecordsRequest {
   seed?: number;
   temperature?: number;
   constraints?: ConstraintRequest[];
+  quality?: "light" | "medium" | "heavy";
+  generation_profile?: "auto" | "fast" | "balanced" | "quality";
+  include_reasoning_trace?: boolean;
+  reasoning_effort?: "low" | "medium" | "high";
+  min_criteria?: number;
+  target_choices?: number;
+  split_ratio?: Record<string, number>;
+  negative_ratio?: number;
+  multiplicator?: Record<string, unknown>;
+  project_id?: string;
+  use_meta_felix?: boolean;
+  type?: "training" | "evaluation" | "split";
+  visibility?: "private" | "public";
+  classified_examples?: Record<string, unknown>[];
 }
 
 export async function generateRecords(
@@ -987,6 +1080,20 @@ export async function generateRecords(
     seed: req.seed,
     temperature: req.temperature,
     constraints: req.constraints,
+    project_id: req.project_id,
+    generation_profile: req.generation_profile,
+    include_reasoning_trace: req.include_reasoning_trace,
+    reasoning_effort: req.reasoning_effort,
+    quality: req.quality,
+    min_criteria: req.min_criteria,
+    target_choices: req.target_choices,
+    split_ratio: req.split_ratio,
+    negative_ratio: req.negative_ratio,
+    multiplicator: req.multiplicator,
+    use_meta_felix: req.use_meta_felix,
+    type: req.type,
+    visibility: req.visibility,
+    classified_examples: req.classified_examples,
   });
 }
 
@@ -1001,6 +1108,20 @@ export interface GenerateFieldsRequest {
   temperature?: number;
   constraints?: ConstraintRequest[];
   save_dataset?: boolean;
+  quality?: "light" | "medium" | "heavy";
+  generation_profile?: "auto" | "fast" | "balanced" | "quality";
+  include_reasoning_trace?: boolean;
+  reasoning_effort?: "low" | "medium" | "high";
+  min_criteria?: number;
+  target_choices?: number;
+  split_ratio?: Record<string, number>;
+  negative_ratio?: number;
+  multiplicator?: Record<string, unknown>;
+  project_id?: string;
+  use_meta_felix?: boolean;
+  type?: "training" | "evaluation" | "split";
+  visibility?: "private" | "public";
+  classified_examples?: Record<string, unknown>[];
 }
 
 export async function generateFields(
@@ -1016,6 +1137,20 @@ export async function generateFields(
     seed: req.seed,
     temperature: req.temperature,
     constraints: req.constraints,
+    project_id: req.project_id,
+    generation_profile: req.generation_profile,
+    include_reasoning_trace: req.include_reasoning_trace,
+    reasoning_effort: req.reasoning_effort,
+    quality: req.quality,
+    min_criteria: req.min_criteria,
+    target_choices: req.target_choices,
+    split_ratio: req.split_ratio,
+    negative_ratio: req.negative_ratio,
+    multiplicator: req.multiplicator,
+    use_meta_felix: req.use_meta_felix,
+    type: req.type,
+    visibility: req.visibility,
+    classified_examples: req.classified_examples,
   });
 }
 
@@ -1029,6 +1164,7 @@ export interface LabelExistingFieldsRequest {
   constraints?: ConstraintRequest[];
   save_dataset?: boolean;
   dataset_name?: string;
+  project_id?: string;
 }
 
 export async function labelExistingFields(
@@ -1047,6 +1183,52 @@ export async function inferFields(
   return request<InferFieldsResponse>("POST", "/generate/fields/infer-fields", req);
 }
 
+export interface ImprovePromptRequest {
+  prompt: string;
+  data_type?: string;
+}
+
+export interface ImprovePromptResponse {
+  success: boolean;
+  improved_prompt: string;
+  summary: string;
+}
+
+export async function improvePrompt(
+  req: ImprovePromptRequest
+): Promise<ApiResult<ImprovePromptResponse>> {
+  return request<ImprovePromptResponse>("POST", "/generate/improve-prompt", req);
+}
+
+export interface InferAdvancedRequest {
+  prompt: string;
+  data_type?: string;
+  labels?: string[];
+}
+
+export interface InferredConstraint {
+  description: string;
+  choices?: string[];
+  probability?: number;
+}
+
+export interface InferredMultiplicator {
+  prompt: string;
+  choices: string[];
+}
+
+export interface InferAdvancedResponse {
+  success: boolean;
+  constraints: InferredConstraint[];
+  multiplicator?: InferredMultiplicator;
+}
+
+export async function inferAdvanced(
+  req: InferAdvancedRequest
+): Promise<ApiResult<InferAdvancedResponse>> {
+  return request<InferAdvancedResponse>("POST", "/generate/infer-advanced", req);
+}
+
 // ─────────────────────────────────────────────────────────────────────────────
 // Training Jobs
 // ─────────────────────────────────────────────────────────────────────────────
@@ -1341,12 +1523,14 @@ export async function listTrainedModels(): Promise<
 export interface AllModelsResponse {
   deployed: DeployedModel[];
   trained: TrainedModel[];
+  projects: ProjectResponse[];
 }
 
 export async function listAllModels(): Promise<ApiResult<AllModelsResponse>> {
-  const [deployedResult, trainedResult] = await Promise.all([
+  const [deployedResult, trainedResult, projectsResult] = await Promise.all([
     listModels(),
     listTrainedModels(),
+    listProjects(),
   ]);
 
   if (!deployedResult.ok) {
@@ -1355,6 +1539,9 @@ export async function listAllModels(): Promise<ApiResult<AllModelsResponse>> {
   if (!trainedResult.ok) {
     return { ok: false, status: trainedResult.status, error: trainedResult.error };
   }
+  if (!projectsResult.ok) {
+    return { ok: false, status: projectsResult.status, error: projectsResult.error };
+  }
 
   return {
     ok: true,
@@ -1362,6 +1549,7 @@ export async function listAllModels(): Promise<ApiResult<AllModelsResponse>> {
     data: {
       deployed: deployedResult.data?.models ?? [],
       trained: trainedResult.data?.training_jobs ?? [],
+      projects: projectsResult.data?.projects ?? [],
     },
   };
 }
@@ -1397,21 +1585,34 @@ export interface EncoderInferenceRequest {
   include_confidence?: boolean;
   include_spans?: boolean;
   format_results?: boolean;
+  is_warmup?: boolean;
+  store?: boolean;
+  project_id?: string;
+}
+
+export interface InferenceMessage {
+  role: "system" | "user" | "assistant";
+  content: string;
 }
 
 export interface GenerateInferenceRequest {
   model_id: string;
   task: "generate";
-  messages: Array<{ role: "system" | "user" | "assistant"; content: string }>;
+  messages: InferenceMessage[];
   max_tokens?: number;
   temperature?: number;
   top_p?: number;
+  include_reasoning_trace?: boolean;
+  is_warmup?: boolean;
+  store?: boolean;
+  project_id?: string;
 }
 
 export type InferenceRequest = EncoderInferenceRequest | GenerateInferenceRequest;
 
 export interface EncoderInferenceResponse {
   type: "encoder";
+  inference_id: string;
   result: Record<string, unknown> | unknown[];
   model_id: string;
   latency_ms: number;
@@ -1421,7 +1622,9 @@ export interface EncoderInferenceResponse {
 
 export interface GenerateInferenceResponse {
   type: "decoder";
+  inference_id: string;
   completion: string;
+  reasoning_trace?: string | null;
   model_id: string;
   latency_ms: number;
 }
@@ -1434,11 +1637,52 @@ export async function runInference(
   return request<InferenceResponse>("POST", "/inference", req);
 }
 
+export interface TextCompletionRequest {
+  model: string;
+  prompt: string;
+  temperature?: number;
+  max_tokens?: number;
+  stop?: string[] | string;
+  echo?: boolean;
+  extra_body?: Record<string, unknown>;
+}
+
+export interface TextCompletionUsage {
+  prompt_tokens: number;
+  completion_tokens: number;
+  total_tokens: number;
+}
+
+export interface TextCompletionChoice {
+  index?: number;
+  text: string;
+  finish_reason?: string | null;
+}
+
+export interface TextCompletionResponse {
+  id?: string;
+  object?: string;
+  created?: number;
+  model: string;
+  choices: TextCompletionChoice[];
+  usage: TextCompletionUsage;
+}
+
+export async function runTextCompletion(
+  req: TextCompletionRequest
+): Promise<ApiResult<TextCompletionResponse>> {
+  return request<TextCompletionResponse>("POST", "/v1/completions", req);
+}
+
 export interface BaseModelInfo {
   id: string;
   name?: string;
   label?: string;
-  type: "encoder" | "decoder" | string;
+  type?: "encoder" | "decoder" | string;
+  task_type?: string;
+  supports_inference?: boolean;
+  supports_on_demand_inference?: boolean;
+  supports_training?: boolean;
   description?: string;
 }
 
@@ -1449,42 +1693,325 @@ export interface BaseModelsResponse {
 export async function listBaseModels(): Promise<
   ApiResult<BaseModelsResponse | BaseModelInfo[]>
 > {
-  return request<BaseModelsResponse | BaseModelInfo[]>("GET", "/inference/base-models");
+  return request<BaseModelsResponse | BaseModelInfo[]>("GET", "/base-models");
 }
 
 // ─────────────────────────────────────────────────────────────────────────────
-// Adaptive Fine-tuning
+// Agent chat
 // ─────────────────────────────────────────────────────────────────────────────
 
-export interface AdaptiveFinetuningHistoryItem {
+export interface AgentChatHistoryItem {
   role: "user" | "assistant" | "system" | string;
   content: string;
 }
 
-export interface AdaptiveFinetuningQueryFilters {
-  start_time?: string;
-  end_time?: string;
+export interface ChatSessionMessage {
+  id?: string;
+  session_id?: string;
+  role: string;
+  content: string;
+  message_index?: number;
+  created_at?: string;
+  tool_call_id?: string | null;
+  tool_calls?: Array<{
+    id?: string;
+    name?: string;
+    args?: Record<string, unknown>;
+  }> | null;
+  images?: Array<Record<string, unknown>> | null;
+}
+
+export interface ChatSessionResponse {
+  id: string;
+  user_id: string;
+  title: string;
+  created_at: string;
+  updated_at: string;
+  is_archived: boolean;
+  project_id: string | null;
+  modal_sandbox_id: string | null;
+  raw_message_tree_present?: boolean;
+  messages: ChatSessionMessage[];
+}
+
+export interface CreateChatSessionRequest {
+  first_message?: string;
+  title?: string;
   project_id?: string;
-  model_id?: string;
 }
 
-export interface AdaptiveFinetuningChatRequest {
+export interface UpdateChatSessionRequest {
+  title?: string;
+  is_archived?: boolean;
+}
+
+export interface AppendSessionMessagesRequest {
+  messages: Array<{
+    role: string;
+    content: string;
+    tool_call_id?: string;
+    tool_calls?: Array<{
+      id?: string;
+      name?: string;
+      args?: Record<string, unknown>;
+      [key: string]: unknown;
+    }>;
+  }>;
+}
+
+export interface ChatSessionListResponse {
+  success: boolean;
+  sessions: Array<Pick<ChatSessionResponse, "id" | "user_id" | "title" | "created_at" | "updated_at" | "is_archived">>;
+}
+
+export interface AgentChatRequest {
   message: string;
   conversation_id?: string;
-  history?: AdaptiveFinetuningHistoryItem[];
-  filters?: AdaptiveFinetuningQueryFilters;
+  history?: AgentChatHistoryItem[];
+  resume_value?: string;
 }
 
-export interface AdaptiveFinetuningChatResponse {
+export interface AgentChatResponse {
   answer: string;
   conversation_id: string;
   tool_calls_made?: number;
 }
 
-export async function adaptiveFinetuningChat(
-  req: AdaptiveFinetuningChatRequest
-): Promise<ApiResult<AdaptiveFinetuningChatResponse>> {
-  return request<AdaptiveFinetuningChatResponse>("POST", "/adaptive-finetuning/chat", req);
+export interface AutoAgentRun {
+  agent_run_id: string;
+  status: string;
+  created_at?: string;
+  updated_at?: string;
+  project_id?: string;
+  model_id?: string;
+  result_url?: string;
+  run_type?: string;
+  [key: string]: unknown;
+}
+
+export async function agentChat(
+  req: AgentChatRequest
+): Promise<ApiResult<AgentChatResponse>> {
+  const baseUrl = getBaseUrl().replace(/\/$/, "");
+  const apiKey = getApiKey();
+  const url = `${baseUrl}/auto-agent/clarify`;
+  const headers: Record<string, string> = {
+    "Content-Type": "application/json",
+    "User-Agent": "pioneer-cli/0.1.0",
+  };
+
+  if (apiKey) {
+    headers["X-API-Key"] = apiKey;
+  }
+
+  try {
+    const res = await fetch(url, {
+      method: "POST",
+      headers,
+      body: JSON.stringify(req),
+    });
+
+    const text = await res.text();
+
+    let parsedJson: unknown;
+    try {
+      parsedJson = JSON.parse(text);
+    } catch {
+      // Ignore; some responses may be SSE/event text or empty.
+    }
+
+    const parseAgentStream = (raw: string): AgentChatResponse | undefined => {
+      const direct = parseAgentPayload(raw);
+      if (direct) return direct;
+
+      const events = raw
+        .split("\n")
+        .map((line) => line.trim())
+        .filter((line) => line.startsWith("data:"));
+      const fallback: AgentChatResponse = {
+        answer: "",
+        conversation_id: "",
+      };
+
+      for (const eventLine of events) {
+        const payload = eventLine.slice(5).trim();
+        if (!payload) continue;
+
+        let event: Record<string, unknown> | undefined;
+        try {
+          event = JSON.parse(payload);
+        } catch {
+          continue;
+        }
+        if (typeof event === "object" && event) {
+          const type = typeof event.type === "string" ? event.type : undefined;
+
+          if (type === "delta" && typeof event.content === "string") {
+            fallback.answer += event.content;
+          }
+
+          if (typeof event.content === "string" && !fallback.answer) {
+            fallback.answer = event.content;
+          }
+
+          if (typeof event.answer === "string") {
+            fallback.answer = event.answer;
+          }
+
+          if (typeof event.conversation_id === "string") {
+            fallback.conversation_id = event.conversation_id;
+          }
+
+          if (typeof event.tool_calls_made === "number") {
+            fallback.tool_calls_made = event.tool_calls_made;
+          }
+
+          if (type === "complete") {
+            if (typeof event.conversation_id === "string") {
+              fallback.conversation_id = event.conversation_id;
+            }
+            if (typeof event.answer === "string") {
+              fallback.answer = event.answer;
+            }
+          }
+        }
+      }
+
+      if (fallback.answer || fallback.conversation_id) {
+        return fallback;
+      }
+      return undefined;
+    };
+
+    const parseAgentPayload = (raw: string): AgentChatResponse | undefined => {
+      if (!raw.trim()) {
+        return undefined;
+      }
+      if (typeof parsedJson !== "object" || parsedJson === null) {
+        return undefined;
+      }
+
+      const payload = parsedJson as Record<string, unknown>;
+      const answer = typeof payload.answer === "string" ? payload.answer : "";
+      const conversationId =
+        typeof payload.answer === "string" && typeof payload.conversation_id === "string"
+          ? payload.conversation_id
+          : typeof payload.conversation_id === "string"
+            ? payload.conversation_id
+            : "";
+
+      if (!answer && !conversationId && typeof payload.detail !== "string") {
+        return undefined;
+      }
+
+      return {
+        answer,
+        conversation_id: conversationId,
+        ...(typeof payload.tool_calls_made === "number"
+          ? { tool_calls_made: payload.tool_calls_made }
+          : {}),
+      };
+    };
+
+    const parsedResponse = parseAgentStream(text);
+
+    if (!res.ok) {
+      const rawError = parsedJson ? JSON.stringify(parsedJson) : text || `HTTP ${res.status}`;
+      const extractedError = extractErrorMessage(parsedJson) || text || `HTTP ${res.status}`;
+      if (isAuthError(res.status, rawError)) {
+        return {
+          ok: false,
+          status: res.status,
+          error: formatAuthError(extractedError),
+        };
+      }
+      return {
+        ok: false,
+        status: res.status,
+        error: extractedError,
+      };
+    }
+
+    if (parsedResponse) {
+      return { ok: true, status: res.status, data: parsedResponse };
+    }
+
+    return {
+      ok: true,
+      status: res.status,
+      data: {
+        answer: text || "",
+        conversation_id: "",
+      },
+    };
+  } catch (err) {
+    return {
+      ok: false,
+      status: 0,
+      error: err instanceof Error ? err.message : String(err),
+    };
+  }
+}
+
+// Agent session persistence APIs (Pioneer MLE agent)
+export async function createAgentSession(
+  req: CreateChatSessionRequest
+): Promise<ApiResult<ChatSessionResponse>> {
+  return request<ChatSessionResponse>("POST", "/mle-agent/sessions", req);
+}
+
+export async function getAgentSession(
+  sessionId: string
+): Promise<ApiResult<ChatSessionResponse>> {
+  return request<ChatSessionResponse>("GET", `/mle-agent/sessions/${sessionId}`);
+}
+
+export async function listAgentSessions(): Promise<ApiResult<ChatSessionListResponse>> {
+  return request<ChatSessionListResponse>("GET", "/mle-agent/sessions");
+}
+
+export async function appendSessionMessages(
+  sessionId: string,
+  req: AppendSessionMessagesRequest
+): Promise<ApiResult<{ success: boolean; session_id: string; messages_added: number }>> {
+  return request<{ success: boolean; session_id: string; messages_added: number }>(
+    "POST",
+    `/mle-agent/sessions/${sessionId}/messages`,
+    req
+  );
+}
+
+export async function updateAgentSession(
+  sessionId: string,
+  req: UpdateChatSessionRequest
+): Promise<ApiResult<ChatSessionResponse>> {
+  return request<ChatSessionResponse>("PATCH", `/mle-agent/sessions/${sessionId}`, req);
+}
+
+export async function deleteAgentSession(sessionId: string): Promise<ApiResult<{ success: boolean }>> {
+  return request<{ success: boolean }>("DELETE", `/mle-agent/sessions/${sessionId}`);
+}
+
+export async function listAutoAgentRuns(
+  limit = 20,
+  offset = 0
+): Promise<ApiResult<{ runs: AutoAgentRun[] }>> {
+  return request<{ runs: AutoAgentRun[] }>(
+    "GET",
+    `/auto-agent/runs?limit=${limit}&offset=${offset}`
+  );
+}
+
+export async function getAutoAgentRun(
+  runId: string
+): Promise<ApiResult<AutoAgentRun>> {
+  return request<AutoAgentRun>("GET", `/auto-agent/run/${runId}`);
+}
+
+export async function stopAutoAgentRun(
+  runId: string
+): Promise<ApiResult<{ success: boolean }>> {
+  return request<{ success: boolean }>("POST", `/auto-agent/run/${runId}/stop`);
 }
 
 // ─────────────────────────────────────────────────────────────────────────────
@@ -1496,15 +2023,22 @@ export interface GenerateDecoderRequest {
   instruction?: string;
   num_examples?: number;
   temperature?: number;
+  generation_profile?: "auto" | "fast" | "balanced" | "quality";
   quality?: "light" | "medium" | "heavy";
+  include_reasoning_trace?: boolean;
+  reasoning_effort?: "low" | "medium" | "high";
   constraints?: ConstraintRequest[];
   save_dataset?: boolean;
   dataset_name?: string;
   project_id?: string;
+  use_meta_felix?: boolean;
   session_id?: string;
   config_num_examples?: number;
   min_criteria?: number;
   target_choices?: number;
+  split_ratio?: Record<string, number>;
+  negative_ratio?: number;
+  multiplicator?: Record<string, unknown>;
   type?: "training" | "evaluation" | "split";
   visibility?: "private" | "public";
 }
@@ -1519,13 +2053,20 @@ export async function generateDecoder(
     instruction: req.instruction,
     num_examples: req.num_examples,
     temperature: req.temperature,
+    generation_profile: req.generation_profile,
     quality: req.quality,
     constraints: req.constraints,
     project_id: req.project_id,
+    include_reasoning_trace: req.include_reasoning_trace,
+    reasoning_effort: req.reasoning_effort,
+    use_meta_felix: req.use_meta_felix,
     session_id: req.session_id,
     config_num_examples: req.config_num_examples,
     min_criteria: req.min_criteria,
     target_choices: req.target_choices,
+    split_ratio: req.split_ratio,
+    negative_ratio: req.negative_ratio,
+    multiplicator: req.multiplicator,
     type: req.type,
     visibility: req.visibility,
   });
@@ -1535,6 +2076,23 @@ export async function generateDecoder(
 // Evaluations
 // ─────────────────────────────────────────────────────────────────────────────
 
+export interface BaselineModel {
+  id: string;
+  name: string;
+  provider: string;
+  description: string;
+}
+
+export interface BaselineModelsResponse {
+  success: boolean;
+  models: BaselineModel[];
+  count: number;
+}
+
+export async function listBaselineModels(): Promise<ApiResult<BaselineModelsResponse>> {
+  return request<BaselineModelsResponse>("GET", "/felix/baseline-models");
+}
+
 export interface Evaluation {
   id: string;
   user_id: string;
@@ -1640,6 +2198,37 @@ export async function getDatasetEvaluations(
   );
 }
 
+export interface DeleteEvaluationResponse {
+  success: boolean;
+  message: string;
+}
+
+export async function deleteEvaluation(
+  evaluationId: string
+): Promise<ApiResult<DeleteEvaluationResponse>> {
+  return request<DeleteEvaluationResponse>("DELETE", `/felix/evaluations/${evaluationId}`);
+}
+
+export interface UpdateEvaluationProjectRequest {
+  evaluation_id: string;
+  project_id?: string;
+}
+
+export interface UpdateEvaluationProjectResponse {
+  success: boolean;
+  message: string;
+}
+
+export async function updateEvaluationProject(
+  req: UpdateEvaluationProjectRequest
+): Promise<ApiResult<UpdateEvaluationProjectResponse>> {
+  return request<UpdateEvaluationProjectResponse>(
+    "PATCH",
+    `/felix/evaluations/${req.evaluation_id}/project`,
+    { project_id: req.project_id }
+  );
+}
+
 // ─────────────────────────────────────────────────────────────────────────────
 // Benchmarks
 // ─────────────────────────────────────────────────────────────────────────────
@@ -1759,6 +2348,26 @@ export async function scanForPHD(
   });
 }
 
+export interface DataEditingDismissOutlierRequest {
+  dataset: DatasetRef;
+  fingerprint: string;
+}
+
+export interface DataEditingDismissOutlierResponse {
+  success: boolean;
+}
+
+export async function dismissOutlier(
+  req: DataEditingDismissOutlierRequest
+): Promise<ApiResult<DataEditingDismissOutlierResponse>> {
+  const { dataset, ...rest } = req;
+  return request<DataEditingDismissOutlierResponse>("POST", "/felix/dataset/outliers/dismiss", {
+    ...rest,
+    dataset_name: dataset.name,
+    dataset_version: dataset.version,
+  });
+}
+
 export interface DataEditingRemoveRequest {
   dataset: DatasetRef;
   findings: PIIFinding[];
@@ -2163,10 +2772,11 @@ export async function listActivity(
 // ─────────────────────────────────────────────────────────────────────────────
 
 export interface ProjectCreateRequest {
-  name: string;
+  name?: string;
   icon?: string;
   repo?: string;
   description?: string;
+  active_model_id?: string;
   selected_model_id?: string;
   example?: Record<string, unknown>;
 }
@@ -2204,6 +2814,19 @@ export interface ProjectDeleteResponse {
   project_id: string;
 }
 
+export interface ProjectDeploymentCreate {
+  training_job_id: string;
+  reason?: string;
+}
+
+export interface ProjectDeploymentResponse {
+  success: boolean;
+  message?: string;
+  project_id?: string;
+  training_job_id?: string;
+  deployment_id?: string;
+}
+
 export interface ProjectDatasetCountResponse {
   project_id: string;
   dataset_count: number;
@@ -2265,6 +2888,27 @@ export async function getProjectQualityMetrics(
   );
 }
 
+export async function deployTrainingJobToProject(
+  projectId: string,
+  req: ProjectDeploymentCreate
+): Promise<ApiResult<ProjectDeploymentResponse>> {
+  return request<ProjectDeploymentResponse>(
+    "POST",
+    `/projects/${projectId}/deployments`,
+    req
+  );
+}
+
+export async function rollbackProjectDeployment(
+  projectId: string,
+  deploymentId: string
+): Promise<ApiResult<ProjectDeploymentResponse>> {
+  return request<ProjectDeploymentResponse>(
+    "POST",
+    `/projects/${projectId}/deployments/${deploymentId}/rollback`
+  );
+}
+
 // ─────────────────────────────────────────────────────────────────────────────
 // Presets
 // ─────────────────────────────────────────────────────────────────────────────
@@ -2440,205 +3084,3 @@ export async function endAnnotationSession(
   );
 }
 
-// ─────────────────────────────────────────────────────────────────────────────
-// Leaderboard / Competitions
-// ─────────────────────────────────────────────────────────────────────────────
-
-export interface LeaderboardEntry {
-  id: string;
-  dataset_id: string;
-  evaluation_id: string;
-  user_id: string;
-  display_name: string;
-  model_name: string;
-  model_id?: string;
-  f1_score: number;
-  precision_score?: number;
-  recall_score?: number;
-  accuracy?: number;
-  created_at: string;
-  updated_at: string;
-  rank?: number;
-}
-
-export interface CompetitionInfo {
-  dataset_id: string;
-  dataset_name: string;
-  dataset_type: string;
-  description?: string;
-  sample_count?: number;
-  labels?: string[];
-  sample_rows?: Record<string, unknown>[];
-  winner?: LeaderboardEntry;
-  total_entries: number;
-}
-
-export interface CompetitionsResponse {
-  success: boolean;
-  competitions: CompetitionInfo[];
-}
-
-export interface CompetitionSamplesResponse {
-  dataset_id: string;
-  sample_rows: Record<string, unknown>[];
-}
-
-export interface LeaderboardEntriesResponse {
-  success: boolean;
-  dataset_id: string;
-  dataset_name: string;
-  entries: LeaderboardEntry[];
-  total_entries: number;
-}
-
-export interface LeaderboardSubmission {
-  evaluation_id: string;
-  display_name: string;
-}
-
-export interface LeaderboardSubmitResponse {
-  success: boolean;
-  entry: LeaderboardEntry;
-  rank: number;
-  is_new_best: boolean;
-}
-
-export async function listCompetitions(): Promise<ApiResult<CompetitionsResponse>> {
-  return request<CompetitionsResponse>("GET", "/leaderboard/competitions");
-}
-
-export async function getCompetitionSamples(
-  datasetId: string
-): Promise<ApiResult<CompetitionSamplesResponse>> {
-  return request<CompetitionSamplesResponse>(
-    "GET",
-    `/leaderboard/competitions/${datasetId}/samples`
-  );
-}
-
-export async function getLeaderboardEntries(
-  datasetId: string,
-  limit?: number
-): Promise<ApiResult<LeaderboardEntriesResponse>> {
-  const params = new URLSearchParams();
-  if (limit) params.set("limit", String(limit));
-  const query = params.toString();
-  const url = `/leaderboard/datasets/${datasetId}/entries${query ? `?${query}` : ""}`;
-  return request<LeaderboardEntriesResponse>("GET", url);
-}
-
-export async function submitToLeaderboard(
-  datasetId: string,
-  submission: LeaderboardSubmission
-): Promise<ApiResult<LeaderboardSubmitResponse>> {
-  return request<LeaderboardSubmitResponse>(
-    "POST",
-    `/leaderboard/datasets/${datasetId}/submit`,
-    submission
-  );
-}
-
-// ─────────────────────────────────────────────────────────────────────────────
-// Notebook Execution
-// ─────────────────────────────────────────────────────────────────────────────
-
-export interface CreateNotebookSessionRequest {
-  gpu?: string;
-  load_felix_helpers?: boolean;
-  auto_terminate_seconds?: number;
-  tags?: string[];
-}
-
-export interface CreateNotebookSessionResponse {
-  session_id: string;
-  gpu: string;
-  message: string;
-  status: "creating" | "ready";
-}
-
-export interface SessionStatusResponse {
-  session_id: string;
-  status: "creating" | "ready" | "failed";
-  error?: string;
-}
-
-export interface ExecuteOutput {
-  type: string;
-  name?: string;
-  text?: string;
-  data?: Record<string, string>;
-  ename?: string;
-  evalue?: string;
-  traceback?: string[];
-}
-
-export interface ExecuteCodeRequest {
-  code: string;
-  session_id?: string;
-  cell_id?: string;
-  include_variables?: boolean;
-  gpu?: string;
-  load_felix_helpers?: boolean;
-}
-
-export interface ExecuteCodeResponse {
-  success: boolean;
-  outputs: ExecuteOutput[];
-  variables: Record<string, unknown>;
-  execution_time_ms: number;
-  gpu_used?: string;
-  cell_id?: string;
-  session_id: string;
-  error?: string;
-  session_created?: boolean;
-}
-
-export interface NotebookSessionInfo {
-  session_id: string;
-  gpu: string;
-  created_at: string;
-  last_activity: string;
-  status: string;
-}
-
-export interface ListSessionsResponse {
-  sessions: NotebookSessionInfo[];
-}
-
-export async function createNotebookSession(
-  req?: CreateNotebookSessionRequest
-): Promise<ApiResult<CreateNotebookSessionResponse>> {
-  return request<CreateNotebookSessionResponse>(
-    "POST",
-    "/felix/notebook/sessions",
-    req ?? {}
-  );
-}
-
-export async function getNotebookSessionStatus(
-  sessionId: string
-): Promise<ApiResult<SessionStatusResponse>> {
-  return request<SessionStatusResponse>(
-    "GET",
-    `/felix/notebook/sessions/${sessionId}/status`
-  );
-}
-
-export async function listNotebookSessions(): Promise<ApiResult<ListSessionsResponse>> {
-  return request<ListSessionsResponse>("GET", "/felix/notebook/sessions");
-}
-
-export async function executeNotebookCode(
-  req: ExecuteCodeRequest
-): Promise<ApiResult<ExecuteCodeResponse>> {
-  return request<ExecuteCodeResponse>("POST", "/felix/notebook/execute", req);
-}
-
-export async function terminateNotebookSession(
-  sessionId: string
-): Promise<ApiResult<{ message: string }>> {
-  return request<{ message: string }>(
-    "DELETE",
-    `/felix/notebook/sessions/${sessionId}`
-  );
-}