-
Notifications
You must be signed in to change notification settings - Fork 35
/
runDeveloperAgent.ts
98 lines (93 loc) · 2.89 KB
/
runDeveloperAgent.ts
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
import * as $ from "js-agent";
export async function runDeveloperAgent({
openAiApiKey,
task,
projectInstructions,
setupCommands,
}: {
openAiApiKey: string;
task: string;
projectInstructions: string;
setupCommands: string[];
}) {
const model = $.provider.openai.chatModel({
apiKey: openAiApiKey,
model: "gpt-4",
});
const executeRemote = $.tool.executeRemoteTool({
baseUrl: "http://localhost:3001",
}) as any;
return $.runAgent({
properties: { task, projectInstructions },
agent: $.step.createFixedStepsLoop({
steps: [
$.step.createFixedStepsLoop({
type: "setup",
steps: setupCommands.map(
(command) => async (run) =>
$.step.createActionStep({
action: $.tool.runCommand({ execute: executeRemote }),
input: { command },
run,
})
),
}),
$.step.generateNextStepLoop({
actions: [
$.tool.readFile({ execute: executeRemote }),
$.tool.writeFile({ execute: executeRemote }),
$.tool.runCommand({ execute: executeRemote }),
$.tool.askUser({
execute: $.tool.executeAskUser(),
}),
],
actionFormat: $.action.format.json(),
prompt: $.prompt.concatChatPrompts(
$.prompt.sectionsChatPrompt({
role: "system",
getSections: async ({ runState: { projectInstructions } }) => [
{
title: "role",
content: `You are a software developer that creates and modifies JavaScript programs.
You are working in a Linux environment.`,
},
{ title: "project", content: projectInstructions },
{
title: "constraints",
content: `You must verify that the changes that you make are working.`,
},
],
}),
$.prompt.availableActionsChatPrompt(),
$.prompt.sectionsChatPrompt({
role: "user",
getSections: async ({ runState: { task } }) => [
{ title: "Task", content: task },
],
}),
$.prompt.recentStepsChatPrompt({ maxSteps: 10 })
),
model,
}),
],
}),
observer: $.agent.observer.combineObservers(
$.agent.observer.showRunInConsole({ name: "JavaScript Developer Agent" }),
{
async onRunFinished({ run }) {
const runCostInMillicent = await $.agent.calculateRunCostInMillicent({
run,
});
console.log(
`Run cost: $${(runCostInMillicent / 1000 / 100).toFixed(2)}`
);
console.log(
`LLM calls: ${
run.recordedCalls.filter((call) => call.success).length
}`
);
},
}
),
});
}