116 lines
4.2 KiB
JavaScript
116 lines
4.2 KiB
JavaScript
import { createOpencode, createOpencodeClient, } from "@opencode-ai/sdk/v2";
|
||
import { config } from "../config.js";
|
||
import { logger } from "../logger.js";
|
||
export class OpencodeRuntimeAdapter {
|
||
clientPromise = null;
|
||
closeServer = null;
|
||
async ensureClient() {
|
||
if (!this.clientPromise) {
|
||
this.clientPromise = this.bootstrapClient();
|
||
}
|
||
return this.clientPromise;
|
||
}
|
||
async health() {
|
||
const client = await this.ensureClient();
|
||
const response = await client.global.health();
|
||
return requireData(response.data, "global.health");
|
||
}
|
||
async createSession(title) {
|
||
const client = await this.ensureClient();
|
||
const response = await client.session.create({
|
||
title,
|
||
});
|
||
return requireData(response.data, "session.create");
|
||
}
|
||
async getSession(id) {
|
||
const client = await this.ensureClient();
|
||
const response = await client.session.get({
|
||
sessionID: id,
|
||
});
|
||
return requireData(response.data, "session.get");
|
||
}
|
||
async sendPrompt(sessionId, text) {
|
||
await this.prompt(sessionId, text);
|
||
// 当前 SDK 响应风格下，prompt() 本身不会直接返回完整 assistant parts，
|
||
// 所以这里紧跟一次 messages() 回读，给上层路由统一消费。
|
||
return this.messages(sessionId);
|
||
}
|
||
async prompt(sessionId, text) {
|
||
const client = await this.ensureClient();
|
||
await client.session.prompt({
|
||
sessionID: sessionId,
|
||
parts: [{ type: "text", text }],
|
||
});
|
||
}
|
||
async messages(sessionId, limit = 20) {
|
||
const client = await this.ensureClient();
|
||
const messages = await client.session.messages({
|
||
sessionID: sessionId,
|
||
limit,
|
||
});
|
||
return requireData(messages.data, "session.messages");
|
||
}
|
||
async forkSession(sessionId, messageId) {
|
||
const client = await this.ensureClient();
|
||
const response = await client.session.fork({
|
||
sessionID: sessionId,
|
||
messageID: messageId,
|
||
});
|
||
return requireData(response.data, "session.fork");
|
||
}
|
||
async abortSession(sessionId) {
|
||
const client = await this.ensureClient();
|
||
const response = await client.session.abort({
|
||
sessionID: sessionId,
|
||
});
|
||
return requireData(response.data, "session.abort");
|
||
}
|
||
async subscribeEvents() {
|
||
const client = await this.ensureClient();
|
||
const response = await client.event.subscribe();
|
||
return response.stream;
|
||
}
|
||
async dispose() {
|
||
this.closeServer?.();
|
||
this.closeServer = null;
|
||
this.clientPromise = null;
|
||
}
|
||
async bootstrapClient() {
|
||
if (config.OPENCODE_BASE_URL) {
|
||
logger.info({ baseUrl: config.OPENCODE_BASE_URL }, "connecting to external opencode server");
|
||
return createOpencodeClient({
|
||
baseUrl: config.OPENCODE_BASE_URL,
|
||
});
|
||
}
|
||
// embedded 模式下，把服务内工具桥地址注入到 opencode 进程环境里，
|
||
// 这样 .opencode/tools 下的自定义工具可以回调本服务。
|
||
process.env.TJWATER_AGENT_INTERNAL_BASE_URL = `http://127.0.0.1:${config.PORT}`;
|
||
process.env.TJWATER_AGENT_INTERNAL_TOKEN =
|
||
config.AGENT_INTERNAL_TOKEN ?? process.env.TJWATER_AGENT_INTERNAL_TOKEN ?? "";
|
||
logger.info({
|
||
hostname: config.OPENCODE_HOSTNAME,
|
||
port: config.OPENCODE_PORT,
|
||
model: config.OPENCODE_MODEL,
|
||
}, "starting embedded opencode server");
|
||
const runtime = await createOpencode({
|
||
hostname: config.OPENCODE_HOSTNAME,
|
||
port: config.OPENCODE_PORT,
|
||
timeout: config.OPENCODE_TIMEOUT_MS,
|
||
config: {
|
||
model: config.OPENCODE_MODEL,
|
||
},
|
||
});
|
||
this.closeServer = () => {
|
||
runtime.server.close();
|
||
};
|
||
return runtime.client;
|
||
}
|
||
}
|
||
export const opencodeRuntime = new OpencodeRuntimeAdapter();
|
||
function requireData(data, operation) {
|
||
if (data === undefined) {
|
||
throw new Error(`${operation} returned no data`);
|
||
}
|
||
return data;
|
||
}
|