fix: improve realtime controls layout and add agent status bar
- Change realtime controls to horizontal layout with fixed height (88px) - Add orientation prop to VolumeMeter for horizontal/vertical modes - Create AgentStatusBar component showing mode badge and agent status - Add agent persistence system for managing agent state across restarts - Fix bottom sheet dismiss handling to clear state properly - Remove debug timer from realtime controls 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,145 @@
|
||||
import { View, Text, Pressable } from "react-native";
|
||||
import { StyleSheet, useUnistyles } from "react-native-unistyles";
|
||||
import { ChevronDown } from "lucide-react-native";
|
||||
import { useSession } from "@/contexts/session-context";
|
||||
import { useState } from "react";
|
||||
import { ModeSelectorModal } from "./mode-selector-modal";
|
||||
|
||||
interface AgentStatusBarProps {
|
||||
agentId: string;
|
||||
}
|
||||
|
||||
export function AgentStatusBar({ agentId }: AgentStatusBarProps) {
|
||||
const { theme } = useUnistyles();
|
||||
const { agents, setAgentMode } = useSession();
|
||||
const [showModeSelector, setShowModeSelector] = useState(false);
|
||||
|
||||
const agent = agents.get(agentId);
|
||||
|
||||
if (!agent) {
|
||||
return null;
|
||||
}
|
||||
|
||||
function handleModeChange(modeId: string) {
|
||||
setAgentMode(agentId, modeId);
|
||||
}
|
||||
|
||||
function getStatusLabel(status: string): string {
|
||||
switch (status) {
|
||||
case "ready":
|
||||
return "Ready";
|
||||
case "processing":
|
||||
return "Working";
|
||||
case "completed":
|
||||
return "Completed";
|
||||
case "failed":
|
||||
return "Failed";
|
||||
case "killed":
|
||||
return "Stopped";
|
||||
default:
|
||||
return status;
|
||||
}
|
||||
}
|
||||
|
||||
function getStatusColor(status: string): string {
|
||||
switch (status) {
|
||||
case "ready":
|
||||
return theme.colors.palette.green[500];
|
||||
case "processing":
|
||||
return theme.colors.palette.blue[500];
|
||||
case "completed":
|
||||
return theme.colors.palette.gray[500];
|
||||
case "failed":
|
||||
return theme.colors.palette.red[500];
|
||||
case "killed":
|
||||
return theme.colors.palette.orange[500];
|
||||
default:
|
||||
return theme.colors.mutedForeground;
|
||||
}
|
||||
}
|
||||
|
||||
return (
|
||||
<View style={styles.container}>
|
||||
{/* Agent Mode Badge */}
|
||||
{agent.availableModes && agent.availableModes.length > 0 && (
|
||||
<Pressable
|
||||
onPress={() => setShowModeSelector(true)}
|
||||
style={({ pressed }) => [
|
||||
styles.modeBadge,
|
||||
pressed && styles.modeBadgePressed,
|
||||
]}
|
||||
>
|
||||
<Text style={styles.modeBadgeText}>
|
||||
{agent.availableModes?.find((m) => m.id === agent.currentModeId)?.name ||
|
||||
agent.currentModeId ||
|
||||
"default"}
|
||||
</Text>
|
||||
<ChevronDown size={14} color={theme.colors.mutedForeground} />
|
||||
</Pressable>
|
||||
)}
|
||||
|
||||
{/* Agent Status Indicator */}
|
||||
<View style={styles.statusIndicator}>
|
||||
<View
|
||||
style={[
|
||||
styles.statusDot,
|
||||
{ backgroundColor: getStatusColor(agent.status) },
|
||||
]}
|
||||
/>
|
||||
<Text style={styles.statusText}>{getStatusLabel(agent.status)}</Text>
|
||||
</View>
|
||||
|
||||
{/* Mode selector modal */}
|
||||
<ModeSelectorModal
|
||||
visible={showModeSelector}
|
||||
agent={agent}
|
||||
onModeChange={handleModeChange}
|
||||
onClose={() => setShowModeSelector(false)}
|
||||
/>
|
||||
</View>
|
||||
);
|
||||
}
|
||||
|
||||
const styles = StyleSheet.create((theme) => ({
|
||||
container: {
|
||||
flexDirection: "row",
|
||||
alignItems: "center",
|
||||
gap: theme.spacing[3],
|
||||
paddingHorizontal: theme.spacing[4],
|
||||
paddingTop: theme.spacing[2],
|
||||
paddingBottom: theme.spacing[1],
|
||||
},
|
||||
modeBadge: {
|
||||
flexDirection: "row",
|
||||
alignItems: "center",
|
||||
gap: theme.spacing[2],
|
||||
paddingHorizontal: theme.spacing[3],
|
||||
paddingVertical: theme.spacing[2],
|
||||
backgroundColor: theme.colors.muted,
|
||||
borderRadius: theme.borderRadius.full,
|
||||
},
|
||||
modeBadgePressed: {
|
||||
backgroundColor: theme.colors.accent,
|
||||
},
|
||||
modeBadgeText: {
|
||||
color: theme.colors.mutedForeground,
|
||||
fontSize: theme.fontSize.xs,
|
||||
fontWeight: theme.fontWeight.semibold,
|
||||
textTransform: "capitalize",
|
||||
},
|
||||
statusIndicator: {
|
||||
flexDirection: "row",
|
||||
alignItems: "center",
|
||||
gap: theme.spacing[2],
|
||||
},
|
||||
statusDot: {
|
||||
width: 8,
|
||||
height: 8,
|
||||
borderRadius: 4,
|
||||
},
|
||||
statusText: {
|
||||
color: theme.colors.mutedForeground,
|
||||
fontSize: theme.fontSize.xs,
|
||||
fontWeight: theme.fontWeight.semibold,
|
||||
},
|
||||
}));
|
||||
@@ -45,7 +45,14 @@ export function AgentStreamView({
|
||||
|
||||
function handleOpenToolCallDetails(toolCall: SelectedToolCall) {
|
||||
setSelectedToolCall(toolCall);
|
||||
bottomSheetRef.current?.present();
|
||||
// Delay present to next frame to ensure component is mounted
|
||||
setTimeout(() => {
|
||||
bottomSheetRef.current?.present();
|
||||
}, 0);
|
||||
}
|
||||
|
||||
function handleBottomSheetDismiss() {
|
||||
setSelectedToolCall(null);
|
||||
}
|
||||
|
||||
return (
|
||||
@@ -171,6 +178,7 @@ export function AgentStreamView({
|
||||
<ToolCallBottomSheet
|
||||
bottomSheetRef={bottomSheetRef}
|
||||
selectedToolCall={selectedToolCall}
|
||||
onDismiss={handleBottomSheetDismiss}
|
||||
/>
|
||||
</View>
|
||||
);
|
||||
|
||||
@@ -72,6 +72,9 @@ const styles = StyleSheet.create((theme) => ({
|
||||
borderTopWidth: theme.borderWidth[1],
|
||||
borderTopColor: theme.colors.border,
|
||||
},
|
||||
realtimeContainer: {
|
||||
backgroundColor: theme.colors.background,
|
||||
},
|
||||
centeredButtonContainer: {
|
||||
padding: theme.spacing[6],
|
||||
alignItems: "center",
|
||||
|
||||
@@ -23,6 +23,7 @@ export const OrchestratorMessagesView = forwardRef<ScrollView, OrchestratorMessa
|
||||
const { theme } = useUnistyles();
|
||||
const bottomSheetRef = useRef<BottomSheetModal | null>(null);
|
||||
const [selectedToolCall, setSelectedToolCall] = useState<SelectedToolCall | null>(null);
|
||||
const [isBottomSheetVisible, setIsBottomSheetVisible] = useState(false);
|
||||
|
||||
function handleOpenToolCallDetails(toolCall: {
|
||||
toolName: string;
|
||||
@@ -45,7 +46,16 @@ export const OrchestratorMessagesView = forwardRef<ScrollView, OrchestratorMessa
|
||||
},
|
||||
},
|
||||
});
|
||||
bottomSheetRef.current?.present();
|
||||
setIsBottomSheetVisible(true);
|
||||
// Delay present to next frame to ensure component is mounted
|
||||
setTimeout(() => {
|
||||
bottomSheetRef.current?.present();
|
||||
}, 0);
|
||||
}
|
||||
|
||||
function handleBottomSheetDismiss() {
|
||||
setIsBottomSheetVisible(false);
|
||||
setSelectedToolCall(null);
|
||||
}
|
||||
|
||||
return (
|
||||
@@ -139,10 +149,13 @@ export const OrchestratorMessagesView = forwardRef<ScrollView, OrchestratorMessa
|
||||
)}
|
||||
</ScrollView>
|
||||
|
||||
<ToolCallBottomSheet
|
||||
bottomSheetRef={bottomSheetRef}
|
||||
selectedToolCall={selectedToolCall}
|
||||
/>
|
||||
{isBottomSheetVisible && (
|
||||
<ToolCallBottomSheet
|
||||
bottomSheetRef={bottomSheetRef}
|
||||
selectedToolCall={selectedToolCall}
|
||||
onDismiss={handleBottomSheetDismiss}
|
||||
/>
|
||||
)}
|
||||
</>
|
||||
);
|
||||
}
|
||||
|
||||
@@ -31,13 +31,8 @@ export function RealtimeControls() {
|
||||
isMuted={isMuted}
|
||||
isDetecting={isDetecting}
|
||||
isSpeaking={isSpeaking}
|
||||
orientation="horizontal"
|
||||
/>
|
||||
{/* Debug timer */}
|
||||
{(isDetecting || isSpeaking) && (
|
||||
<Text style={styles.debugTimer}>
|
||||
{(segmentDuration / 1000).toFixed(1)}s
|
||||
</Text>
|
||||
)}
|
||||
</View>
|
||||
<View style={styles.buttons}>
|
||||
{/* Mute button */}
|
||||
@@ -71,20 +66,23 @@ export function RealtimeControls() {
|
||||
|
||||
const styles = StyleSheet.create((theme) => ({
|
||||
container: {
|
||||
minHeight: 200,
|
||||
padding: theme.spacing[4],
|
||||
flexDirection: "row",
|
||||
alignItems: "center",
|
||||
justifyContent: "space-between",
|
||||
paddingHorizontal: theme.spacing[4],
|
||||
paddingVertical: theme.spacing[4],
|
||||
height: 88,
|
||||
},
|
||||
volumeContainer: {
|
||||
flex: 1,
|
||||
justifyContent: "center",
|
||||
alignItems: "center",
|
||||
alignItems: "flex-start",
|
||||
},
|
||||
buttons: {
|
||||
flexDirection: "row",
|
||||
alignItems: "center",
|
||||
justifyContent: "center",
|
||||
gap: theme.spacing[3],
|
||||
paddingTop: theme.spacing[4],
|
||||
},
|
||||
muteButton: {
|
||||
width: 48,
|
||||
@@ -108,10 +106,4 @@ const styles = StyleSheet.create((theme) => ({
|
||||
justifyContent: "center",
|
||||
backgroundColor: theme.colors.palette.red[600],
|
||||
},
|
||||
debugTimer: {
|
||||
marginTop: theme.spacing[2],
|
||||
color: theme.colors.mutedForeground,
|
||||
fontSize: theme.fontSize.sm,
|
||||
fontFamily: "monospace",
|
||||
},
|
||||
}));
|
||||
|
||||
@@ -15,11 +15,13 @@ import type { SelectedToolCall } from "@/types/shared";
|
||||
interface ToolCallBottomSheetProps {
|
||||
bottomSheetRef: React.RefObject<BottomSheetModal | null>;
|
||||
selectedToolCall: SelectedToolCall | null;
|
||||
onDismiss: () => void;
|
||||
}
|
||||
|
||||
export function ToolCallBottomSheet({
|
||||
bottomSheetRef,
|
||||
selectedToolCall,
|
||||
onDismiss,
|
||||
}: ToolCallBottomSheetProps) {
|
||||
const insets = useSafeAreaInsets();
|
||||
const snapPoints = useMemo(() => ["80%"], []);
|
||||
@@ -88,6 +90,7 @@ export function ToolCallBottomSheet({
|
||||
handleIndicatorStyle={styles.handleIndicator}
|
||||
backgroundStyle={styles.background}
|
||||
topInset={insets.top}
|
||||
onDismiss={onDismiss}
|
||||
>
|
||||
{/* Header */}
|
||||
<View style={styles.header}>
|
||||
|
||||
@@ -16,15 +16,16 @@ interface VolumeMeterProps {
|
||||
isMuted?: boolean;
|
||||
isDetecting?: boolean;
|
||||
isSpeaking?: boolean;
|
||||
orientation?: "vertical" | "horizontal";
|
||||
}
|
||||
|
||||
export function VolumeMeter({ volume, isMuted = false, isDetecting = false, isSpeaking = false }: VolumeMeterProps) {
|
||||
export function VolumeMeter({ volume, isMuted = false, isDetecting = false, isSpeaking = false, orientation = "vertical" }: VolumeMeterProps) {
|
||||
const { theme } = useUnistyles();
|
||||
|
||||
// Base dimensions
|
||||
const LINE_SPACING = 8;
|
||||
const MAX_HEIGHT = 50;
|
||||
const MIN_HEIGHT = 20;
|
||||
const MAX_HEIGHT = orientation === "horizontal" ? 30 : 50;
|
||||
const MIN_HEIGHT = orientation === "horizontal" ? 12 : 20;
|
||||
|
||||
// Shared values for each line's height
|
||||
const line1Height = useSharedValue(MIN_HEIGHT);
|
||||
@@ -162,8 +163,10 @@ export function VolumeMeter({ volume, isMuted = false, isDetecting = false, isSp
|
||||
const lineColor = "#FFFFFF";
|
||||
const lineWidth = 8;
|
||||
|
||||
const containerHeight = orientation === "horizontal" ? 60 : 100;
|
||||
|
||||
return (
|
||||
<View style={styles.container}>
|
||||
<View style={[styles.container, { height: containerHeight }]}>
|
||||
<ReanimatedAnimated.View
|
||||
style={[
|
||||
styles.line,
|
||||
@@ -205,7 +208,6 @@ const styles = StyleSheet.create((theme) => ({
|
||||
flexDirection: "row",
|
||||
alignItems: "center",
|
||||
justifyContent: "center",
|
||||
height: 100,
|
||||
},
|
||||
line: {
|
||||
borderRadius: theme.borderRadius.full,
|
||||
|
||||
@@ -0,0 +1,9 @@
|
||||
[
|
||||
{
|
||||
"id": "bcc0019c-9293-4235-adb7-921f8e7d67eb",
|
||||
"title": "Fix Real-time Footer Spacing",
|
||||
"sessionId": "019a1575-706b-73fc-8955-d2c7b524e707",
|
||||
"createdAt": "2025-10-24T09:03:28.577Z",
|
||||
"cwd": "/Users/moboudra/dev/voice-dev"
|
||||
}
|
||||
]
|
||||
@@ -18,6 +18,7 @@
|
||||
"dependencies": {
|
||||
"@agentclientprotocol/sdk": "^0.4.9",
|
||||
"@ai-sdk/openai": "^2.0.52",
|
||||
"@boudra/claude-code-acp": "^0.8.0",
|
||||
"@deepgram/sdk": "^3.4.0",
|
||||
"@modelcontextprotocol/sdk": "^1.20.1",
|
||||
"@openrouter/ai-sdk-provider": "^1.2.0",
|
||||
|
||||
@@ -26,6 +26,7 @@ import type {
|
||||
EnrichedSessionNotification,
|
||||
EnrichedSessionUpdate,
|
||||
} from "./types.js";
|
||||
import { AgentPersistence, type PersistedAgent } from "./agent-persistence.js";
|
||||
|
||||
interface PendingPermission {
|
||||
requestId: string;
|
||||
@@ -107,6 +108,153 @@ class ACPClient implements Client {
|
||||
*/
|
||||
export class AgentManager {
|
||||
private agents = new Map<string, ManagedAgent>();
|
||||
private persistence = new AgentPersistence();
|
||||
|
||||
/**
|
||||
* Initialize the agent manager and load persisted agents
|
||||
*/
|
||||
async initialize(): Promise<void> {
|
||||
console.log("[AgentManager] Initializing and loading persisted agents...");
|
||||
const persistedAgents = await this.persistence.load();
|
||||
|
||||
for (const persistedAgent of persistedAgents) {
|
||||
try {
|
||||
console.log(`[AgentManager] Resuming agent ${persistedAgent.id} with session ${persistedAgent.sessionId}`);
|
||||
await this.resumeAgent(persistedAgent);
|
||||
} catch (error) {
|
||||
console.error(`[AgentManager] Failed to resume agent ${persistedAgent.id}:`, error);
|
||||
// Remove failed agent from persistence
|
||||
await this.persistence.remove(persistedAgent.id);
|
||||
}
|
||||
}
|
||||
|
||||
console.log(`[AgentManager] Loaded ${this.agents.size} agents`);
|
||||
}
|
||||
|
||||
/**
|
||||
* Resume an existing agent from persisted data
|
||||
*/
|
||||
async resumeAgent(persisted: PersistedAgent): Promise<string> {
|
||||
const agentId = persisted.id;
|
||||
const cwd = expandTilde(persisted.cwd);
|
||||
|
||||
// Validate that the working directory exists
|
||||
try {
|
||||
await access(cwd, constants.R_OK | constants.X_OK);
|
||||
} catch (error) {
|
||||
throw new Error(
|
||||
`Working directory does not exist or is not accessible: ${cwd}`
|
||||
);
|
||||
}
|
||||
|
||||
// Spawn the ACP process without any special flags
|
||||
const agentProcess = spawn("npx", ["@boudra/claude-code-acp"], {
|
||||
stdio: ["pipe", "pipe", "pipe"],
|
||||
cwd,
|
||||
});
|
||||
|
||||
// Create streams for ACP communication
|
||||
const input = Writable.toWeb(agentProcess.stdin);
|
||||
const output = Readable.toWeb(agentProcess.stdout);
|
||||
const stream = ndJsonStream(input, output);
|
||||
|
||||
// Create the ACP client and connection
|
||||
const client = new ACPClient(
|
||||
agentId,
|
||||
(id, update) => {
|
||||
this.handleSessionNotification(id, update);
|
||||
},
|
||||
(id, params) => {
|
||||
return this.handlePermissionRequest(id, params);
|
||||
}
|
||||
);
|
||||
const connection = new ClientSideConnection(() => client, stream);
|
||||
|
||||
// Create the managed agent record
|
||||
const agent: ManagedAgent = {
|
||||
id: agentId,
|
||||
status: "initializing",
|
||||
createdAt: new Date(persisted.createdAt),
|
||||
process: agentProcess,
|
||||
connection,
|
||||
subscribers: new Set(),
|
||||
updates: [],
|
||||
cwd,
|
||||
title: persisted.title,
|
||||
titleGenerationTriggered: true, // Already has a title
|
||||
pendingPermissions: new Map(),
|
||||
currentAssistantMessageId: null,
|
||||
currentThoughtId: null,
|
||||
sessionId: persisted.sessionId,
|
||||
};
|
||||
|
||||
this.agents.set(agentId, agent);
|
||||
|
||||
// Handle process errors
|
||||
agentProcess.on("error", (error) => {
|
||||
this.handleAgentError(agentId, `Process error: ${error.message}`);
|
||||
});
|
||||
|
||||
agentProcess.on("exit", (code, signal) => {
|
||||
const agent = this.agents.get(agentId);
|
||||
if (!agent) return;
|
||||
|
||||
if (agent.status !== "completed" && agent.status !== "killed") {
|
||||
this.handleAgentError(
|
||||
agentId,
|
||||
`Process exited unexpectedly: code=${code}, signal=${signal}`
|
||||
);
|
||||
}
|
||||
});
|
||||
|
||||
// Capture stderr for debugging
|
||||
agentProcess.stderr.on("data", (data) => {
|
||||
console.error(`[Agent ${agentId}] stderr:`, data.toString());
|
||||
});
|
||||
|
||||
try {
|
||||
// Initialize the connection
|
||||
await agent.connection.initialize({
|
||||
protocolVersion: PROTOCOL_VERSION,
|
||||
clientCapabilities: {
|
||||
fs: {
|
||||
readTextFile: true,
|
||||
writeTextFile: true,
|
||||
},
|
||||
},
|
||||
});
|
||||
|
||||
// Use the loadSession API to resume the existing session
|
||||
console.log(`[Agent ${agentId}] Loading session ${persisted.sessionId} via loadSession API`);
|
||||
const sessionResponse = await agent.connection.loadSession({
|
||||
sessionId: persisted.sessionId,
|
||||
cwd,
|
||||
mcpServers: [],
|
||||
});
|
||||
|
||||
console.log(`[Agent ${agentId}] Session loaded:`, JSON.stringify(sessionResponse, null, 2));
|
||||
|
||||
// Store session modes from response
|
||||
if (sessionResponse.modes) {
|
||||
agent.currentModeId = sessionResponse.modes.currentModeId;
|
||||
agent.availableModes = sessionResponse.modes.availableModes;
|
||||
console.log(
|
||||
`[Agent ${agentId}] Session loaded with mode: ${agent.currentModeId}`,
|
||||
`Available modes:`, agent.availableModes?.map(m => m.id).join(', ')
|
||||
);
|
||||
}
|
||||
|
||||
agent.status = "ready";
|
||||
|
||||
return agentId;
|
||||
} catch (error) {
|
||||
this.handleAgentError(
|
||||
agentId,
|
||||
`Resume failed: ${error instanceof Error ? error.message : String(error)}`
|
||||
);
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a new agent
|
||||
@@ -126,7 +274,7 @@ export class AgentManager {
|
||||
}
|
||||
|
||||
// Spawn the ACP process
|
||||
const agentProcess = spawn("npx", ["@zed-industries/claude-code-acp"], {
|
||||
const agentProcess = spawn("npx", ["@boudra/claude-code-acp"], {
|
||||
stdio: ["pipe", "pipe", "pipe"],
|
||||
cwd,
|
||||
});
|
||||
@@ -231,6 +379,17 @@ export class AgentManager {
|
||||
|
||||
agent.status = "ready";
|
||||
|
||||
// Persist the new agent
|
||||
if (agent.sessionId) {
|
||||
await this.persistence.upsert({
|
||||
id: agentId,
|
||||
title: agent.title || `Agent ${agentId.slice(0, 8)}`,
|
||||
sessionId: agent.sessionId,
|
||||
createdAt: agent.createdAt.toISOString(),
|
||||
cwd: agent.cwd,
|
||||
});
|
||||
}
|
||||
|
||||
// If an initial prompt was provided, send it
|
||||
if (options.initialPrompt) {
|
||||
console.log(`[Agent ${agentId}] Sending initial prompt`);
|
||||
@@ -433,6 +592,9 @@ export class AgentManager {
|
||||
// This ensures subscribers can still query agent info
|
||||
this.notifySubscribers(agentId);
|
||||
|
||||
// Remove from persistence
|
||||
await this.persistence.remove(agentId);
|
||||
|
||||
// Kill the process
|
||||
agent.process.kill("SIGTERM");
|
||||
|
||||
@@ -717,7 +879,7 @@ export class AgentManager {
|
||||
/**
|
||||
* Set the title for an agent
|
||||
*/
|
||||
setAgentTitle(agentId: string, title: string): void {
|
||||
async setAgentTitle(agentId: string, title: string): Promise<void> {
|
||||
const agent = this.agents.get(agentId);
|
||||
if (!agent) {
|
||||
throw new Error(`Agent ${agentId} not found`);
|
||||
@@ -725,6 +887,11 @@ export class AgentManager {
|
||||
|
||||
agent.title = title;
|
||||
console.log(`[Agent ${agentId}] Title set to: "${title}"`);
|
||||
|
||||
// Persist the title update
|
||||
if (agent.sessionId) {
|
||||
await this.persistence.updateTitle(agentId, title);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
|
||||
@@ -0,0 +1,100 @@
|
||||
import { promises as fs } from "fs";
|
||||
import path from "path";
|
||||
import { fileURLToPath } from "url";
|
||||
import { dirname } from "path";
|
||||
|
||||
const __filename = fileURLToPath(import.meta.url);
|
||||
const __dirname = dirname(__filename);
|
||||
|
||||
export interface PersistedAgent {
|
||||
id: string;
|
||||
title: string;
|
||||
sessionId: string;
|
||||
createdAt: string;
|
||||
cwd: string;
|
||||
}
|
||||
|
||||
export class AgentPersistence {
|
||||
private persistencePath: string;
|
||||
|
||||
constructor() {
|
||||
// Store agents.json in the project root
|
||||
this.persistencePath = path.join(__dirname, "../../../agents.json");
|
||||
}
|
||||
|
||||
/**
|
||||
* Load all persisted agents
|
||||
*/
|
||||
async load(): Promise<PersistedAgent[]> {
|
||||
try {
|
||||
const data = await fs.readFile(this.persistencePath, "utf-8");
|
||||
return JSON.parse(data);
|
||||
} catch (error) {
|
||||
// File doesn't exist or is invalid, return empty array
|
||||
console.log("[AgentPersistence] No existing agents found or file is invalid");
|
||||
return [];
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Save agents to disk
|
||||
*/
|
||||
async save(agents: PersistedAgent[]): Promise<void> {
|
||||
try {
|
||||
await fs.writeFile(
|
||||
this.persistencePath,
|
||||
JSON.stringify(agents, null, 2),
|
||||
"utf-8"
|
||||
);
|
||||
console.log(`[AgentPersistence] Saved ${agents.length} agents to disk`);
|
||||
} catch (error) {
|
||||
console.error("[AgentPersistence] Failed to save agents:", error);
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Add or update an agent
|
||||
*/
|
||||
async upsert(agent: PersistedAgent): Promise<void> {
|
||||
const agents = await this.load();
|
||||
const existingIndex = agents.findIndex((a) => a.id === agent.id);
|
||||
|
||||
if (existingIndex >= 0) {
|
||||
agents[existingIndex] = agent;
|
||||
console.log(`[AgentPersistence] Updated agent ${agent.id}`);
|
||||
} else {
|
||||
agents.push(agent);
|
||||
console.log(`[AgentPersistence] Added new agent ${agent.id}`);
|
||||
}
|
||||
|
||||
await this.save(agents);
|
||||
}
|
||||
|
||||
/**
|
||||
* Remove an agent
|
||||
*/
|
||||
async remove(agentId: string): Promise<void> {
|
||||
const agents = await this.load();
|
||||
const filtered = agents.filter((a) => a.id !== agentId);
|
||||
|
||||
if (filtered.length < agents.length) {
|
||||
await this.save(filtered);
|
||||
console.log(`[AgentPersistence] Removed agent ${agentId}`);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Update agent title
|
||||
*/
|
||||
async updateTitle(agentId: string, title: string): Promise<void> {
|
||||
const agents = await this.load();
|
||||
const agent = agents.find((a) => a.id === agentId);
|
||||
|
||||
if (agent) {
|
||||
agent.title = title;
|
||||
await this.save(agents);
|
||||
console.log(`[AgentPersistence] Updated title for agent ${agentId}: "${title}"`);
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -60,7 +60,7 @@ function createServer() {
|
||||
return app;
|
||||
}
|
||||
|
||||
function main() {
|
||||
async function main() {
|
||||
const port = parseInt(process.env.PORT || "6767", 10);
|
||||
|
||||
const app = createServer();
|
||||
@@ -68,7 +68,8 @@ function main() {
|
||||
|
||||
// Initialize global agent manager
|
||||
const agentManager = new AgentManager();
|
||||
console.log("✓ Global agent manager initialized");
|
||||
await agentManager.initialize(); // Load persisted agents
|
||||
console.log("✓ Global agent manager initialized with persisted agents");
|
||||
|
||||
// Initialize WebSocket server
|
||||
const wsServer = new VoiceAssistantWebSocketServer(httpServer, agentManager);
|
||||
|
||||
Reference in New Issue
Block a user