stagent 0.1.0

package/src/lib/agents/runtime/task-assist-types.ts

@@ -0,0 +1,8 @@
+export interface TaskAssistResponse {
+  improvedDescription: string;
+  breakdown: { title: string; description: string }[];
+  recommendedPattern: "single" | "sequence" | "planner-executor" | "checkpoint";
+  complexity: "simple" | "moderate" | "complex";
+  needsCheckpoint: boolean;
+  reasoning: string;
+}

package/src/lib/agents/runtime/types.ts

@@ -0,0 +1,30 @@
+import type { ApiKeySource } from "@/lib/constants/settings";
+import type { ProfileTestReport } from "@/lib/agents/profiles/test-types";
+import type { RuntimeCapabilities, RuntimeCatalogEntry } from "./catalog";
+import type { TaskAssistResponse } from "./task-assist-types";
+
+export interface RuntimeConnectionResult {
+  connected: boolean;
+  apiKeySource?: ApiKeySource;
+  error?: string;
+}
+
+export interface TaskAssistInput {
+  title?: string;
+  description?: string;
+}
+
+export interface AgentRuntimeAdapter {
+  metadata: RuntimeCatalogEntry;
+  executeTask(taskId: string): Promise<void>;
+  resumeTask(taskId: string): Promise<void>;
+  cancelTask(taskId: string): Promise<void>;
+  runTaskAssist?(input: TaskAssistInput): Promise<TaskAssistResponse>;
+  runProfileTests?(profileId: string): Promise<ProfileTestReport>;
+  testConnection?(): Promise<RuntimeConnectionResult>;
+}
+
+export interface RuntimeSummary {
+  runtime: RuntimeCatalogEntry;
+  capabilities: RuntimeCapabilities;
+}

package/src/lib/constants/settings.ts

@@ -0,0 +1,13 @@
+export const SETTINGS_KEYS = {
+  AUTH_METHOD: "auth.method",
+  AUTH_API_KEY: "auth.apiKey",
+  AUTH_API_KEY_SOURCE: "auth.apiKeySource",
+  OPENAI_AUTH_API_KEY: "openai.authApiKey",
+  OPENAI_AUTH_API_KEY_SOURCE: "openai.authApiKeySource",
+  PERMISSIONS_ALLOW: "permissions.allow",
+  BUDGET_POLICY: "usage.budgetPolicy",
+  BUDGET_WARNING_STATE: "usage.budgetWarningState",
+} as const;
+
+export type AuthMethod = "api_key" | "oauth";
+export type ApiKeySource = "db" | "env" | "oauth" | "unknown";

package/src/lib/constants/status-colors.ts

@@ -0,0 +1,44 @@
+/**
+ * Shared status-to-badge-variant mappings.
+ * Single source of truth — consumed by every component that renders status badges.
+ */
+
+export const taskStatusVariant: Record<string, "default" | "secondary" | "destructive" | "outline" | "success"> = {
+  planned: "outline",
+  queued: "secondary",
+  running: "default",
+  completed: "success",
+  failed: "destructive",
+  cancelled: "secondary",
+};
+
+export const projectStatusVariant: Record<string, "default" | "secondary" | "outline"> = {
+  active: "default",
+  paused: "secondary",
+  completed: "outline",
+};
+
+export const workflowStatusVariant: Record<string, "default" | "secondary" | "destructive" | "outline"> = {
+  draft: "outline",
+  active: "default",
+  paused: "secondary",
+  completed: "default",
+  failed: "destructive",
+};
+
+export const scheduleStatusVariant: Record<string, "default" | "secondary" | "destructive" | "outline" | "success"> = {
+  active: "default",
+  paused: "secondary",
+  completed: "success",
+  expired: "outline",
+};
+
+export const patternLabels: Record<string, string> = {
+  single: "Single Task",
+  sequence: "Sequence",
+  "planner-executor": "Planner → Executor",
+  checkpoint: "Human-in-the-Loop",
+  loop: "Autonomous Loop",
+  parallel: "Parallel Research",
+  swarm: "Multi-Agent Swarm",
+};

package/src/lib/constants/task-status.ts

@@ -0,0 +1,49 @@
+export const TASK_STATUSES = [
+  "planned",
+  "queued",
+  "running",
+  "completed",
+  "failed",
+  "cancelled",
+] as const;
+
+export type TaskStatus = (typeof TASK_STATUSES)[number];
+
+export const COLUMN_ORDER: TaskStatus[] = [
+  "planned",
+  "queued",
+  "running",
+  "completed",
+  "failed",
+];
+
+// Valid status transitions (system/agent can do these)
+export const VALID_TRANSITIONS: Record<TaskStatus, TaskStatus[]> = {
+  planned: ["queued", "cancelled"],
+  queued: ["running", "planned", "cancelled"],
+  running: ["completed", "failed", "cancelled"],
+  completed: ["planned"],
+  failed: ["planned", "queued", "running"],
+  cancelled: ["planned", "running"],
+};
+
+// Transitions a user can trigger via drag-and-drop
+export const USER_DRAG_TRANSITIONS: Record<TaskStatus, TaskStatus[]> = {
+  planned: ["queued"],
+  queued: ["planned"],
+  running: [],
+  completed: ["planned"],
+  failed: ["planned"],
+  cancelled: ["planned"],
+};
+
+export function isValidTransition(from: TaskStatus, to: TaskStatus): boolean {
+  return VALID_TRANSITIONS[from]?.includes(to) ?? false;
+}
+
+export function isValidDragTransition(from: TaskStatus, to: TaskStatus): boolean {
+  return USER_DRAG_TRANSITIONS[from]?.includes(to) ?? false;
+}
+
+/** Maximum number of times a task can be resumed before requiring a fresh start */
+export const MAX_RESUME_COUNT = 3;

package/src/lib/data/clear.ts

@@ -0,0 +1,63 @@
+import { db } from "@/lib/db";
+import {
+  agentLogs,
+  notifications,
+  documents,
+  tasks,
+  workflows,
+  schedules,
+  projects,
+  usageLedger,
+} from "@/lib/db/schema";
+import { readdirSync, unlinkSync, mkdirSync } from "fs";
+import { join } from "path";
+import { homedir } from "os";
+import { clearSampleProfiles } from "./seed-data/profiles";
+
+const uploadsDir = join(
+  process.env.STAGENT_DATA_DIR || join(homedir(), ".stagent"),
+  "uploads"
+);
+
+/**
+ * Wipe all data tables (FK-safe order) and uploaded files.
+ * Preserves the settings table (auth config).
+ */
+export function clearAllData() {
+  const sampleProfilesDeleted = clearSampleProfiles();
+
+  // Delete in FK-safe order: children before parents
+  const usageLedgerDeleted = db.delete(usageLedger).run().changes;
+  const logsDeleted = db.delete(agentLogs).run().changes;
+  const notificationsDeleted = db.delete(notifications).run().changes;
+  const documentsDeleted = db.delete(documents).run().changes;
+  const tasksDeleted = db.delete(tasks).run().changes;
+  const workflowsDeleted = db.delete(workflows).run().changes;
+  const schedulesDeleted = db.delete(schedules).run().changes;
+  const projectsDeleted = db.delete(projects).run().changes;
+
+  // Wipe uploaded files
+  let filesDeleted = 0;
+  mkdirSync(uploadsDir, { recursive: true });
+  try {
+    for (const file of readdirSync(uploadsDir)) {
+      unlinkSync(join(uploadsDir, file));
+      filesDeleted++;
+    }
+  } catch {
+    // Directory may not exist yet — that's fine
+  }
+
+  return {
+    sampleProfiles: sampleProfilesDeleted,
+    projects: projectsDeleted,
+    tasks: tasksDeleted,
+    workflows: workflowsDeleted,
+    schedules: schedulesDeleted,
+    usageLedger: usageLedgerDeleted,
+    agentLogs: logsDeleted,
+    notifications: notificationsDeleted,
+    documents: documentsDeleted,
+    files: filesDeleted,
+  };
+}