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
|
import "zod-openapi/extend";
import { App } from "./app";
import { Server } from "./server/server";
import fs from "fs/promises";
import path from "path";
import { Bus } from "./bus";
import { Session } from "./session/session";
import cac from "cac";
import { Share } from "./share/share";
import { Storage } from "./storage/storage";
import { LLM } from "./llm/llm";
const cli = cac("opencode");
cli.command("", "Start the opencode in interactive mode").action(async () => {
await App.provide({ directory: process.cwd() }, async () => {
await Share.init();
Server.listen();
});
});
cli.command("generate", "Generate OpenAPI and event specs").action(async () => {
const specs = await Server.openapi();
const dir = "gen";
await fs.rmdir(dir, { recursive: true }).catch(() => {});
await fs.mkdir(dir, { recursive: true });
await Bun.write(
path.join(dir, "openapi.json"),
JSON.stringify(specs, null, 2),
);
await Bun.write(
path.join(dir, "event.json"),
JSON.stringify(Bus.specs(), null, 2),
);
});
cli
.command("run [...message]", "Run a chat message")
.action(async (message: string[]) => {
await App.provide({ directory: process.cwd() }, async () => {
console.log("Thinking...");
await Share.init();
const session = await Session.create();
console.log(
`Share ID: ${Share.URL.replace("api.", "")}/share?id=${session.id}`,
);
let index = 0;
Bus.subscribe(Storage.Event.Write, async (payload) => {
const [root, , type, messageID] = payload.properties.key.split("/");
if (root !== "session" && type !== "message") return;
const message = await Session.messages(session.id).then((x) =>
x.find((x) => x.id === messageID),
);
if (!message) return;
for (; index < message.parts.length; index++) {
const part = message.parts[index];
if (part.type === "text") continue;
if (part.type === "step-start") continue;
if (
part.type === "tool-invocation" &&
part.toolInvocation.state !== "result"
)
break;
if (part.type === "tool-invocation") {
console.log(`🔧 ${part.toolInvocation.toolName}`);
if (
part.toolInvocation.state === "result" &&
"result" in part.toolInvocation
) {
const result = part.toolInvocation.result;
if (typeof result === "string") {
const lines = result.split("\n");
const truncated = lines.slice(0, 4);
if (lines.length > 4) truncated.push("...");
console.log(truncated.join("\n"));
} else if (result && typeof result === "object") {
const jsonStr = JSON.stringify(result, null, 2);
const lines = jsonStr.split("\n");
const truncated = lines.slice(0, 4);
if (lines.length > 4) truncated.push("...");
console.log(truncated.join("\n"));
}
}
continue;
}
console.log(part);
}
});
const providers = await LLM.providers();
const providerID = Object.keys(providers)[0];
const modelID = Object.keys(providers[providerID].info.models!)[0];
console.log("using", providerID, modelID);
const result = await Session.chat({
sessionID: session.id,
providerID,
modelID,
parts: [
{
type: "text",
text: message.join(" "),
},
],
});
for (const part of result.parts) {
if (part.type === "text") {
console.log("opencode:", part.text);
}
}
});
});
cli.help();
cli.version("1.0.0");
cli.parse();
|
