daycare-cli 2026.2.26 → 2026.2.27
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/engine/agents/agent.d.ts +2 -0
- package/dist/engine/agents/agent.d.ts.map +1 -1
- package/dist/engine/agents/agent.js +7 -3
- package/dist/engine/agents/agent.js.map +1 -1
- package/dist/engine/agents/agentSystem.d.ts +5 -0
- package/dist/engine/agents/agentSystem.d.ts.map +1 -1
- package/dist/engine/agents/agentSystem.js +16 -1
- package/dist/engine/agents/agentSystem.js.map +1 -1
- package/dist/engine/agents/agentSystem.spec.js +59 -0
- package/dist/engine/agents/agentSystem.spec.js.map +1 -1
- package/dist/engine/agents/ops/agentLoopRun.d.ts +0 -2
- package/dist/engine/agents/ops/agentLoopRun.d.ts.map +1 -1
- package/dist/engine/agents/ops/agentLoopRun.js +4 -7
- package/dist/engine/agents/ops/agentLoopRun.js.map +1 -1
- package/dist/engine/agents/ops/agentLoopRun.spec.js +21 -24
- package/dist/engine/agents/ops/agentLoopRun.spec.js.map +1 -1
- package/dist/engine/apps/appExecute.spec.js +2 -10
- package/dist/engine/apps/appExecute.spec.js.map +1 -1
- package/dist/engine/apps/appInstallToolBuild.js +1 -1
- package/dist/engine/apps/appInstallToolBuild.js.map +1 -1
- package/dist/engine/apps/appInstallToolBuild.spec.js +1 -5
- package/dist/engine/apps/appInstallToolBuild.spec.js.map +1 -1
- package/dist/engine/apps/appRuleToolBuild.spec.js +1 -5
- package/dist/engine/apps/appRuleToolBuild.spec.js.map +1 -1
- package/dist/engine/apps/appToolExecutorBuild.spec.js +1 -5
- package/dist/engine/apps/appToolExecutorBuild.spec.js.map +1 -1
- package/dist/engine/engine.d.ts.map +1 -1
- package/dist/engine/engine.js +10 -0
- package/dist/engine/engine.js.map +1 -1
- package/dist/engine/friends/usertagGenerate.d.ts +6 -0
- package/dist/engine/friends/usertagGenerate.d.ts.map +1 -0
- package/dist/engine/friends/usertagGenerate.js +311 -0
- package/dist/engine/friends/usertagGenerate.js.map +1 -0
- package/dist/engine/friends/usertagGenerate.spec.d.ts +2 -0
- package/dist/engine/friends/usertagGenerate.spec.d.ts.map +1 -0
- package/dist/engine/friends/usertagGenerate.spec.js +13 -0
- package/dist/engine/friends/usertagGenerate.spec.js.map +1 -0
- package/dist/engine/modules/executablePrompts/executablePromptExpand.spec.js +1 -5
- package/dist/engine/modules/executablePrompts/executablePromptExpand.spec.js.map +1 -1
- package/dist/engine/modules/monty/montyPythonTypeFromSchemaRuntime.spec.js +1 -5
- package/dist/engine/modules/monty/montyPythonTypeFromSchemaRuntime.spec.js.map +1 -1
- package/dist/engine/modules/rlm/rlmExecute.spec.js +1 -5
- package/dist/engine/modules/rlm/rlmExecute.spec.js.map +1 -1
- package/dist/engine/modules/rlm/rlmRestore.spec.js +1 -5
- package/dist/engine/modules/rlm/rlmRestore.spec.js.map +1 -1
- package/dist/engine/modules/rlm/rlmTool.spec.js +1 -5
- package/dist/engine/modules/rlm/rlmTool.spec.js.map +1 -1
- package/dist/engine/modules/say/sayFileResolve.d.ts +4 -5
- package/dist/engine/modules/say/sayFileResolve.d.ts.map +1 -1
- package/dist/engine/modules/say/sayFileResolve.js +24 -14
- package/dist/engine/modules/say/sayFileResolve.js.map +1 -1
- package/dist/engine/modules/say/sayFileResolve.spec.js +28 -29
- package/dist/engine/modules/say/sayFileResolve.spec.js.map +1 -1
- package/dist/engine/modules/toolResolver.spec.js +1 -5
- package/dist/engine/modules/toolResolver.spec.js.map +1 -1
- package/dist/engine/modules/tools/agentCompactTool.spec.js +1 -5
- package/dist/engine/modules/tools/agentCompactTool.spec.js.map +1 -1
- package/dist/engine/modules/tools/agentResetTool.spec.js +1 -5
- package/dist/engine/modules/tools/agentResetTool.spec.js.map +1 -1
- package/dist/engine/modules/tools/background.spec.js +3 -11
- package/dist/engine/modules/tools/background.spec.js.map +1 -1
- package/dist/engine/modules/tools/channelCreateTool.spec.js +1 -5
- package/dist/engine/modules/tools/channelCreateTool.spec.js.map +1 -1
- package/dist/engine/modules/tools/channelHistoryTool.spec.js +1 -5
- package/dist/engine/modules/tools/channelHistoryTool.spec.js.map +1 -1
- package/dist/engine/modules/tools/channelMemberTool.spec.js +1 -5
- package/dist/engine/modules/tools/channelMemberTool.spec.js.map +1 -1
- package/dist/engine/modules/tools/channelSendTool.spec.js +1 -5
- package/dist/engine/modules/tools/channelSendTool.spec.js.map +1 -1
- package/dist/engine/modules/tools/friendAddToolBuild.d.ts +7 -0
- package/dist/engine/modules/tools/friendAddToolBuild.d.ts.map +1 -0
- package/dist/engine/modules/tools/friendAddToolBuild.js +163 -0
- package/dist/engine/modules/tools/friendAddToolBuild.js.map +1 -0
- package/dist/engine/modules/tools/friendAddToolBuild.spec.d.ts +2 -0
- package/dist/engine/modules/tools/friendAddToolBuild.spec.d.ts.map +1 -0
- package/dist/engine/modules/tools/friendAddToolBuild.spec.js +151 -0
- package/dist/engine/modules/tools/friendAddToolBuild.spec.js.map +1 -0
- package/dist/engine/modules/tools/friendRemoveToolBuild.d.ts +7 -0
- package/dist/engine/modules/tools/friendRemoveToolBuild.d.ts.map +1 -0
- package/dist/engine/modules/tools/friendRemoveToolBuild.js +174 -0
- package/dist/engine/modules/tools/friendRemoveToolBuild.js.map +1 -0
- package/dist/engine/modules/tools/friendRemoveToolBuild.spec.d.ts +2 -0
- package/dist/engine/modules/tools/friendRemoveToolBuild.spec.d.ts.map +1 -0
- package/dist/engine/modules/tools/friendRemoveToolBuild.spec.js +172 -0
- package/dist/engine/modules/tools/friendRemoveToolBuild.spec.js.map +1 -0
- package/dist/engine/modules/tools/friendSendToolBuild.d.ts +7 -0
- package/dist/engine/modules/tools/friendSendToolBuild.d.ts.map +1 -0
- package/dist/engine/modules/tools/friendSendToolBuild.js +104 -0
- package/dist/engine/modules/tools/friendSendToolBuild.js.map +1 -0
- package/dist/engine/modules/tools/friendSendToolBuild.spec.d.ts +2 -0
- package/dist/engine/modules/tools/friendSendToolBuild.spec.d.ts.map +1 -0
- package/dist/engine/modules/tools/friendSendToolBuild.spec.js +120 -0
- package/dist/engine/modules/tools/friendSendToolBuild.spec.js.map +1 -0
- package/dist/engine/modules/tools/friendShareSubuserToolBuild.d.ts +7 -0
- package/dist/engine/modules/tools/friendShareSubuserToolBuild.d.ts.map +1 -0
- package/dist/engine/modules/tools/friendShareSubuserToolBuild.js +133 -0
- package/dist/engine/modules/tools/friendShareSubuserToolBuild.js.map +1 -0
- package/dist/engine/modules/tools/friendShareSubuserToolBuild.spec.d.ts +2 -0
- package/dist/engine/modules/tools/friendShareSubuserToolBuild.spec.d.ts.map +1 -0
- package/dist/engine/modules/tools/friendShareSubuserToolBuild.spec.js +119 -0
- package/dist/engine/modules/tools/friendShareSubuserToolBuild.spec.js.map +1 -0
- package/dist/engine/modules/tools/friendUnshareSubuserToolBuild.d.ts +7 -0
- package/dist/engine/modules/tools/friendUnshareSubuserToolBuild.d.ts.map +1 -0
- package/dist/engine/modules/tools/friendUnshareSubuserToolBuild.js +118 -0
- package/dist/engine/modules/tools/friendUnshareSubuserToolBuild.js.map +1 -0
- package/dist/engine/modules/tools/friendUnshareSubuserToolBuild.spec.d.ts +2 -0
- package/dist/engine/modules/tools/friendUnshareSubuserToolBuild.spec.d.ts.map +1 -0
- package/dist/engine/modules/tools/friendUnshareSubuserToolBuild.spec.js +100 -0
- package/dist/engine/modules/tools/friendUnshareSubuserToolBuild.spec.js.map +1 -0
- package/dist/engine/modules/tools/image-generation.d.ts.map +1 -1
- package/dist/engine/modules/tools/image-generation.js +26 -10
- package/dist/engine/modules/tools/image-generation.js.map +1 -1
- package/dist/engine/modules/tools/image-generation.spec.js +13 -10
- package/dist/engine/modules/tools/image-generation.spec.js.map +1 -1
- package/dist/engine/modules/tools/mermaid-png.d.ts.map +1 -1
- package/dist/engine/modules/tools/mermaid-png.js +16 -11
- package/dist/engine/modules/tools/mermaid-png.js.map +1 -1
- package/dist/engine/modules/tools/mermaid-png.spec.js +20 -24
- package/dist/engine/modules/tools/mermaid-png.spec.js.map +1 -1
- package/dist/engine/modules/tools/pdf-process.d.ts.map +1 -1
- package/dist/engine/modules/tools/pdf-process.js +10 -38
- package/dist/engine/modules/tools/pdf-process.js.map +1 -1
- package/dist/engine/modules/tools/pdf-process.spec.js +15 -5
- package/dist/engine/modules/tools/pdf-process.spec.js.map +1 -1
- package/dist/engine/modules/tools/permanentAgentToolBuild.spec.js +5 -13
- package/dist/engine/modules/tools/permanentAgentToolBuild.spec.js.map +1 -1
- package/dist/engine/modules/tools/send-file.d.ts.map +1 -1
- package/dist/engine/modules/tools/send-file.js +20 -11
- package/dist/engine/modules/tools/send-file.js.map +1 -1
- package/dist/engine/modules/tools/sendUserMessageTool.spec.js +1 -5
- package/dist/engine/modules/tools/sendUserMessageTool.spec.js.map +1 -1
- package/dist/engine/modules/tools/sessionHistoryToolBuild.spec.js +1 -2
- package/dist/engine/modules/tools/sessionHistoryToolBuild.spec.js.map +1 -1
- package/dist/engine/modules/tools/signal.spec.js +1 -5
- package/dist/engine/modules/tools/signal.spec.js.map +1 -1
- package/dist/engine/modules/tools/signalEventsCsvToolBuild.spec.js +1 -5
- package/dist/engine/modules/tools/signalEventsCsvToolBuild.spec.js.map +1 -1
- package/dist/engine/modules/tools/signalSubscribeToolBuild.spec.js +1 -5
- package/dist/engine/modules/tools/signalSubscribeToolBuild.spec.js.map +1 -1
- package/dist/engine/modules/tools/signalUnsubscribeToolBuild.spec.js +1 -5
- package/dist/engine/modules/tools/signalUnsubscribeToolBuild.spec.js.map +1 -1
- package/dist/engine/modules/tools/skillToolBuild.js +3 -3
- package/dist/engine/modules/tools/skillToolBuild.js.map +1 -1
- package/dist/engine/modules/tools/skillToolBuild.spec.js +12 -9
- package/dist/engine/modules/tools/skillToolBuild.spec.js.map +1 -1
- package/dist/engine/modules/tools/subuserConfigureToolBuild.spec.js +1 -2
- package/dist/engine/modules/tools/subuserConfigureToolBuild.spec.js.map +1 -1
- package/dist/engine/modules/tools/subuserCreateToolBuild.spec.js +1 -2
- package/dist/engine/modules/tools/subuserCreateToolBuild.spec.js.map +1 -1
- package/dist/engine/modules/tools/subuserListToolBuild.spec.js +1 -2
- package/dist/engine/modules/tools/subuserListToolBuild.spec.js.map +1 -1
- package/dist/engine/modules/tools/topologyToolBuild.d.ts.map +1 -1
- package/dist/engine/modules/tools/topologyToolBuild.js +111 -2
- package/dist/engine/modules/tools/topologyToolBuild.js.map +1 -1
- package/dist/engine/modules/tools/topologyToolBuild.spec.js +99 -2
- package/dist/engine/modules/tools/topologyToolBuild.spec.js.map +1 -1
- package/dist/engine/modules/tools/types.d.ts +3 -4
- package/dist/engine/modules/tools/types.d.ts.map +1 -1
- package/dist/plugins/dashboard/site/404.html +1 -1
- package/dist/plugins/dashboard/site/agent.html +1 -1
- package/dist/plugins/dashboard/site/agent.txt +1 -1
- package/dist/plugins/dashboard/site/agents.html +1 -1
- package/dist/plugins/dashboard/site/agents.txt +1 -1
- package/dist/plugins/dashboard/site/automations.html +1 -1
- package/dist/plugins/dashboard/site/automations.txt +1 -1
- package/dist/plugins/dashboard/site/connectors.html +1 -1
- package/dist/plugins/dashboard/site/connectors.txt +1 -1
- package/dist/plugins/dashboard/site/index.html +1 -1
- package/dist/plugins/dashboard/site/index.txt +1 -1
- package/dist/plugins/dashboard/site/memory.html +1 -1
- package/dist/plugins/dashboard/site/memory.txt +1 -1
- package/dist/plugins/dashboard/site/processes.html +1 -1
- package/dist/plugins/dashboard/site/processes.txt +1 -1
- package/dist/plugins/dashboard/site/providers.html +1 -1
- package/dist/plugins/dashboard/site/providers.txt +1 -1
- package/dist/plugins/dashboard/site/signals.html +1 -1
- package/dist/plugins/dashboard/site/signals.txt +1 -1
- package/dist/plugins/dashboard/site/telemetry.html +1 -1
- package/dist/plugins/dashboard/site/telemetry.txt +1 -1
- package/dist/plugins/dashboard/site/tools.html +1 -1
- package/dist/plugins/dashboard/site/tools.txt +1 -1
- package/dist/plugins/database/__tests__/plugin.spec.js +4 -2
- package/dist/plugins/database/__tests__/plugin.spec.js.map +1 -1
- package/dist/plugins/monty-python/tool.spec.js +1 -2
- package/dist/plugins/monty-python/tool.spec.js.map +1 -1
- package/dist/plugins/shell/processTools.js +1 -1
- package/dist/plugins/shell/processTools.js.map +1 -1
- package/dist/plugins/shell/processTools.spec.js +2 -3
- package/dist/plugins/shell/processTools.spec.js.map +1 -1
- package/dist/plugins/shell/tool.d.ts.map +1 -1
- package/dist/plugins/shell/tool.js +90 -486
- package/dist/plugins/shell/tool.js.map +1 -1
- package/dist/plugins/shell/tool.spec.js +11 -20
- package/dist/plugins/shell/tool.spec.js.map +1 -1
- package/dist/sandbox/sandbox.d.ts +26 -0
- package/dist/sandbox/sandbox.d.ts.map +1 -0
- package/dist/sandbox/sandbox.js +466 -0
- package/dist/sandbox/sandbox.js.map +1 -0
- package/dist/sandbox/sandbox.spec.d.ts +2 -0
- package/dist/sandbox/sandbox.spec.d.ts.map +1 -0
- package/dist/sandbox/sandbox.spec.js +167 -0
- package/dist/sandbox/sandbox.spec.js.map +1 -0
- package/dist/sandbox/sandboxCanRead.d.ts.map +1 -1
- package/dist/sandbox/sandboxCanRead.js +4 -8
- package/dist/sandbox/sandboxCanRead.js.map +1 -1
- package/dist/sandbox/sandboxCanRead.spec.js +17 -11
- package/dist/sandbox/sandboxCanRead.spec.js.map +1 -1
- package/dist/sandbox/sandboxFilesystemPolicyBuild.d.ts +1 -1
- package/dist/sandbox/sandboxFilesystemPolicyBuild.d.ts.map +1 -1
- package/dist/sandbox/sandboxFilesystemPolicyBuild.js +14 -6
- package/dist/sandbox/sandboxFilesystemPolicyBuild.js.map +1 -1
- package/dist/sandbox/sandboxFilesystemPolicyBuild.spec.js +7 -2
- package/dist/sandbox/sandboxFilesystemPolicyBuild.spec.js.map +1 -1
- package/dist/sandbox/sandboxReadDenyPathsBuild.d.ts +13 -0
- package/dist/sandbox/sandboxReadDenyPathsBuild.d.ts.map +1 -0
- package/dist/sandbox/sandboxReadDenyPathsBuild.js +28 -0
- package/dist/sandbox/sandboxReadDenyPathsBuild.js.map +1 -0
- package/dist/sandbox/sandboxReadDenyPathsBuild.spec.d.ts +2 -0
- package/dist/sandbox/sandboxReadDenyPathsBuild.spec.d.ts.map +1 -0
- package/dist/sandbox/sandboxReadDenyPathsBuild.spec.js +29 -0
- package/dist/sandbox/sandboxReadDenyPathsBuild.spec.js.map +1 -0
- package/dist/sandbox/sandboxTypes.d.ts +67 -0
- package/dist/sandbox/sandboxTypes.d.ts.map +1 -0
- package/dist/sandbox/sandboxTypes.js +2 -0
- package/dist/sandbox/sandboxTypes.js.map +1 -0
- package/dist/storage/agentsRepository.d.ts +1 -0
- package/dist/storage/agentsRepository.d.ts.map +1 -1
- package/dist/storage/agentsRepository.js +16 -0
- package/dist/storage/agentsRepository.js.map +1 -1
- package/dist/storage/agentsRepository.spec.js +45 -0
- package/dist/storage/agentsRepository.spec.js.map +1 -1
- package/dist/storage/connectionsRepository.d.ts +18 -0
- package/dist/storage/connectionsRepository.d.ts.map +1 -0
- package/dist/storage/connectionsRepository.js +163 -0
- package/dist/storage/connectionsRepository.js.map +1 -0
- package/dist/storage/connectionsRepository.spec.d.ts +2 -0
- package/dist/storage/connectionsRepository.spec.d.ts.map +1 -0
- package/dist/storage/connectionsRepository.spec.js +115 -0
- package/dist/storage/connectionsRepository.spec.js.map +1 -0
- package/dist/storage/databaseTypes.d.ts +19 -0
- package/dist/storage/databaseTypes.d.ts.map +1 -1
- package/dist/storage/migrations/20260222_add_usertag_connections.d.ts +7 -0
- package/dist/storage/migrations/20260222_add_usertag_connections.d.ts.map +1 -0
- package/dist/storage/migrations/20260222_add_usertag_connections.js +32 -0
- package/dist/storage/migrations/20260222_add_usertag_connections.js.map +1 -0
- package/dist/storage/migrations/20260222_add_usertag_connections.spec.d.ts +2 -0
- package/dist/storage/migrations/20260222_add_usertag_connections.spec.d.ts.map +1 -0
- package/dist/storage/migrations/20260222_add_usertag_connections.spec.js +55 -0
- package/dist/storage/migrations/20260222_add_usertag_connections.spec.js.map +1 -0
- package/dist/storage/migrations/_migrations.d.ts.map +1 -1
- package/dist/storage/migrations/_migrations.js +3 -1
- package/dist/storage/migrations/_migrations.js.map +1 -1
- package/dist/storage/storage.d.ts +3 -0
- package/dist/storage/storage.d.ts.map +1 -1
- package/dist/storage/storage.js +35 -3
- package/dist/storage/storage.js.map +1 -1
- package/dist/storage/storage.spec.js +1 -0
- package/dist/storage/storage.spec.js.map +1 -1
- package/dist/storage/usersRepository.d.ts +2 -0
- package/dist/storage/usersRepository.d.ts.map +1 -1
- package/dist/storage/usersRepository.js +43 -2
- package/dist/storage/usersRepository.js.map +1 -1
- package/dist/storage/usersRepository.spec.js +9 -2
- package/dist/storage/usersRepository.spec.js.map +1 -1
- package/dist/types.d.ts +2 -0
- package/dist/types.d.ts.map +1 -1
- package/dist/types.js +2 -0
- package/dist/types.js.map +1 -1
- package/package.json +1 -1
- /package/dist/plugins/dashboard/site/_next/static/{X_oqQhoSTmj1_qmNPx-r5 → Hr0soHgJ1L7WevXil6GIk}/_buildManifest.js +0 -0
- /package/dist/plugins/dashboard/site/_next/static/{X_oqQhoSTmj1_qmNPx-r5 → Hr0soHgJ1L7WevXil6GIk}/_ssgManifest.js +0 -0
|
@@ -0,0 +1,119 @@
|
|
|
1
|
+
import { describe, expect, it, vi } from "vitest";
|
|
2
|
+
import { Storage } from "../../../storage/storage.js";
|
|
3
|
+
import { friendShareSubuserToolBuild } from "./friendShareSubuserToolBuild.js";
|
|
4
|
+
const toolCall = { id: "tool-1", name: "friend_share_subuser" };
|
|
5
|
+
describe("friendShareSubuserToolBuild", () => {
|
|
6
|
+
it("shares a subuser with an existing friend", async () => {
|
|
7
|
+
const storage = Storage.open(":memory:");
|
|
8
|
+
try {
|
|
9
|
+
const alice = await storage.users.create({ id: "alice", usertag: "happy-penguin-42" });
|
|
10
|
+
const bob = await storage.users.create({ id: "bob", usertag: "swift-fox-42" });
|
|
11
|
+
await storage.users.create({ id: "charlie", usertag: "brave-eagle-77" });
|
|
12
|
+
const subuser = await storage.users.create({
|
|
13
|
+
id: "alice-sub-1",
|
|
14
|
+
parentUserId: alice.id,
|
|
15
|
+
name: "helper",
|
|
16
|
+
usertag: "cool-cat-11"
|
|
17
|
+
});
|
|
18
|
+
await storage.connections.upsertRequest(alice.id, bob.id, 100);
|
|
19
|
+
await storage.connections.upsertRequest(bob.id, alice.id, 200);
|
|
20
|
+
const postToUserAgents = vi.fn(async () => undefined);
|
|
21
|
+
const tool = friendShareSubuserToolBuild();
|
|
22
|
+
const result = await tool.execute({ friendUsertag: "swift-fox-42", subuserId: subuser.id }, contextBuild(alice.id, storage, postToUserAgents), toolCall);
|
|
23
|
+
expect(result.typedResult.status).toBe("offered");
|
|
24
|
+
expect(postToUserAgents).toHaveBeenCalledWith(bob.id, expect.objectContaining({
|
|
25
|
+
type: "system_message",
|
|
26
|
+
origin: "friend:happy-penguin-42",
|
|
27
|
+
text: expect.stringContaining('Use friend_add("cool-cat-11") to accept.')
|
|
28
|
+
}));
|
|
29
|
+
const share = await storage.connections.find(subuser.id, bob.id);
|
|
30
|
+
expect(share).toMatchObject({
|
|
31
|
+
requestedA: true,
|
|
32
|
+
requestedB: false
|
|
33
|
+
});
|
|
34
|
+
}
|
|
35
|
+
finally {
|
|
36
|
+
storage.close();
|
|
37
|
+
}
|
|
38
|
+
});
|
|
39
|
+
it("fails when owner and friend are not connected", async () => {
|
|
40
|
+
const storage = Storage.open(":memory:");
|
|
41
|
+
try {
|
|
42
|
+
const alice = await storage.users.create({ id: "alice", usertag: "happy-penguin-42" });
|
|
43
|
+
await storage.users.create({ id: "bob", usertag: "swift-fox-42" });
|
|
44
|
+
const subuser = await storage.users.create({
|
|
45
|
+
id: "alice-sub-1",
|
|
46
|
+
parentUserId: alice.id,
|
|
47
|
+
name: "helper",
|
|
48
|
+
usertag: "cool-cat-11"
|
|
49
|
+
});
|
|
50
|
+
const tool = friendShareSubuserToolBuild();
|
|
51
|
+
await expect(tool.execute({ friendUsertag: "swift-fox-42", subuserId: subuser.id }, contextBuild(alice.id, storage, vi.fn(async () => undefined)), toolCall)).rejects.toThrow("You are not friends");
|
|
52
|
+
}
|
|
53
|
+
finally {
|
|
54
|
+
storage.close();
|
|
55
|
+
}
|
|
56
|
+
});
|
|
57
|
+
it("fails when caller does not own the subuser", async () => {
|
|
58
|
+
const storage = Storage.open(":memory:");
|
|
59
|
+
try {
|
|
60
|
+
const alice = await storage.users.create({ id: "alice", usertag: "happy-penguin-42" });
|
|
61
|
+
const bob = await storage.users.create({ id: "bob", usertag: "swift-fox-42" });
|
|
62
|
+
await storage.users.create({ id: "charlie", usertag: "brave-eagle-77" });
|
|
63
|
+
const subuser = await storage.users.create({
|
|
64
|
+
id: "alice-sub-1",
|
|
65
|
+
parentUserId: alice.id,
|
|
66
|
+
name: "helper",
|
|
67
|
+
usertag: "cool-cat-11"
|
|
68
|
+
});
|
|
69
|
+
await storage.connections.upsertRequest(alice.id, bob.id, 100);
|
|
70
|
+
await storage.connections.upsertRequest(bob.id, alice.id, 200);
|
|
71
|
+
const tool = friendShareSubuserToolBuild();
|
|
72
|
+
await expect(tool.execute({ friendUsertag: "brave-eagle-77", subuserId: subuser.id }, contextBuild(bob.id, storage, vi.fn(async () => undefined)), toolCall)).rejects.toThrow("Subuser does not belong to the calling user.");
|
|
73
|
+
}
|
|
74
|
+
finally {
|
|
75
|
+
storage.close();
|
|
76
|
+
}
|
|
77
|
+
});
|
|
78
|
+
it("fails when share is already active", async () => {
|
|
79
|
+
const storage = Storage.open(":memory:");
|
|
80
|
+
try {
|
|
81
|
+
const alice = await storage.users.create({ id: "alice", usertag: "happy-penguin-42" });
|
|
82
|
+
const bob = await storage.users.create({ id: "bob", usertag: "swift-fox-42" });
|
|
83
|
+
const subuser = await storage.users.create({
|
|
84
|
+
id: "alice-sub-1",
|
|
85
|
+
parentUserId: alice.id,
|
|
86
|
+
name: "helper",
|
|
87
|
+
usertag: "cool-cat-11"
|
|
88
|
+
});
|
|
89
|
+
await storage.connections.upsertRequest(alice.id, bob.id, 100);
|
|
90
|
+
await storage.connections.upsertRequest(bob.id, alice.id, 200);
|
|
91
|
+
await storage.connections.upsertRequest(subuser.id, bob.id, 300);
|
|
92
|
+
await storage.connections.upsertRequest(bob.id, subuser.id, 400);
|
|
93
|
+
const tool = friendShareSubuserToolBuild();
|
|
94
|
+
await expect(tool.execute({ friendUsertag: "swift-fox-42", subuserId: subuser.id }, contextBuild(alice.id, storage, vi.fn(async () => undefined)), toolCall)).rejects.toThrow("already shared");
|
|
95
|
+
}
|
|
96
|
+
finally {
|
|
97
|
+
storage.close();
|
|
98
|
+
}
|
|
99
|
+
});
|
|
100
|
+
});
|
|
101
|
+
function contextBuild(userId, storage, postToUserAgents) {
|
|
102
|
+
return {
|
|
103
|
+
connectorRegistry: null,
|
|
104
|
+
sandbox: null,
|
|
105
|
+
auth: null,
|
|
106
|
+
logger: console,
|
|
107
|
+
assistant: null,
|
|
108
|
+
agent: { id: "agent-1" },
|
|
109
|
+
ctx: { userId, agentId: "agent-1" },
|
|
110
|
+
source: "test",
|
|
111
|
+
messageContext: {},
|
|
112
|
+
agentSystem: {
|
|
113
|
+
storage,
|
|
114
|
+
postToUserAgents
|
|
115
|
+
},
|
|
116
|
+
heartbeats: null
|
|
117
|
+
};
|
|
118
|
+
}
|
|
119
|
+
//# sourceMappingURL=friendShareSubuserToolBuild.spec.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"friendShareSubuserToolBuild.spec.js","sourceRoot":"","sources":["../../../../sources/engine/modules/tools/friendShareSubuserToolBuild.spec.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAGlD,OAAO,EAAE,OAAO,EAAE,MAAM,6BAA6B,CAAC;AACtD,OAAO,EAAE,2BAA2B,EAAE,MAAM,kCAAkC,CAAC;AAE/E,MAAM,QAAQ,GAAG,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,sBAAsB,EAAE,CAAC;AAEhE,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IACzC,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACzC,IAAI,CAAC;YACD,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,kBAAkB,EAAE,CAAC,CAAC;YACvF,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC,CAAC;YAC/E,MAAM,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC,CAAC;YACzE,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC;gBACvC,EAAE,EAAE,aAAa;gBACjB,YAAY,EAAE,KAAK,CAAC,EAAE;gBACtB,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,aAAa;aACzB,CAAC,CAAC;YACH,MAAM,OAAO,CAAC,WAAW,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;YAC/D,MAAM,OAAO,CAAC,WAAW,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,EAAE,KAAK,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;YAE/D,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,SAAS,CAAC,CAAC;YACtD,MAAM,IAAI,GAAG,2BAA2B,EAAE,CAAC;YAC3C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAC7B,EAAE,aAAa,EAAE,cAAc,EAAE,SAAS,EAAE,OAAO,CAAC,EAAE,EAAE,EACxD,YAAY,CAAC,KAAK,CAAC,EAAE,EAAE,OAAO,EAAE,gBAAgB,CAAC,EACjD,QAAQ,CACX,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAClD,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CACzC,GAAG,CAAC,EAAE,EACN,MAAM,CAAC,gBAAgB,CAAC;gBACpB,IAAI,EAAE,gBAAgB;gBACtB,MAAM,EAAE,yBAAyB;gBACjC,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAAC,0CAA0C,CAAC;aAC5E,CAAC,CACL,CAAC;YAEF,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;YACjE,MAAM,CAAC,KAAK,CAAC,CAAC,aAAa,CAAC;gBACxB,UAAU,EAAE,IAAI;gBAChB,UAAU,EAAE,KAAK;aACpB,CAAC,CAAC;QACP,CAAC;gBAAS,CAAC;YACP,OAAO,CAAC,KAAK,EAAE,CAAC;QACpB,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACzC,IAAI,CAAC;YACD,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,kBAAkB,EAAE,CAAC,CAAC;YACvF,MAAM,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC,CAAC;YACnE,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC;gBACvC,EAAE,EAAE,aAAa;gBACjB,YAAY,EAAE,KAAK,CAAC,EAAE;gBACtB,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,aAAa;aACzB,CAAC,CAAC;YACH,MAAM,IAAI,GAAG,2BAA2B,EAAE,CAAC;YAE3C,MAAM,MAAM,CACR,IAAI,CAAC,OAAO,CACR,EAAE,aAAa,EAAE,cAAc,EAAE,SAAS,EAAE,OAAO,CAAC,EAAE,EAAE,EACxD,YAAY,CACR,KAAK,CAAC,EAAE,EACR,OAAO,EACP,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,SAAS,CAAC,CAC/B,EACD,QAAQ,CACX,CACJ,CAAC,OAAO,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC;QAC7C,CAAC;gBAAS,CAAC;YACP,OAAO,CAAC,KAAK,EAAE,CAAC;QACpB,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACzC,IAAI,CAAC;YACD,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,kBAAkB,EAAE,CAAC,CAAC;YACvF,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC,CAAC;YAC/E,MAAM,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC,CAAC;YACzE,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC;gBACvC,EAAE,EAAE,aAAa;gBACjB,YAAY,EAAE,KAAK,CAAC,EAAE;gBACtB,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,aAAa;aACzB,CAAC,CAAC;YACH,MAAM,OAAO,CAAC,WAAW,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;YAC/D,MAAM,OAAO,CAAC,WAAW,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,EAAE,KAAK,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;YAC/D,MAAM,IAAI,GAAG,2BAA2B,EAAE,CAAC;YAE3C,MAAM,MAAM,CACR,IAAI,CAAC,OAAO,CACR,EAAE,aAAa,EAAE,gBAAgB,EAAE,SAAS,EAAE,OAAO,CAAC,EAAE,EAAE,EAC1D,YAAY,CACR,GAAG,CAAC,EAAE,EACN,OAAO,EACP,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,SAAS,CAAC,CAC/B,EACD,QAAQ,CACX,CACJ,CAAC,OAAO,CAAC,OAAO,CAAC,8CAA8C,CAAC,CAAC;QACtE,CAAC;gBAAS,CAAC;YACP,OAAO,CAAC,KAAK,EAAE,CAAC;QACpB,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAChD,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACzC,IAAI,CAAC;YACD,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,kBAAkB,EAAE,CAAC,CAAC;YACvF,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC,CAAC;YAC/E,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC;gBACvC,EAAE,EAAE,aAAa;gBACjB,YAAY,EAAE,KAAK,CAAC,EAAE;gBACtB,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,aAAa;aACzB,CAAC,CAAC;YAEH,MAAM,OAAO,CAAC,WAAW,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;YAC/D,MAAM,OAAO,CAAC,WAAW,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,EAAE,KAAK,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;YAC/D,MAAM,OAAO,CAAC,WAAW,CAAC,aAAa,CAAC,OAAO,CAAC,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;YACjE,MAAM,OAAO,CAAC,WAAW,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;YAEjE,MAAM,IAAI,GAAG,2BAA2B,EAAE,CAAC;YAC3C,MAAM,MAAM,CACR,IAAI,CAAC,OAAO,CACR,EAAE,aAAa,EAAE,cAAc,EAAE,SAAS,EAAE,OAAO,CAAC,EAAE,EAAE,EACxD,YAAY,CACR,KAAK,CAAC,EAAE,EACR,OAAO,EACP,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,SAAS,CAAC,CAC/B,EACD,QAAQ,CACX,CACJ,CAAC,OAAO,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;QACxC,CAAC;gBAAS,CAAC;YACP,OAAO,CAAC,KAAK,EAAE,CAAC;QACpB,CAAC;IACL,CAAC,CAAC,CAAC;AACP,CAAC,CAAC,CAAC;AAEH,SAAS,YAAY,CACjB,MAAc,EACd,OAAgB,EAChB,gBAAwE;IAExE,OAAO;QACH,iBAAiB,EAAE,IAA4D;QAC/E,OAAO,EAAE,IAAkD;QAC3D,IAAI,EAAE,IAA+C;QACrD,MAAM,EAAE,OAAoD;QAC5D,SAAS,EAAE,IAAI;QACf,KAAK,EAAE,EAAE,EAAE,EAAE,SAAS,EAA8C;QACpE,GAAG,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAiC;QAClE,MAAM,EAAE,MAAM;QACd,cAAc,EAAE,EAAE;QAClB,WAAW,EAAE;YACT,OAAO;YACP,gBAAgB;SAC+B;QACnD,UAAU,EAAE,IAAqD;KACpE,CAAC;AACN,CAAC"}
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
import type { ToolDefinition } from "@/types";
|
|
2
|
+
/**
|
|
3
|
+
* Revokes a previously offered/active subuser share from a friend.
|
|
4
|
+
* Expects: caller owns subuserId and an existing share request from the subuser side is present.
|
|
5
|
+
*/
|
|
6
|
+
export declare function friendUnshareSubuserToolBuild(): ToolDefinition;
|
|
7
|
+
//# sourceMappingURL=friendUnshareSubuserToolBuild.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"friendUnshareSubuserToolBuild.d.ts","sourceRoot":"","sources":["../../../../sources/engine/modules/tools/friendUnshareSubuserToolBuild.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,cAAc,EAAsB,MAAM,SAAS,CAAC;AA8BlE;;;GAGG;AACH,wBAAgB,6BAA6B,IAAI,cAAc,CAyF9D"}
|
|
@@ -0,0 +1,118 @@
|
|
|
1
|
+
import { Type } from "@sinclair/typebox";
|
|
2
|
+
import { messageBuildSystemText } from "../../messages/messageBuildSystemText.js";
|
|
3
|
+
const schema = Type.Object({
|
|
4
|
+
friendUsertag: Type.String({ minLength: 1 }),
|
|
5
|
+
subuserId: Type.String({ minLength: 1 })
|
|
6
|
+
}, { additionalProperties: false });
|
|
7
|
+
const resultSchema = Type.Object({
|
|
8
|
+
summary: Type.String(),
|
|
9
|
+
status: Type.String(),
|
|
10
|
+
friendUsertag: Type.String(),
|
|
11
|
+
subuserId: Type.String()
|
|
12
|
+
}, { additionalProperties: false });
|
|
13
|
+
const returns = {
|
|
14
|
+
schema: resultSchema,
|
|
15
|
+
toLLMText: (result) => result.summary
|
|
16
|
+
};
|
|
17
|
+
/**
|
|
18
|
+
* Revokes a previously offered/active subuser share from a friend.
|
|
19
|
+
* Expects: caller owns subuserId and an existing share request from the subuser side is present.
|
|
20
|
+
*/
|
|
21
|
+
export function friendUnshareSubuserToolBuild() {
|
|
22
|
+
return {
|
|
23
|
+
tool: {
|
|
24
|
+
name: "friend_unshare_subuser",
|
|
25
|
+
description: "Revoke a subuser share from a friend by usertag.",
|
|
26
|
+
parameters: schema
|
|
27
|
+
},
|
|
28
|
+
returns,
|
|
29
|
+
visibleByDefault: (context) => context.descriptor.type !== "subuser",
|
|
30
|
+
execute: async (args, toolContext, toolCall) => {
|
|
31
|
+
const payload = args;
|
|
32
|
+
const targetUsertag = usertagNormalize(payload.friendUsertag);
|
|
33
|
+
const subuserId = payload.subuserId.trim();
|
|
34
|
+
if (!subuserId) {
|
|
35
|
+
throw new Error("subuserId is required.");
|
|
36
|
+
}
|
|
37
|
+
const users = toolContext.agentSystem.storage.users;
|
|
38
|
+
const connections = toolContext.agentSystem.storage.connections;
|
|
39
|
+
const me = await users.findById(toolContext.ctx.userId);
|
|
40
|
+
if (!me) {
|
|
41
|
+
throw new Error("Current user not found.");
|
|
42
|
+
}
|
|
43
|
+
const myUsertag = me.usertag?.trim() ?? "";
|
|
44
|
+
if (!myUsertag) {
|
|
45
|
+
throw new Error("Current user does not have a usertag.");
|
|
46
|
+
}
|
|
47
|
+
const friend = await users.findByUsertag(targetUsertag);
|
|
48
|
+
if (!friend) {
|
|
49
|
+
throw new Error(`User not found for usertag: ${targetUsertag}`);
|
|
50
|
+
}
|
|
51
|
+
const subuser = await users.findById(subuserId);
|
|
52
|
+
if (!subuser) {
|
|
53
|
+
throw new Error("Subuser not found.");
|
|
54
|
+
}
|
|
55
|
+
if (subuser.parentUserId !== me.id) {
|
|
56
|
+
throw new Error("Subuser does not belong to the calling user.");
|
|
57
|
+
}
|
|
58
|
+
const existing = await connections.find(subuser.id, friend.id);
|
|
59
|
+
if (!existing) {
|
|
60
|
+
throw new Error(`No share exists for subuser ${subuserId} and ${targetUsertag}.`);
|
|
61
|
+
}
|
|
62
|
+
const state = sideStateForUser(existing, subuser.id);
|
|
63
|
+
if (!state.myRequested) {
|
|
64
|
+
throw new Error(`No share exists for subuser ${subuserId} and ${targetUsertag}.`);
|
|
65
|
+
}
|
|
66
|
+
const updated = await connections.clearSide(subuser.id, friend.id);
|
|
67
|
+
if (updated && !updated.requestedA && !updated.requestedB) {
|
|
68
|
+
await connections.delete(subuser.id, friend.id);
|
|
69
|
+
}
|
|
70
|
+
const origin = `friend:${myUsertag}`;
|
|
71
|
+
const subuserName = subuser.name ?? subuser.id;
|
|
72
|
+
const subuserUsertag = subuser.usertag?.trim() ?? "unknown";
|
|
73
|
+
await toolContext.agentSystem.postToUserAgents(friend.id, {
|
|
74
|
+
type: "system_message",
|
|
75
|
+
origin,
|
|
76
|
+
text: messageBuildSystemText(`${myUsertag} revoked your access to subuser "${subuserName}" (${subuserUsertag}).`, origin)
|
|
77
|
+
});
|
|
78
|
+
const summary = `Revoked subuser ${subuserId} share from ${targetUsertag}.`;
|
|
79
|
+
const toolMessage = {
|
|
80
|
+
role: "toolResult",
|
|
81
|
+
toolCallId: toolCall.id,
|
|
82
|
+
toolName: toolCall.name,
|
|
83
|
+
content: [{ type: "text", text: summary }],
|
|
84
|
+
isError: false,
|
|
85
|
+
timestamp: Date.now()
|
|
86
|
+
};
|
|
87
|
+
return {
|
|
88
|
+
toolMessage,
|
|
89
|
+
typedResult: {
|
|
90
|
+
summary,
|
|
91
|
+
status: "revoked",
|
|
92
|
+
friendUsertag: targetUsertag,
|
|
93
|
+
subuserId
|
|
94
|
+
}
|
|
95
|
+
};
|
|
96
|
+
}
|
|
97
|
+
};
|
|
98
|
+
}
|
|
99
|
+
function usertagNormalize(value) {
|
|
100
|
+
const normalized = value.trim().toLowerCase();
|
|
101
|
+
if (!normalized) {
|
|
102
|
+
throw new Error("friendUsertag is required.");
|
|
103
|
+
}
|
|
104
|
+
return normalized;
|
|
105
|
+
}
|
|
106
|
+
function sideStateForUser(connection, userId) {
|
|
107
|
+
if (connection.userAId === userId) {
|
|
108
|
+
return {
|
|
109
|
+
myRequested: connection.requestedA,
|
|
110
|
+
otherRequested: connection.requestedB
|
|
111
|
+
};
|
|
112
|
+
}
|
|
113
|
+
return {
|
|
114
|
+
myRequested: connection.requestedB,
|
|
115
|
+
otherRequested: connection.requestedA
|
|
116
|
+
};
|
|
117
|
+
}
|
|
118
|
+
//# sourceMappingURL=friendUnshareSubuserToolBuild.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"friendUnshareSubuserToolBuild.js","sourceRoot":"","sources":["../../../../sources/engine/modules/tools/friendUnshareSubuserToolBuild.ts"],"names":[],"mappings":"AACA,OAAO,EAAe,IAAI,EAAE,MAAM,mBAAmB,CAAC;AAEtD,OAAO,EAAE,sBAAsB,EAAE,MAAM,0CAA0C,CAAC;AAElF,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CACtB;IACI,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;IAC5C,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;CAC3C,EACD,EAAE,oBAAoB,EAAE,KAAK,EAAE,CAClC,CAAC;AAIF,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,CAC5B;IACI,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE;IACtB,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE;IACrB,aAAa,EAAE,IAAI,CAAC,MAAM,EAAE;IAC5B,SAAS,EAAE,IAAI,CAAC,MAAM,EAAE;CAC3B,EACD,EAAE,oBAAoB,EAAE,KAAK,EAAE,CAClC,CAAC;AAIF,MAAM,OAAO,GAAmD;IAC5D,MAAM,EAAE,YAAY;IACpB,SAAS,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,OAAO;CACxC,CAAC;AAEF;;;GAGG;AACH,MAAM,UAAU,6BAA6B;IACzC,OAAO;QACH,IAAI,EAAE;YACF,IAAI,EAAE,wBAAwB;YAC9B,WAAW,EAAE,kDAAkD;YAC/D,UAAU,EAAE,MAAM;SACrB;QACD,OAAO;QACP,gBAAgB,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,KAAK,SAAS;QACpE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,EAAE,EAAE;YAC3C,MAAM,OAAO,GAAG,IAAgC,CAAC;YACjD,MAAM,aAAa,GAAG,gBAAgB,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAC9D,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;YAC3C,IAAI,CAAC,SAAS,EAAE,CAAC;gBACb,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;YAC9C,CAAC;YAED,MAAM,KAAK,GAAG,WAAW,CAAC,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC;YACpD,MAAM,WAAW,GAAG,WAAW,CAAC,WAAW,CAAC,OAAO,CAAC,WAAW,CAAC;YAChE,MAAM,EAAE,GAAG,MAAM,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACxD,IAAI,CAAC,EAAE,EAAE,CAAC;gBACN,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;YAC/C,CAAC;YACD,MAAM,SAAS,GAAG,EAAE,CAAC,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;YAC3C,IAAI,CAAC,SAAS,EAAE,CAAC;gBACb,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;YAC7D,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,aAAa,CAAC,aAAa,CAAC,CAAC;YACxD,IAAI,CAAC,MAAM,EAAE,CAAC;gBACV,MAAM,IAAI,KAAK,CAAC,+BAA+B,aAAa,EAAE,CAAC,CAAC;YACpE,CAAC;YAED,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;YAChD,IAAI,CAAC,OAAO,EAAE,CAAC;gBACX,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;YAC1C,CAAC;YACD,IAAI,OAAO,CAAC,YAAY,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;gBACjC,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;YACpE,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,CAAC,EAAE,CAAC,CAAC;YAC/D,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACZ,MAAM,IAAI,KAAK,CAAC,+BAA+B,SAAS,QAAQ,aAAa,GAAG,CAAC,CAAC;YACtF,CAAC;YAED,MAAM,KAAK,GAAG,gBAAgB,CAAC,QAAQ,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC;YACrD,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;gBACrB,MAAM,IAAI,KAAK,CAAC,+BAA+B,SAAS,QAAQ,aAAa,GAAG,CAAC,CAAC;YACtF,CAAC;YAED,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,CAAC,EAAE,CAAC,CAAC;YACnE,IAAI,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;gBACxD,MAAM,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,CAAC,EAAE,CAAC,CAAC;YACpD,CAAC;YAED,MAAM,MAAM,GAAG,UAAU,SAAS,EAAE,CAAC;YACrC,MAAM,WAAW,GAAG,OAAO,CAAC,IAAI,IAAI,OAAO,CAAC,EAAE,CAAC;YAC/C,MAAM,cAAc,GAAG,OAAO,CAAC,OAAO,EAAE,IAAI,EAAE,IAAI,SAAS,CAAC;YAC5D,MAAM,WAAW,CAAC,WAAW,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,EAAE;gBACtD,IAAI,EAAE,gBAAgB;gBACtB,MAAM;gBACN,IAAI,EAAE,sBAAsB,CACxB,GAAG,SAAS,oCAAoC,WAAW,MAAM,cAAc,IAAI,EACnF,MAAM,CACT;aACJ,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,mBAAmB,SAAS,eAAe,aAAa,GAAG,CAAC;YAC5E,MAAM,WAAW,GAAsB;gBACnC,IAAI,EAAE,YAAY;gBAClB,UAAU,EAAE,QAAQ,CAAC,EAAE;gBACvB,QAAQ,EAAE,QAAQ,CAAC,IAAI;gBACvB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;gBAC1C,OAAO,EAAE,KAAK;gBACd,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACxB,CAAC;YAEF,OAAO;gBACH,WAAW;gBACX,WAAW,EAAE;oBACT,OAAO;oBACP,MAAM,EAAE,SAAS;oBACjB,aAAa,EAAE,aAAa;oBAC5B,SAAS;iBACZ;aACJ,CAAC;QACN,CAAC;KACJ,CAAC;AACN,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAa;IACnC,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAC9C,IAAI,CAAC,UAAU,EAAE,CAAC;QACd,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;IAClD,CAAC;IACD,OAAO,UAAU,CAAC;AACtB,CAAC;AAED,SAAS,gBAAgB,CACrB,UAIC,EACD,MAAc;IAKd,IAAI,UAAU,CAAC,OAAO,KAAK,MAAM,EAAE,CAAC;QAChC,OAAO;YACH,WAAW,EAAE,UAAU,CAAC,UAAU;YAClC,cAAc,EAAE,UAAU,CAAC,UAAU;SACxC,CAAC;IACN,CAAC;IACD,OAAO;QACH,WAAW,EAAE,UAAU,CAAC,UAAU;QAClC,cAAc,EAAE,UAAU,CAAC,UAAU;KACxC,CAAC;AACN,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"friendUnshareSubuserToolBuild.spec.d.ts","sourceRoot":"","sources":["../../../../sources/engine/modules/tools/friendUnshareSubuserToolBuild.spec.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,100 @@
|
|
|
1
|
+
import { describe, expect, it, vi } from "vitest";
|
|
2
|
+
import { Storage } from "../../../storage/storage.js";
|
|
3
|
+
import { friendUnshareSubuserToolBuild } from "./friendUnshareSubuserToolBuild.js";
|
|
4
|
+
const toolCall = { id: "tool-1", name: "friend_unshare_subuser" };
|
|
5
|
+
describe("friendUnshareSubuserToolBuild", () => {
|
|
6
|
+
it("revokes an active share and notifies the friend", async () => {
|
|
7
|
+
const storage = Storage.open(":memory:");
|
|
8
|
+
try {
|
|
9
|
+
const alice = await storage.users.create({ id: "alice", usertag: "happy-penguin-42" });
|
|
10
|
+
const bob = await storage.users.create({ id: "bob", usertag: "swift-fox-42" });
|
|
11
|
+
const subuser = await storage.users.create({
|
|
12
|
+
id: "alice-sub-1",
|
|
13
|
+
parentUserId: alice.id,
|
|
14
|
+
name: "helper",
|
|
15
|
+
usertag: "cool-cat-11"
|
|
16
|
+
});
|
|
17
|
+
await storage.connections.upsertRequest(subuser.id, bob.id, 100);
|
|
18
|
+
await storage.connections.upsertRequest(bob.id, subuser.id, 200);
|
|
19
|
+
const postToUserAgents = vi.fn(async () => undefined);
|
|
20
|
+
const tool = friendUnshareSubuserToolBuild();
|
|
21
|
+
const result = await tool.execute({ friendUsertag: "swift-fox-42", subuserId: subuser.id }, contextBuild(alice.id, storage, postToUserAgents), toolCall);
|
|
22
|
+
expect(result.typedResult.status).toBe("revoked");
|
|
23
|
+
expect(postToUserAgents).toHaveBeenCalledWith(bob.id, expect.objectContaining({
|
|
24
|
+
type: "system_message",
|
|
25
|
+
origin: "friend:happy-penguin-42"
|
|
26
|
+
}));
|
|
27
|
+
const state = await storage.connections.find(subuser.id, bob.id);
|
|
28
|
+
expect(state).toMatchObject({
|
|
29
|
+
requestedA: false,
|
|
30
|
+
requestedB: true
|
|
31
|
+
});
|
|
32
|
+
}
|
|
33
|
+
finally {
|
|
34
|
+
storage.close();
|
|
35
|
+
}
|
|
36
|
+
});
|
|
37
|
+
it("revokes a pending share offer and removes the empty row", async () => {
|
|
38
|
+
const storage = Storage.open(":memory:");
|
|
39
|
+
try {
|
|
40
|
+
const alice = await storage.users.create({ id: "alice", usertag: "happy-penguin-42" });
|
|
41
|
+
const bob = await storage.users.create({ id: "bob", usertag: "swift-fox-42" });
|
|
42
|
+
const subuser = await storage.users.create({
|
|
43
|
+
id: "alice-sub-1",
|
|
44
|
+
parentUserId: alice.id,
|
|
45
|
+
name: "helper",
|
|
46
|
+
usertag: "cool-cat-11"
|
|
47
|
+
});
|
|
48
|
+
await storage.connections.upsertRequest(subuser.id, bob.id, 100);
|
|
49
|
+
const postToUserAgents = vi.fn(async () => undefined);
|
|
50
|
+
const tool = friendUnshareSubuserToolBuild();
|
|
51
|
+
const result = await tool.execute({ friendUsertag: "swift-fox-42", subuserId: subuser.id }, contextBuild(alice.id, storage, postToUserAgents), toolCall);
|
|
52
|
+
expect(result.typedResult.status).toBe("revoked");
|
|
53
|
+
expect(await storage.connections.find(subuser.id, bob.id)).toBeNull();
|
|
54
|
+
expect(postToUserAgents).toHaveBeenCalledWith(bob.id, expect.objectContaining({
|
|
55
|
+
type: "system_message",
|
|
56
|
+
origin: "friend:happy-penguin-42"
|
|
57
|
+
}));
|
|
58
|
+
}
|
|
59
|
+
finally {
|
|
60
|
+
storage.close();
|
|
61
|
+
}
|
|
62
|
+
});
|
|
63
|
+
it("fails when no share exists", async () => {
|
|
64
|
+
const storage = Storage.open(":memory:");
|
|
65
|
+
try {
|
|
66
|
+
const alice = await storage.users.create({ id: "alice", usertag: "happy-penguin-42" });
|
|
67
|
+
await storage.users.create({ id: "bob", usertag: "swift-fox-42" });
|
|
68
|
+
await storage.users.create({
|
|
69
|
+
id: "alice-sub-1",
|
|
70
|
+
parentUserId: alice.id,
|
|
71
|
+
name: "helper",
|
|
72
|
+
usertag: "cool-cat-11"
|
|
73
|
+
});
|
|
74
|
+
const tool = friendUnshareSubuserToolBuild();
|
|
75
|
+
await expect(tool.execute({ friendUsertag: "swift-fox-42", subuserId: "alice-sub-1" }, contextBuild(alice.id, storage, vi.fn(async () => undefined)), toolCall)).rejects.toThrow("No share exists");
|
|
76
|
+
}
|
|
77
|
+
finally {
|
|
78
|
+
storage.close();
|
|
79
|
+
}
|
|
80
|
+
});
|
|
81
|
+
});
|
|
82
|
+
function contextBuild(userId, storage, postToUserAgents) {
|
|
83
|
+
return {
|
|
84
|
+
connectorRegistry: null,
|
|
85
|
+
sandbox: null,
|
|
86
|
+
auth: null,
|
|
87
|
+
logger: console,
|
|
88
|
+
assistant: null,
|
|
89
|
+
agent: { id: "agent-1" },
|
|
90
|
+
ctx: { userId, agentId: "agent-1" },
|
|
91
|
+
source: "test",
|
|
92
|
+
messageContext: {},
|
|
93
|
+
agentSystem: {
|
|
94
|
+
storage,
|
|
95
|
+
postToUserAgents
|
|
96
|
+
},
|
|
97
|
+
heartbeats: null
|
|
98
|
+
};
|
|
99
|
+
}
|
|
100
|
+
//# sourceMappingURL=friendUnshareSubuserToolBuild.spec.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"friendUnshareSubuserToolBuild.spec.js","sourceRoot":"","sources":["../../../../sources/engine/modules/tools/friendUnshareSubuserToolBuild.spec.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAGlD,OAAO,EAAE,OAAO,EAAE,MAAM,6BAA6B,CAAC;AACtD,OAAO,EAAE,6BAA6B,EAAE,MAAM,oCAAoC,CAAC;AAEnF,MAAM,QAAQ,GAAG,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,wBAAwB,EAAE,CAAC;AAElE,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;IAC3C,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACzC,IAAI,CAAC;YACD,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,kBAAkB,EAAE,CAAC,CAAC;YACvF,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC,CAAC;YAC/E,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC;gBACvC,EAAE,EAAE,aAAa;gBACjB,YAAY,EAAE,KAAK,CAAC,EAAE;gBACtB,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,aAAa;aACzB,CAAC,CAAC;YACH,MAAM,OAAO,CAAC,WAAW,CAAC,aAAa,CAAC,OAAO,CAAC,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;YACjE,MAAM,OAAO,CAAC,WAAW,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;YAEjE,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,SAAS,CAAC,CAAC;YACtD,MAAM,IAAI,GAAG,6BAA6B,EAAE,CAAC;YAC7C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAC7B,EAAE,aAAa,EAAE,cAAc,EAAE,SAAS,EAAE,OAAO,CAAC,EAAE,EAAE,EACxD,YAAY,CAAC,KAAK,CAAC,EAAE,EAAE,OAAO,EAAE,gBAAgB,CAAC,EACjD,QAAQ,CACX,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAClD,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CACzC,GAAG,CAAC,EAAE,EACN,MAAM,CAAC,gBAAgB,CAAC;gBACpB,IAAI,EAAE,gBAAgB;gBACtB,MAAM,EAAE,yBAAyB;aACpC,CAAC,CACL,CAAC;YAEF,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;YACjE,MAAM,CAAC,KAAK,CAAC,CAAC,aAAa,CAAC;gBACxB,UAAU,EAAE,KAAK;gBACjB,UAAU,EAAE,IAAI;aACnB,CAAC,CAAC;QACP,CAAC;gBAAS,CAAC;YACP,OAAO,CAAC,KAAK,EAAE,CAAC;QACpB,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACzC,IAAI,CAAC;YACD,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,kBAAkB,EAAE,CAAC,CAAC;YACvF,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC,CAAC;YAC/E,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC;gBACvC,EAAE,EAAE,aAAa;gBACjB,YAAY,EAAE,KAAK,CAAC,EAAE;gBACtB,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,aAAa;aACzB,CAAC,CAAC;YACH,MAAM,OAAO,CAAC,WAAW,CAAC,aAAa,CAAC,OAAO,CAAC,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;YAEjE,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,SAAS,CAAC,CAAC;YACtD,MAAM,IAAI,GAAG,6BAA6B,EAAE,CAAC;YAC7C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAC7B,EAAE,aAAa,EAAE,cAAc,EAAE,SAAS,EAAE,OAAO,CAAC,EAAE,EAAE,EACxD,YAAY,CAAC,KAAK,CAAC,EAAE,EAAE,OAAO,EAAE,gBAAgB,CAAC,EACjD,QAAQ,CACX,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;YACtE,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CACzC,GAAG,CAAC,EAAE,EACN,MAAM,CAAC,gBAAgB,CAAC;gBACpB,IAAI,EAAE,gBAAgB;gBACtB,MAAM,EAAE,yBAAyB;aACpC,CAAC,CACL,CAAC;QACN,CAAC;gBAAS,CAAC;YACP,OAAO,CAAC,KAAK,EAAE,CAAC;QACpB,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;QACxC,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACzC,IAAI,CAAC;YACD,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,kBAAkB,EAAE,CAAC,CAAC;YACvF,MAAM,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC,CAAC;YACnE,MAAM,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC;gBACvB,EAAE,EAAE,aAAa;gBACjB,YAAY,EAAE,KAAK,CAAC,EAAE;gBACtB,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,aAAa;aACzB,CAAC,CAAC;YACH,MAAM,IAAI,GAAG,6BAA6B,EAAE,CAAC;YAE7C,MAAM,MAAM,CACR,IAAI,CAAC,OAAO,CACR,EAAE,aAAa,EAAE,cAAc,EAAE,SAAS,EAAE,aAAa,EAAE,EAC3D,YAAY,CACR,KAAK,CAAC,EAAE,EACR,OAAO,EACP,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,SAAS,CAAC,CAC/B,EACD,QAAQ,CACX,CACJ,CAAC,OAAO,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;QACzC,CAAC;gBAAS,CAAC;YACP,OAAO,CAAC,KAAK,EAAE,CAAC;QACpB,CAAC;IACL,CAAC,CAAC,CAAC;AACP,CAAC,CAAC,CAAC;AAEH,SAAS,YAAY,CACjB,MAAc,EACd,OAAgB,EAChB,gBAAwE;IAExE,OAAO;QACH,iBAAiB,EAAE,IAA4D;QAC/E,OAAO,EAAE,IAAkD;QAC3D,IAAI,EAAE,IAA+C;QACrD,MAAM,EAAE,OAAoD;QAC5D,SAAS,EAAE,IAAI;QACf,KAAK,EAAE,EAAE,EAAE,EAAE,SAAS,EAA8C;QACpE,GAAG,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAiC;QAClE,MAAM,EAAE,MAAM;QACd,cAAc,EAAE,EAAE;QAClB,WAAW,EAAE;YACT,OAAO;YACP,gBAAgB;SAC+B;QACnD,UAAU,EAAE,IAAqD;KACpE,CAAC;AACN,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"image-generation.d.ts","sourceRoot":"","sources":["../../../../sources/engine/modules/tools/image-generation.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"image-generation.d.ts","sourceRoot":"","sources":["../../../../sources/engine/modules/tools/image-generation.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,cAAc,EAAsB,MAAM,SAAS,CAAC;AAGlE,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,+BAA+B,CAAC;AA0E7E,wBAAgB,wBAAwB,CAAC,aAAa,EAAE,uBAAuB,GAAG,cAAc,CA6H/F"}
|
|
@@ -1,4 +1,8 @@
|
|
|
1
|
+
import { promises as fs } from "node:fs";
|
|
2
|
+
import path from "node:path";
|
|
1
3
|
import { Type } from "@sinclair/typebox";
|
|
4
|
+
import { sanitizeFilename } from "../../../util/filename.js";
|
|
5
|
+
import { FileFolder } from "../../files/fileFolder.js";
|
|
2
6
|
const schema = Type.Object({
|
|
3
7
|
prompt: Type.String({ minLength: 1 }),
|
|
4
8
|
provider: Type.Optional(Type.String({ minLength: 1 })),
|
|
@@ -74,20 +78,22 @@ export function buildImageGenerationTool(imageRegistry) {
|
|
|
74
78
|
if (!provider) {
|
|
75
79
|
throw new Error(`Unknown image provider: ${providerId}`);
|
|
76
80
|
}
|
|
81
|
+
const tempFileStore = new FileFolder(path.join(toolContext.sandbox.homeDir, "tmp", "image-generation"));
|
|
77
82
|
const result = await provider.generate({
|
|
78
83
|
prompt: payload.prompt,
|
|
79
84
|
size: payload.size,
|
|
80
85
|
count: payload.count,
|
|
81
86
|
model: payload.model
|
|
82
87
|
}, {
|
|
83
|
-
fileStore:
|
|
88
|
+
fileStore: tempFileStore,
|
|
84
89
|
auth: toolContext.auth,
|
|
85
90
|
logger: toolContext.logger
|
|
86
91
|
});
|
|
87
|
-
const downloadsDir = toolContext.
|
|
92
|
+
const downloadsDir = path.join(toolContext.sandbox.homeDir, "downloads");
|
|
88
93
|
const createdAt = new Date();
|
|
89
94
|
const timestamp = createdAt.toISOString().replace(/[:.]/g, "-");
|
|
90
|
-
const
|
|
95
|
+
const downloadsDirSandboxPath = `~/${path.relative(toolContext.sandbox.homeDir, downloadsDir)}`;
|
|
96
|
+
const summary = `Generated ${result.files.length} image(s) with ${providerId}. Saved under ${downloadsDirSandboxPath}.`;
|
|
91
97
|
const content = [
|
|
92
98
|
{
|
|
93
99
|
type: "text",
|
|
@@ -101,15 +107,24 @@ export function buildImageGenerationTool(imageRegistry) {
|
|
|
101
107
|
}
|
|
102
108
|
const suffix = result.files.length > 1 ? `-${index + 1}` : "";
|
|
103
109
|
const extension = imageExtensionResolve(file.mimeType);
|
|
104
|
-
const
|
|
105
|
-
|
|
110
|
+
const fileName = sanitizeFilename(`${timestamp}${suffix}${extension}`);
|
|
111
|
+
const targetPath = path.join(downloadsDir, fileName);
|
|
112
|
+
const sourceContent = await fs.readFile(file.path);
|
|
113
|
+
const saved = await toolContext.sandbox.write({
|
|
114
|
+
path: targetPath,
|
|
115
|
+
content: sourceContent
|
|
116
|
+
});
|
|
117
|
+
savedFiles.push({
|
|
118
|
+
id: saved.sandboxPath,
|
|
119
|
+
name: fileName,
|
|
120
|
+
path: saved.sandboxPath,
|
|
121
|
+
resolvedPath: saved.resolvedPath,
|
|
106
122
|
mimeType: file.mimeType,
|
|
107
|
-
|
|
123
|
+
size: saved.bytes
|
|
108
124
|
});
|
|
109
|
-
savedFiles.push(saved);
|
|
110
125
|
content.push({
|
|
111
126
|
type: "text",
|
|
112
|
-
text: `Image file: ${saved.
|
|
127
|
+
text: `Image file: ${saved.sandboxPath} (${file.mimeType})`
|
|
113
128
|
});
|
|
114
129
|
}
|
|
115
130
|
const toolMessage = {
|
|
@@ -126,11 +141,12 @@ export function buildImageGenerationTool(imageRegistry) {
|
|
|
126
141
|
size: file.size
|
|
127
142
|
})),
|
|
128
143
|
downloads: {
|
|
129
|
-
dir:
|
|
144
|
+
dir: downloadsDirSandboxPath,
|
|
130
145
|
files: savedFiles.map((file) => ({
|
|
131
146
|
id: file.id,
|
|
132
147
|
name: file.name,
|
|
133
148
|
path: file.path,
|
|
149
|
+
resolvedPath: file.resolvedPath,
|
|
134
150
|
mimeType: file.mimeType,
|
|
135
151
|
size: file.size
|
|
136
152
|
}))
|
|
@@ -145,7 +161,7 @@ export function buildImageGenerationTool(imageRegistry) {
|
|
|
145
161
|
summary,
|
|
146
162
|
provider: providerId,
|
|
147
163
|
fileCount: savedFiles.length,
|
|
148
|
-
downloadsDir
|
|
164
|
+
downloadsDir: downloadsDirSandboxPath
|
|
149
165
|
}
|
|
150
166
|
};
|
|
151
167
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"image-generation.js","sourceRoot":"","sources":["../../../../sources/engine/modules/tools/image-generation.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"image-generation.js","sourceRoot":"","sources":["../../../../sources/engine/modules/tools/image-generation.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,IAAI,EAAE,EAAE,MAAM,SAAS,CAAC;AACzC,OAAO,IAAI,MAAM,WAAW,CAAC;AAE7B,OAAO,EAAE,IAAI,EAAE,MAAM,mBAAmB,CAAC;AAEzC,OAAO,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAC7D,OAAO,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAIvD,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CACtB;IACI,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;IACrC,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;IACtD,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;IAClD,KAAK,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;IAC7D,KAAK,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;CACtD,EACD,EAAE,oBAAoB,EAAE,KAAK,EAAE,CAClC,CAAC;AAEF,MAAM,2BAA2B,GAAG,IAAI,CAAC,MAAM,CAC3C;IACI,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE;IACtB,QAAQ,EAAE,IAAI,CAAC,MAAM,EAAE;IACvB,SAAS,EAAE,IAAI,CAAC,MAAM,EAAE;IACxB,YAAY,EAAE,IAAI,CAAC,MAAM,EAAE;CAC9B,EACD,EAAE,oBAAoB,EAAE,KAAK,EAAE,CAClC,CAAC;AASF,MAAM,sBAAsB,GAA8C;IACtE,MAAM,EAAE,2BAA2B;IACnC,SAAS,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,OAAO;CACxC,CAAC;AAEF,MAAM,wBAAwB,GAA2B;IACrD,YAAY,EAAE,MAAM;IACpB,WAAW,EAAE,MAAM;IACnB,WAAW,EAAE,MAAM;IACnB,WAAW,EAAE,MAAM;IACnB,YAAY,EAAE,OAAO;IACrB,eAAe,EAAE,MAAM;IACvB,WAAW,EAAE,MAAM;IACnB,YAAY,EAAE,OAAO;IACrB,YAAY,EAAE,OAAO;IACrB,cAAc,EAAE,MAAM;IACtB,0BAA0B,EAAE,MAAM;CACrC,CAAC;AAEF,SAAS,qBAAqB,CAAC,QAAgB;IAC3C,MAAM,UAAU,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACjD,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC1B,OAAO,MAAM,CAAC;IAClB,CAAC;IACD,MAAM,MAAM,GAAG,wBAAwB,CAAC,UAAU,CAAC,CAAC;IACpD,IAAI,MAAM,EAAE,CAAC;QACT,OAAO,MAAM,CAAC;IAClB,CAAC;IACD,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QACnC,OAAO,MAAM,CAAC;IAClB,CAAC;IACD,MAAM,OAAO,GAAG,UAAU,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;IAC9E,MAAM,WAAW,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;IACxD,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC3B,OAAO,MAAM,CAAC;IAClB,CAAC;IACD,MAAM,WAAW,GAAG,WAAW,CAAC,OAAO,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;IAC5D,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC3B,OAAO,MAAM,CAAC;IAClB,CAAC;IACD,OAAO,WAAW,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,WAAW,EAAE,CAAC;AACzF,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,aAAsC;IAC3E,OAAO;QACH,IAAI,EAAE;YACF,IAAI,EAAE,gBAAgB;YACtB,WAAW,EAAE,kEAAkE;YAC/E,UAAU,EAAE,MAAM;SACrB;QACD,OAAO,EAAE,sBAAsB;QAC/B,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,EAAE,EAAE;YAC3C,MAAM,OAAO,GAAG,IAAsD,CAAC;YACvE,MAAM,SAAS,GAAG,aAAa,CAAC,IAAI,EAAE,CAAC;YACvC,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;YAC/D,CAAC;YACD,MAAM,UAAU,GAAG,OAAO,CAAC,QAAQ,IAAI,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAE,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YAC1F,IAAI,CAAC,UAAU,EAAE,CAAC;gBACd,MAAM,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAC;YAC5E,CAAC;YACD,MAAM,QAAQ,GAAG,aAAa,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YAC/C,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACZ,MAAM,IAAI,KAAK,CAAC,2BAA2B,UAAU,EAAE,CAAC,CAAC;YAC7D,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,OAAO,EAAE,KAAK,EAAE,kBAAkB,CAAC,CAAC,CAAC;YAExG,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,QAAQ,CAClC;gBACI,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,KAAK,EAAE,OAAO,CAAC,KAAK;aACvB,EACD;gBACI,SAAS,EAAE,aAAa;gBACxB,IAAI,EAAE,WAAW,CAAC,IAAI;gBACtB,MAAM,EAAE,WAAW,CAAC,MAAM;aAC7B,CACJ,CAAC;YAEF,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;YACzE,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;YAC7B,MAAM,SAAS,GAAG,SAAS,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;YAChE,MAAM,uBAAuB,GAAG,KAAK,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,OAAO,CAAC,OAAO,EAAE,YAAY,CAAC,EAAE,CAAC;YAEhG,MAAM,OAAO,GAAG,aAAa,MAAM,CAAC,KAAK,CAAC,MAAM,kBAAkB,UAAU,iBAAiB,uBAAuB,GAAG,CAAC;YACxH,MAAM,OAAO,GAA0C;gBACnD;oBACI,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,OAAO;iBAChB;aACJ,CAAC;YACF,MAAM,UAAU,GAOX,EAAE,CAAC;YACR,KAAK,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC;gBACjD,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;oBACtC,SAAS;gBACb,CAAC;gBACD,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC9D,MAAM,SAAS,GAAG,qBAAqB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACvD,MAAM,QAAQ,GAAG,gBAAgB,CAAC,GAAG,SAAS,GAAG,MAAM,GAAG,SAAS,EAAE,CAAC,CAAC;gBACvE,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;gBACrD,MAAM,aAAa,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACnD,MAAM,KAAK,GAAG,MAAM,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC;oBAC1C,IAAI,EAAE,UAAU;oBAChB,OAAO,EAAE,aAAa;iBACzB,CAAC,CAAC;gBACH,UAAU,CAAC,IAAI,CAAC;oBACZ,EAAE,EAAE,KAAK,CAAC,WAAW;oBACrB,IAAI,EAAE,QAAQ;oBACd,IAAI,EAAE,KAAK,CAAC,WAAW;oBACvB,YAAY,EAAE,KAAK,CAAC,YAAY;oBAChC,QAAQ,EAAE,IAAI,CAAC,QAAQ;oBACvB,IAAI,EAAE,KAAK,CAAC,KAAK;iBACpB,CAAC,CAAC;gBACH,OAAO,CAAC,IAAI,CAAC;oBACT,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,eAAe,KAAK,CAAC,WAAW,KAAK,IAAI,CAAC,QAAQ,GAAG;iBAC9D,CAAC,CAAC;YACP,CAAC;YAED,MAAM,WAAW,GAAsB;gBACnC,IAAI,EAAE,YAAY;gBAClB,UAAU,EAAE,QAAQ,CAAC,EAAE;gBACvB,QAAQ,EAAE,QAAQ,CAAC,IAAI;gBACvB,OAAO;gBACP,OAAO,EAAE;oBACL,QAAQ,EAAE,UAAU;oBACpB,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;wBAC/B,EAAE,EAAE,IAAI,CAAC,EAAE;wBACX,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,QAAQ,EAAE,IAAI,CAAC,QAAQ;wBACvB,IAAI,EAAE,IAAI,CAAC,IAAI;qBAClB,CAAC,CAAC;oBACH,SAAS,EAAE;wBACP,GAAG,EAAE,uBAAuB;wBAC5B,KAAK,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;4BAC7B,EAAE,EAAE,IAAI,CAAC,EAAE;4BACX,IAAI,EAAE,IAAI,CAAC,IAAI;4BACf,IAAI,EAAE,IAAI,CAAC,IAAI;4BACf,YAAY,EAAE,IAAI,CAAC,YAAY;4BAC/B,QAAQ,EAAE,IAAI,CAAC,QAAQ;4BACvB,IAAI,EAAE,IAAI,CAAC,IAAI;yBAClB,CAAC,CAAC;qBACN;iBACJ;gBACD,OAAO,EAAE,KAAK;gBACd,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACxB,CAAC;YAEF,OAAO;gBACH,WAAW;gBACX,WAAW,EAAE;oBACT,OAAO;oBACP,QAAQ,EAAE,UAAU;oBACpB,SAAS,EAAE,UAAU,CAAC,MAAM;oBAC5B,YAAY,EAAE,uBAAuB;iBACxC;aACJ,CAAC;QACN,CAAC;KACJ,CAAC;AACN,CAAC"}
|
|
@@ -2,7 +2,7 @@ import { promises as fs } from "node:fs";
|
|
|
2
2
|
import os from "node:os";
|
|
3
3
|
import path from "node:path";
|
|
4
4
|
import { describe, expect, it } from "vitest";
|
|
5
|
-
import {
|
|
5
|
+
import { Sandbox } from "../../../sandbox/sandbox.js";
|
|
6
6
|
import { ImageGenerationRegistry } from "../imageGenerationRegistry.js";
|
|
7
7
|
import { buildImageGenerationTool } from "./image-generation.js";
|
|
8
8
|
describe("buildImageGenerationTool", () => {
|
|
@@ -33,14 +33,15 @@ describe("buildImageGenerationTool", () => {
|
|
|
33
33
|
name: "generate_image"
|
|
34
34
|
});
|
|
35
35
|
const downloads = result.toolMessage.details?.downloads;
|
|
36
|
-
expect(downloads?.dir).toBe(
|
|
36
|
+
expect(downloads?.dir).toBe("~/downloads");
|
|
37
37
|
expect(downloads?.files).toHaveLength(1);
|
|
38
38
|
const generatedPath = downloads?.files[0]?.path;
|
|
39
|
+
const generatedResolvedPath = downloads?.files[0]?.resolvedPath;
|
|
39
40
|
expect(generatedPath).toBeTruthy();
|
|
40
|
-
if (!generatedPath) {
|
|
41
|
+
if (!generatedPath || !generatedResolvedPath) {
|
|
41
42
|
throw new Error("Missing generated file path");
|
|
42
43
|
}
|
|
43
|
-
const generatedData = await fs.readFile(
|
|
44
|
+
const generatedData = await fs.readFile(generatedResolvedPath, "utf8");
|
|
44
45
|
expect(generatedData).toBe("png-bytes");
|
|
45
46
|
const content = result.toolMessage.content;
|
|
46
47
|
expect(content.every((block) => block.type === "text")).toBe(true);
|
|
@@ -53,17 +54,19 @@ describe("buildImageGenerationTool", () => {
|
|
|
53
54
|
});
|
|
54
55
|
});
|
|
55
56
|
function contextBuild(workingDir) {
|
|
56
|
-
const
|
|
57
|
+
const permissions = {
|
|
58
|
+
workingDir,
|
|
59
|
+
writeDirs: [workingDir]
|
|
60
|
+
};
|
|
57
61
|
return {
|
|
58
62
|
connectorRegistry: null,
|
|
59
|
-
|
|
63
|
+
sandbox: new Sandbox({
|
|
64
|
+
homeDir: workingDir,
|
|
65
|
+
permissions
|
|
66
|
+
}),
|
|
60
67
|
auth: null,
|
|
61
68
|
logger: console,
|
|
62
69
|
assistant: null,
|
|
63
|
-
permissions: {
|
|
64
|
-
workingDir,
|
|
65
|
-
writeDirs: []
|
|
66
|
-
},
|
|
67
70
|
agent: { id: "agent-test" },
|
|
68
71
|
ctx: null,
|
|
69
72
|
source: "test",
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"image-generation.spec.js","sourceRoot":"","sources":["../../../../sources/engine/modules/tools/image-generation.spec.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,IAAI,EAAE,EAAE,MAAM,SAAS,CAAC;AACzC,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAE7B,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAG9C,OAAO,EAAE,
|
|
1
|
+
{"version":3,"file":"image-generation.spec.js","sourceRoot":"","sources":["../../../../sources/engine/modules/tools/image-generation.spec.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,IAAI,EAAE,EAAE,MAAM,SAAS,CAAC;AACzC,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAE7B,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAG9C,OAAO,EAAE,OAAO,EAAE,MAAM,6BAA6B,CAAC;AACtD,OAAO,EAAE,uBAAuB,EAAE,MAAM,+BAA+B,CAAC;AACxE,OAAO,EAAE,wBAAwB,EAAE,MAAM,uBAAuB,CAAC;AAEjE,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACtC,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QACxE,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,qBAAqB,CAAC,CAAC,CAAC;QAChF,IAAI,CAAC;YACD,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;YACpD,MAAM,EAAE,CAAC,SAAS,CAAC,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC;YAEzD,MAAM,aAAa,GAAG,IAAI,uBAAuB,EAAE,CAAC;YACpD,aAAa,CAAC,QAAQ,CAAC,aAAa,EAAE;gBAClC,EAAE,EAAE,eAAe;gBACnB,KAAK,EAAE,eAAe;gBACtB,QAAQ,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;oBACnB,KAAK,EAAE;wBACH;4BACI,EAAE,EAAE,QAAQ;4BACZ,IAAI,EAAE,YAAY;4BAClB,QAAQ,EAAE,WAAW;4BACrB,IAAI,EAAE,CAAC;4BACP,IAAI,EAAE,UAAU;yBACnB;qBACJ;iBACJ,CAAC;aACL,CAAC,CAAC;YAEH,MAAM,IAAI,GAAG,wBAAwB,CAAC,aAAa,CAAC,CAAC;YACrD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE,YAAY,CAAC,OAAO,CAAC,EAAE;gBACzE,EAAE,EAAE,QAAQ;gBACZ,IAAI,EAAE,gBAAgB;aACzB,CAAC,CAAC;YAEH,MAAM,SAAS,GAAG,MAAM,CAAC,WAAW,CAAC,OAAO,EAAE,SAE/B,CAAC;YAChB,MAAM,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC3C,MAAM,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACzC,MAAM,aAAa,GAAG,SAAS,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC;YAChD,MAAM,qBAAqB,GAAG,SAAS,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,YAAY,CAAC;YAChE,MAAM,CAAC,aAAa,CAAC,CAAC,UAAU,EAAE,CAAC;YACnC,IAAI,CAAC,aAAa,IAAI,CAAC,qBAAqB,EAAE,CAAC;gBAC3C,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;YACnD,CAAC;YACD,MAAM,aAAa,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,qBAAqB,EAAE,MAAM,CAAC,CAAC;YACvE,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACxC,MAAM,OAAO,GAAG,MAAM,CAAC,WAAW,CAAC,OAAgE,CAAC;YACpG,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACnE,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACvF,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpE,CAAC;gBAAS,CAAC;YACP,MAAM,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAC3D,CAAC;IACL,CAAC,CAAC,CAAC;AACP,CAAC,CAAC,CAAC;AAEH,SAAS,YAAY,CAAC,UAAkB;IACpC,MAAM,WAAW,GAAG;QAChB,UAAU;QACV,SAAS,EAAE,CAAC,UAAU,CAAC;KAC1B,CAAC;IACF,OAAO;QACH,iBAAiB,EAAE,IAA4D;QAC/E,OAAO,EAAE,IAAI,OAAO,CAAC;YACjB,OAAO,EAAE,UAAU;YACnB,WAAW;SACd,CAAC;QACF,IAAI,EAAE,IAA+C;QACrD,MAAM,EAAE,OAAoD;QAC5D,SAAS,EAAE,IAAI;QACf,KAAK,EAAE,EAAE,EAAE,EAAE,YAAY,EAA8C;QACvE,GAAG,EAAE,IAA8C;QACnD,MAAM,EAAE,MAAM;QACd,cAAc,EAAE,EAAuD;QACvE,WAAW,EAAE,IAAsD;QACnE,UAAU,EAAE,IAAqD;KACpE,CAAC;AACN,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"mermaid-png.d.ts","sourceRoot":"","sources":["../../../../sources/engine/modules/tools/mermaid-png.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"mermaid-png.d.ts","sourceRoot":"","sources":["../../../../sources/engine/modules/tools/mermaid-png.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,cAAc,EAAsB,MAAM,SAAS,CAAC;AAIlE,QAAA,MAAM,MAAM;;;;;EAQX,CAAC;AAiCF;;;GAGG;AACH,wBAAgB,mBAAmB,IAAI,cAAc,CAAC,OAAO,MAAM,CAAC,CAyEnE"}
|