-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathopencode.ts
More file actions
154 lines (132 loc) · 5.03 KB
/
Copy pathopencode.ts
File metadata and controls
154 lines (132 loc) · 5.03 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
import { createOpencode, OpencodeClient } from '@opencode-ai/sdk';
import type { Task, TaskResult } from './types.js';
import { Router } from './router.js';
import { logger } from './app/logger.js';
export interface AuthStatus {
authenticated: boolean;
connected: string[];
defaults: Record<string, string>;
/** True when the opencode binary is missing from PATH (affects CLI fallback only). */
binaryMissing: boolean;
}
export class OpencodeOrchestrator {
public client: OpencodeClient;
private serverCloseCallback: () => void;
private router: Router;
private constructor(client: OpencodeClient, serverCloseCallback: () => void, router?: Router) {
this.client = client;
this.serverCloseCallback = serverCloseCallback;
this.router = router || new Router();
}
static async initialize(router?: Router, mockClient?: any, mockServer?: any): Promise<OpencodeOrchestrator> {
let client = mockClient;
let server = mockServer;
if (!client || !server) {
const res = await createOpencode();
client = client ?? res.client;
server = server ?? res.server;
}
const orchestrator = new OpencodeOrchestrator(client, () => server?.close(), router);
// NOTE: no interactive auth prompting here. Callers use getAuthStatus() and drive the UI.
return orchestrator;
}
async getAuthStatus(): Promise<AuthStatus> {
try {
const { data } = await this.client.provider.list();
const connected = data?.connected ?? [];
const defaults = data?.default ?? {};
const envSatisfied = (data?.all ?? []).some((p) => (p.env ?? []).some((v) => Boolean(process.env[v])));
return {
authenticated: connected.length > 0 || envSatisfied,
connected,
defaults,
binaryMissing: false,
};
} catch (err) {
logger.warn('opencode', `Could not read provider status: ${(err as Error).message}`);
return { authenticated: false, connected: [], defaults: {}, binaryMissing: false };
}
}
async refreshRouterFromProviders(router: Router = this.router): Promise<void> {
try {
const { data: config } = await this.client.config.providers();
if (config && config.providers) {
const readyProviders = config.providers
.filter((p: any) => p.state === 'ready' || p.configured)
.map((p: any) => p.id);
router.updateModelsBasedOnProviders(readyProviders, config.providers, config.default);
}
} catch (err) {
// Ignore provider resolution errors to ensure fallback default routing is preserved
}
}
async executeTask(task: Task): Promise<TaskResult> {
const { data: session, error: createError } = await this.client.session.create({
body: { title: `LoopCode Task: ${task.description}` },
});
if (!session || createError) {
throw new Error('Failed to create OpenCode session: ' + JSON.stringify(createError));
}
const sessionId = session.id;
// Start streaming events in the background for transparency
const { stream } = await this.client.event.subscribe();
// Asynchronous loop to print tool calls
(async () => {
try {
for await (const event of stream) {
if ((event as any).type === 'tool_call' || (event as any).event === 'tool_call') {
// Simplified rendering of tool events
console.log(`[Tool] ${JSON.stringify((event as any).data || (event as any).properties || event)}`);
}
}
} catch (err) {
// Stream may close abruptly
}
})();
// Fold the system prompt/instructions into the main prompt
const fullPrompt = `${task.systemPrompt}\n\nTask Goal:\n${task.goal}`;
// Setup Timeout
const abortController = new AbortController();
const timeoutPromise = new Promise<never>((_, reject) => {
setTimeout(() => {
abortController.abort();
reject(new Error(`Task timed out after ${task.timeout} seconds`));
}, task.timeout * 1000);
});
try {
const modelRoute = this.router.route(task);
const promptPromise = this.client.session.prompt({
path: { id: sessionId },
body: {
parts: [{ type: 'text', text: fullPrompt }],
model: modelRoute,
},
signal: abortController.signal,
});
const { data: result, error: promptError } = await Promise.race([promptPromise, timeoutPromise]);
if (promptError) {
throw new Error('Prompt error: ' + JSON.stringify(promptError));
}
return {
success: true,
message: (result?.info as any)?.text,
};
} catch (error: any) {
// If it timed out, try to gracefully abort the session on the server
if (error.message.includes('timed out')) {
try {
await this.client.session.abort({ path: { id: sessionId } });
} catch (abortErr) {
console.error('Failed to abort session on server:', abortErr);
}
}
return {
success: false,
message: error.message || String(error),
};
}
}
close() {
this.serverCloseCallback();
}
}