feat: add codex (#381)

Author: k11kirky

apps/array/package.json

@@ -121,6 +121,7 @@
     "cmdk": "^1.1.1",
     "date-fns": "^3.3.1",
     "detect-libc": "^1.0.3",
+    "dotenv": "^17.2.3",
     "electron-log": "^5.4.3",
     "electron-store": "^11.0.0",
     "file-icon": "^6.0.0",

apps/array/src/main/index.ts

@@ -3,6 +3,7 @@ declare const __BUILD_DATE__: string | undefined;
 
 import "reflect-metadata";
 import dns from "node:dns";
+
 import { mkdirSync } from "node:fs";
 import os from "node:os";
 import path from "node:path";

apps/array/src/main/services/agent/schemas.ts

@@ -9,6 +9,10 @@ export const credentialsSchema = z.object({
 
 export type Credentials = z.infer<typeof credentialsSchema>;
 
+// Agent framework schema
+export const agentFrameworkSchema = z.enum(["claude", "codex"]);
+export type AgentFramework = z.infer<typeof agentFrameworkSchema>;
+
 // Session config schema
 export const sessionConfigSchema = z.object({
   taskId: z.string(),
@@ -18,11 +22,13 @@ export const sessionConfigSchema = z.object({
   logUrl: z.string().optional(),
   sdkSessionId: z.string().optional(),
   model: z.string().optional(),
+  framework: agentFrameworkSchema.optional(),
 });
 
 export type SessionConfig = z.infer<typeof sessionConfigSchema>;
 
 // Start session input/output
+
 export const startSessionInput = z.object({
   taskId: z.string(),
   taskRunId: z.string(),
@@ -33,6 +39,7 @@ export const startSessionInput = z.object({
   permissionMode: z.string().optional(),
   autoProgress: z.boolean().optional(),
   model: z.string().optional(),
+  framework: agentFrameworkSchema.optional().default("claude"),
   executionMode: z.enum(["plan"]).optional(),
   runMode: z.enum(["local", "cloud"]).optional(),
   createPR: z.boolean().optional(),

apps/array/src/main/services/agent/service.ts

@@ -129,6 +129,7 @@ interface SessionConfig {
   logUrl?: string;
   sdkSessionId?: string;
   model?: string;
+  framework?: "claude" | "codex";
 }
 
 interface ManagedSession {
@@ -230,6 +231,7 @@ export class AgentService extends TypedEventEmitter<AgentServiceEvents> {
       logUrl,
       sdkSessionId,
       model,
+      framework,
     } = config;
 
     if (!isRetry) {
@@ -255,6 +257,7 @@ export class AgentService extends TypedEventEmitter<AgentServiceEvents> {
     try {
       const { clientStreams } = await agent.runTaskV2(taskId, taskRunId, {
         skipGitBranch: true,
+        framework,
       });
 
       const connection = this.createClientConnection(
@@ -457,7 +460,15 @@ export class AgentService extends TypedEventEmitter<AgentServiceEvents> {
     process.env.ANTHROPIC_API_KEY = token;
     process.env.ANTHROPIC_AUTH_TOKEN = token;
 
-    process.env.ANTHROPIC_BASE_URL = getLlmGatewayUrl(credentials.apiHost);
+    const llmGatewayUrl = getLlmGatewayUrl(credentials.apiHost);
+    process.env.ANTHROPIC_BASE_URL = llmGatewayUrl;
+
+    const openaiBaseUrl = llmGatewayUrl.endsWith("/v1")
+      ? llmGatewayUrl
+      : `${llmGatewayUrl}/v1`;
+    process.env.OPENAI_BASE_URL = openaiBaseUrl;
+    process.env.OPENAI_API_KEY = token;
+    process.env.LLM_GATEWAY_URL = llmGatewayUrl;
 
     process.env.CLAUDE_CODE_EXECUTABLE = getClaudeCliPath();
 
@@ -599,6 +610,7 @@ export class AgentService extends TypedEventEmitter<AgentServiceEvents> {
       logUrl: "logUrl" in params ? params.logUrl : undefined,
       sdkSessionId: "sdkSessionId" in params ? params.sdkSessionId : undefined,
       model: "model" in params ? params.model : undefined,
+      framework: "framework" in params ? params.framework : "claude",
     };
   }
 

apps/array/src/renderer/features/sessions/components/FrameworkSelector.tsx

@@ -0,0 +1,74 @@
+import { useSettingsStore } from "@features/settings/stores/settingsStore";
+import { Select, Text } from "@radix-ui/themes";
+import {
+  type AgentFramework,
+  AVAILABLE_FRAMEWORKS,
+} from "@shared/types/models";
+import { useSessionForTask } from "../stores/sessionStore";
+
+interface FrameworkSelectorProps {
+  taskId?: string;
+  disabled?: boolean;
+  onFrameworkChange?: (framework: AgentFramework) => void;
+}
+
+export function FrameworkSelector({
+  taskId,
+  disabled,
+  onFrameworkChange,
+}: FrameworkSelectorProps) {
+  const defaultFramework = useSettingsStore((state) => state.defaultFramework);
+  const setDefaultFramework = useSettingsStore(
+    (state) => state.setDefaultFramework,
+  );
+  const session = useSessionForTask(taskId);
+
+  // Use session framework if available, otherwise fall back to default
+  const activeFramework = session?.framework ?? defaultFramework;
+
+  // Disable if there's an active session (can't change framework mid-session)
+  const isDisabled = disabled || session?.status === "connected";
+
+  const handleChange = (value: string) => {
+    const framework = value as AgentFramework;
+    setDefaultFramework(framework);
+    onFrameworkChange?.(framework);
+  };
+
+  const currentFramework = AVAILABLE_FRAMEWORKS.find(
+    (f) => f.id === activeFramework,
+  );
+  const displayName = currentFramework?.name ?? activeFramework;
+
+  return (
+    <Select.Root
+      value={activeFramework}
+      onValueChange={handleChange}
+      disabled={isDisabled}
+      size="1"
+    >
+      <Select.Trigger
+        variant="ghost"
+        style={{
+          fontSize: "var(--font-size-1)",
+          color: "var(--gray-11)",
+          padding: "4px 8px",
+          marginLeft: "4px",
+          height: "auto",
+          minHeight: "unset",
+        }}
+      >
+        <Text size="1" style={{ fontFamily: "var(--font-mono)" }}>
+          {displayName}
+        </Text>
+      </Select.Trigger>
+      <Select.Content position="popper" sideOffset={4}>
+        {AVAILABLE_FRAMEWORKS.filter((f) => f.enabled).map((framework) => (
+          <Select.Item key={framework.id} value={framework.id}>
+            {framework.name}
+          </Select.Item>
+        ))}
+      </Select.Content>
+    </Select.Root>
+  );
+}

apps/array/src/renderer/features/sessions/stores/sessionStore.ts

@@ -39,6 +39,7 @@ export interface AgentSession {
   logUrl?: string;
   processedLineCount?: number;
   model?: string;
+  framework?: "claude" | "codex";
 }
 
 interface SessionState {
@@ -267,6 +268,7 @@ function createBaseSession(
   taskRunId: string,
   taskId: string,
   isCloud: boolean,
+  framework?: "claude" | "codex",
 ): AgentSession {
   return {
     taskRunId,
@@ -277,6 +279,7 @@ function createBaseSession(
     status: "connecting",
     isPromptPending: false,
     isCloud,
+    framework,
   };
 }
 
@@ -471,7 +474,7 @@ const useStore = create<SessionStore>()(
         return;
       }
 
-      const defaultModel = useSettingsStore.getState().defaultModel;
+      const { defaultModel, defaultFramework } = useSettingsStore.getState();
       const result = await trpcVanilla.agent.start.mutate({
         taskId,
         taskRunId: taskRun.id,
@@ -480,9 +483,15 @@ const useStore = create<SessionStore>()(
         apiHost: auth.apiHost,
         projectId: auth.projectId,
         model: defaultModel,
+        framework: defaultFramework,
       });
 
-      const session = createBaseSession(taskRun.id, taskId, false);
+      const session = createBaseSession(
+        taskRun.id,
+        taskId,
+        false,
+        defaultFramework,
+      );
       session.channel = result.channel;
       session.status = "connected";
       session.model = defaultModel;

apps/array/src/renderer/features/settings/stores/settingsStore.ts

@@ -1,5 +1,9 @@
 import type { WorkspaceMode } from "@shared/types";
-import { DEFAULT_MODEL } from "@shared/types/models";
+import {
+  type AgentFramework,
+  DEFAULT_FRAMEWORK,
+  DEFAULT_MODEL,
+} from "@shared/types/models";
 import { create } from "zustand";
 import { persist } from "zustand/middleware";
 
@@ -14,6 +18,7 @@ interface SettingsStore {
   lastUsedWorkspaceMode: WorkspaceMode;
   createPR: boolean;
   defaultModel: string;
+  defaultFramework: AgentFramework;
 
   setAutoRunTasks: (autoRun: boolean) => void;
   setDefaultRunMode: (mode: DefaultRunMode) => void;
@@ -22,6 +27,7 @@ interface SettingsStore {
   setLastUsedWorkspaceMode: (mode: WorkspaceMode) => void;
   setCreatePR: (createPR: boolean) => void;
   setDefaultModel: (model: string) => void;
+  setDefaultFramework: (framework: AgentFramework) => void;
 }
 
 export const useSettingsStore = create<SettingsStore>()(
@@ -34,6 +40,7 @@ export const useSettingsStore = create<SettingsStore>()(
       lastUsedWorkspaceMode: "worktree",
       createPR: true,
       defaultModel: DEFAULT_MODEL,
+      defaultFramework: DEFAULT_FRAMEWORK,
 
       setAutoRunTasks: (autoRun) => set({ autoRunTasks: autoRun }),
       setDefaultRunMode: (mode) => set({ defaultRunMode: mode }),
@@ -43,6 +50,7 @@ export const useSettingsStore = create<SettingsStore>()(
       setLastUsedWorkspaceMode: (mode) => set({ lastUsedWorkspaceMode: mode }),
       setCreatePR: (createPR) => set({ createPR }),
       setDefaultModel: (model) => set({ defaultModel: model }),
+      setDefaultFramework: (framework) => set({ defaultFramework: framework }),
     }),
     {
       name: "settings-storage",

apps/array/src/renderer/features/task-detail/components/TaskInputEditor.tsx

@@ -2,6 +2,7 @@ import "@features/message-editor/components/message-editor.css";
 import { EditorToolbar } from "@features/message-editor/components/EditorToolbar";
 import type { MessageEditorHandle } from "@features/message-editor/components/MessageEditor";
 import { useTiptapEditor } from "@features/message-editor/tiptap/useTiptapEditor";
+import { FrameworkSelector } from "@features/sessions/components/FrameworkSelector";
 import { ArrowUp, GitBranchIcon } from "@phosphor-icons/react";
 import { Box, Flex, IconButton, Text, Tooltip } from "@radix-ui/themes";
 import { EditorContent } from "@tiptap/react";
@@ -172,12 +173,15 @@ export const TaskInputEditor = forwardRef<
         </Flex>
 
         <Flex justify="between" align="center" px="3" pb="3">
-          <EditorToolbar
-            disabled={isCreatingTask}
-            onInsertChip={insertChip}
-            attachTooltip="Attach files from anywhere"
-            iconSize={16}
-          />
+          <Flex align="center" gap="1">
+            <EditorToolbar
+              disabled={isCreatingTask}
+              onInsertChip={insertChip}
+              attachTooltip="Attach files from anywhere"
+              iconSize={16}
+            />
+            <FrameworkSelector disabled={isCreatingTask} />
+          </Flex>
 
           <Flex align="center" gap="4">
             {!isCloudMode && (

apps/array/src/shared/types/models.ts

@@ -65,6 +65,33 @@ export const AVAILABLE_MODELS: ModelOption[] = [
 
 export const DEFAULT_MODEL = "claude-opus-4-5";
 
+// Agent frameworks
+export type AgentFramework = "claude" | "codex";
+
+export interface FrameworkOption {
+  id: AgentFramework;
+  name: string;
+  description: string;
+  enabled: boolean;
+}
+
+export const AVAILABLE_FRAMEWORKS: FrameworkOption[] = [
+  {
+    id: "claude",
+    name: "Claude Code",
+    description: "Anthropic's Claude Code agent",
+    enabled: true,
+  },
+  {
+    id: "codex",
+    name: "OpenAI Codex",
+    description: "OpenAI's Codex agent",
+    enabled: true,
+  },
+];
+
+export const DEFAULT_FRAMEWORK: AgentFramework = "claude";
+
 export function getModelById(id: string): ModelOption | undefined {
   return AVAILABLE_MODELS.find((m) => m.id === id && m.enabled);
 }

packages/agent/index.ts

@@ -14,9 +14,10 @@ export type {
 export { POSTHOG_NOTIFICATIONS } from "./src/acp-extensions.js";
 export type {
   AcpConnectionConfig,
+  AgentFramework,
   InProcessAcpConnection,
-} from "./src/adapters/claude/claude.js";
-export { createAcpConnection } from "./src/adapters/claude/claude.js";
+} from "./src/adapters/connection.js";
+export { createAcpConnection } from "./src/adapters/connection.js";
 export { Agent } from "./src/agent.js";
 export type {
   AgentEvent,