@nodetool-ai/websocket 0.7.0-rc.11

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.
Files changed (261) hide show
  1. package/dist/agent/agent-runtime.d.ts +47 -0
  2. package/dist/agent/agent-runtime.d.ts.map +1 -0
  3. package/dist/agent/agent-runtime.js +577 -0
  4. package/dist/agent/agent-runtime.js.map +1 -0
  5. package/dist/agent/codex-agent.d.ts +41 -0
  6. package/dist/agent/codex-agent.d.ts.map +1 -0
  7. package/dist/agent/codex-agent.js +334 -0
  8. package/dist/agent/codex-agent.js.map +1 -0
  9. package/dist/agent/index.d.ts +6 -0
  10. package/dist/agent/index.d.ts.map +1 -0
  11. package/dist/agent/index.js +4 -0
  12. package/dist/agent/index.js.map +1 -0
  13. package/dist/agent/llm-agent.d.ts +51 -0
  14. package/dist/agent/llm-agent.d.ts.map +1 -0
  15. package/dist/agent/llm-agent.js +526 -0
  16. package/dist/agent/llm-agent.js.map +1 -0
  17. package/dist/agent/mcp-tool-server.d.ts +33 -0
  18. package/dist/agent/mcp-tool-server.d.ts.map +1 -0
  19. package/dist/agent/mcp-tool-server.js +292 -0
  20. package/dist/agent/mcp-tool-server.js.map +1 -0
  21. package/dist/agent/opencode-agent.d.ts +32 -0
  22. package/dist/agent/opencode-agent.d.ts.map +1 -0
  23. package/dist/agent/opencode-agent.js +332 -0
  24. package/dist/agent/opencode-agent.js.map +1 -0
  25. package/dist/agent/pi-agent.d.ts +42 -0
  26. package/dist/agent/pi-agent.d.ts.map +1 -0
  27. package/dist/agent/pi-agent.js +429 -0
  28. package/dist/agent/pi-agent.js.map +1 -0
  29. package/dist/agent/sdk-provider.d.ts +46 -0
  30. package/dist/agent/sdk-provider.d.ts.map +1 -0
  31. package/dist/agent/sdk-provider.js +50 -0
  32. package/dist/agent/sdk-provider.js.map +1 -0
  33. package/dist/agent/socket-route.d.ts +11 -0
  34. package/dist/agent/socket-route.d.ts.map +1 -0
  35. package/dist/agent/socket-route.js +262 -0
  36. package/dist/agent/socket-route.js.map +1 -0
  37. package/dist/agent/transport.d.ts +39 -0
  38. package/dist/agent/transport.d.ts.map +1 -0
  39. package/dist/agent/transport.js +17 -0
  40. package/dist/agent/transport.js.map +1 -0
  41. package/dist/agent/types.d.ts +11 -0
  42. package/dist/agent/types.d.ts.map +1 -0
  43. package/dist/agent/types.js +10 -0
  44. package/dist/agent/types.js.map +1 -0
  45. package/dist/collection-api.d.ts +14 -0
  46. package/dist/collection-api.d.ts.map +1 -0
  47. package/dist/collection-api.js +75 -0
  48. package/dist/collection-api.js.map +1 -0
  49. package/dist/error-codes.d.ts +3 -0
  50. package/dist/error-codes.d.ts.map +1 -0
  51. package/dist/error-codes.js +2 -0
  52. package/dist/error-codes.js.map +1 -0
  53. package/dist/file-api.d.ts +12 -0
  54. package/dist/file-api.d.ts.map +1 -0
  55. package/dist/file-api.js +76 -0
  56. package/dist/file-api.js.map +1 -0
  57. package/dist/http-api.d.ts +94 -0
  58. package/dist/http-api.d.ts.map +1 -0
  59. package/dist/http-api.js +1472 -0
  60. package/dist/http-api.js.map +1 -0
  61. package/dist/index.d.ts +6 -0
  62. package/dist/index.d.ts.map +1 -0
  63. package/dist/index.js +6 -0
  64. package/dist/index.js.map +1 -0
  65. package/dist/lib/asset-paths.d.ts +10 -0
  66. package/dist/lib/asset-paths.d.ts.map +1 -0
  67. package/dist/lib/asset-paths.js +45 -0
  68. package/dist/lib/asset-paths.js.map +1 -0
  69. package/dist/lib/bridge.d.ts +8 -0
  70. package/dist/lib/bridge.d.ts.map +1 -0
  71. package/dist/lib/bridge.js +70 -0
  72. package/dist/lib/bridge.js.map +1 -0
  73. package/dist/mcp-server.d.ts +32 -0
  74. package/dist/mcp-server.d.ts.map +1 -0
  75. package/dist/mcp-server.js +842 -0
  76. package/dist/mcp-server.js.map +1 -0
  77. package/dist/models-api.d.ts +12 -0
  78. package/dist/models-api.d.ts.map +1 -0
  79. package/dist/models-api.js +730 -0
  80. package/dist/models-api.js.map +1 -0
  81. package/dist/oauth-api.d.ts +28 -0
  82. package/dist/oauth-api.d.ts.map +1 -0
  83. package/dist/oauth-api.js +681 -0
  84. package/dist/oauth-api.js.map +1 -0
  85. package/dist/openai-api.d.ts +63 -0
  86. package/dist/openai-api.d.ts.map +1 -0
  87. package/dist/openai-api.js +359 -0
  88. package/dist/openai-api.js.map +1 -0
  89. package/dist/plugins/websocket.d.ts +14 -0
  90. package/dist/plugins/websocket.d.ts.map +1 -0
  91. package/dist/plugins/websocket.js +431 -0
  92. package/dist/plugins/websocket.js.map +1 -0
  93. package/dist/resolve-media-urls.d.ts +19 -0
  94. package/dist/resolve-media-urls.d.ts.map +1 -0
  95. package/dist/resolve-media-urls.js +109 -0
  96. package/dist/resolve-media-urls.js.map +1 -0
  97. package/dist/routes/assets.d.ts +22 -0
  98. package/dist/routes/assets.d.ts.map +1 -0
  99. package/dist/routes/assets.js +168 -0
  100. package/dist/routes/assets.js.map +1 -0
  101. package/dist/routes/collections.d.ts +16 -0
  102. package/dist/routes/collections.d.ts.map +1 -0
  103. package/dist/routes/collections.js +32 -0
  104. package/dist/routes/collections.js.map +1 -0
  105. package/dist/routes/files.d.ts +12 -0
  106. package/dist/routes/files.d.ts.map +1 -0
  107. package/dist/routes/files.js +11 -0
  108. package/dist/routes/files.js.map +1 -0
  109. package/dist/routes/health.d.ts +4 -0
  110. package/dist/routes/health.d.ts.map +1 -0
  111. package/dist/routes/health.js +41 -0
  112. package/dist/routes/health.js.map +1 -0
  113. package/dist/routes/jobs.d.ts +16 -0
  114. package/dist/routes/jobs.d.ts.map +1 -0
  115. package/dist/routes/jobs.js +25 -0
  116. package/dist/routes/jobs.js.map +1 -0
  117. package/dist/routes/nodes.d.ts +8 -0
  118. package/dist/routes/nodes.d.ts.map +1 -0
  119. package/dist/routes/nodes.js +124 -0
  120. package/dist/routes/nodes.js.map +1 -0
  121. package/dist/routes/oauth.d.ts +8 -0
  122. package/dist/routes/oauth.d.ts.map +1 -0
  123. package/dist/routes/oauth.js +19 -0
  124. package/dist/routes/oauth.js.map +1 -0
  125. package/dist/routes/openai.d.ts +8 -0
  126. package/dist/routes/openai.d.ts.map +1 -0
  127. package/dist/routes/openai.js +20 -0
  128. package/dist/routes/openai.js.map +1 -0
  129. package/dist/routes/storage.d.ts +12 -0
  130. package/dist/routes/storage.d.ts.map +1 -0
  131. package/dist/routes/storage.js +13 -0
  132. package/dist/routes/storage.js.map +1 -0
  133. package/dist/routes/workflows.d.ts +18 -0
  134. package/dist/routes/workflows.d.ts.map +1 -0
  135. package/dist/routes/workflows.js +25 -0
  136. package/dist/routes/workflows.js.map +1 -0
  137. package/dist/routes/workspace.d.ts +15 -0
  138. package/dist/routes/workspace.d.ts.map +1 -0
  139. package/dist/routes/workspace.js +24 -0
  140. package/dist/routes/workspace.js.map +1 -0
  141. package/dist/screenshot-server.d.ts +21 -0
  142. package/dist/screenshot-server.d.ts.map +1 -0
  143. package/dist/screenshot-server.js +222 -0
  144. package/dist/screenshot-server.js.map +1 -0
  145. package/dist/server.d.ts +10 -0
  146. package/dist/server.d.ts.map +1 -0
  147. package/dist/server.js +783 -0
  148. package/dist/server.js.map +1 -0
  149. package/dist/settings-registry.d.ts +29 -0
  150. package/dist/settings-registry.d.ts.map +1 -0
  151. package/dist/settings-registry.js +117 -0
  152. package/dist/settings-registry.js.map +1 -0
  153. package/dist/storage-api.d.ts +5 -0
  154. package/dist/storage-api.d.ts.map +1 -0
  155. package/dist/storage-api.js +221 -0
  156. package/dist/storage-api.js.map +1 -0
  157. package/dist/test-ui-server.d.ts +19 -0
  158. package/dist/test-ui-server.d.ts.map +1 -0
  159. package/dist/test-ui-server.js +1256 -0
  160. package/dist/test-ui-server.js.map +1 -0
  161. package/dist/trpc/context.d.ts +21 -0
  162. package/dist/trpc/context.d.ts.map +1 -0
  163. package/dist/trpc/context.js +10 -0
  164. package/dist/trpc/context.js.map +1 -0
  165. package/dist/trpc/error-formatter.d.ts +18 -0
  166. package/dist/trpc/error-formatter.d.ts.map +1 -0
  167. package/dist/trpc/error-formatter.js +41 -0
  168. package/dist/trpc/error-formatter.js.map +1 -0
  169. package/dist/trpc/index.d.ts +18 -0
  170. package/dist/trpc/index.d.ts.map +1 -0
  171. package/dist/trpc/index.js +13 -0
  172. package/dist/trpc/index.js.map +1 -0
  173. package/dist/trpc/middleware.d.ts +8 -0
  174. package/dist/trpc/middleware.d.ts.map +1 -0
  175. package/dist/trpc/middleware.js +12 -0
  176. package/dist/trpc/middleware.js.map +1 -0
  177. package/dist/trpc/router.d.ts +2543 -0
  178. package/dist/trpc/router.d.ts.map +1 -0
  179. package/dist/trpc/router.js +43 -0
  180. package/dist/trpc/router.js.map +1 -0
  181. package/dist/trpc/routers/assets.d.ts +229 -0
  182. package/dist/trpc/routers/assets.d.ts.map +1 -0
  183. package/dist/trpc/routers/assets.js +263 -0
  184. package/dist/trpc/routers/assets.js.map +1 -0
  185. package/dist/trpc/routers/collections.d.ts +88 -0
  186. package/dist/trpc/routers/collections.d.ts.map +1 -0
  187. package/dist/trpc/routers/collections.js +166 -0
  188. package/dist/trpc/routers/collections.js.map +1 -0
  189. package/dist/trpc/routers/costs.d.ts +129 -0
  190. package/dist/trpc/routers/costs.d.ts.map +1 -0
  191. package/dist/trpc/routers/costs.js +95 -0
  192. package/dist/trpc/routers/costs.js.map +1 -0
  193. package/dist/trpc/routers/files.d.ts +34 -0
  194. package/dist/trpc/routers/files.d.ts.map +1 -0
  195. package/dist/trpc/routers/files.js +92 -0
  196. package/dist/trpc/routers/files.js.map +1 -0
  197. package/dist/trpc/routers/jobs.d.ts +92 -0
  198. package/dist/trpc/routers/jobs.d.ts.map +1 -0
  199. package/dist/trpc/routers/jobs.js +100 -0
  200. package/dist/trpc/routers/jobs.js.map +1 -0
  201. package/dist/trpc/routers/mcp-config.d.ts +69 -0
  202. package/dist/trpc/routers/mcp-config.d.ts.map +1 -0
  203. package/dist/trpc/routers/mcp-config.js +290 -0
  204. package/dist/trpc/routers/mcp-config.js.map +1 -0
  205. package/dist/trpc/routers/messages.d.ts +121 -0
  206. package/dist/trpc/routers/messages.d.ts.map +1 -0
  207. package/dist/trpc/routers/messages.js +113 -0
  208. package/dist/trpc/routers/messages.js.map +1 -0
  209. package/dist/trpc/routers/models.d.ts +748 -0
  210. package/dist/trpc/routers/models.d.ts.map +1 -0
  211. package/dist/trpc/routers/models.js +1107 -0
  212. package/dist/trpc/routers/models.js.map +1 -0
  213. package/dist/trpc/routers/nodes.d.ts +48 -0
  214. package/dist/trpc/routers/nodes.d.ts.map +1 -0
  215. package/dist/trpc/routers/nodes.js +57 -0
  216. package/dist/trpc/routers/nodes.js.map +1 -0
  217. package/dist/trpc/routers/sandboxes.d.ts +64 -0
  218. package/dist/trpc/routers/sandboxes.d.ts.map +1 -0
  219. package/dist/trpc/routers/sandboxes.js +327 -0
  220. package/dist/trpc/routers/sandboxes.js.map +1 -0
  221. package/dist/trpc/routers/settings.d.ts +113 -0
  222. package/dist/trpc/routers/settings.d.ts.map +1 -0
  223. package/dist/trpc/routers/settings.js +193 -0
  224. package/dist/trpc/routers/settings.js.map +1 -0
  225. package/dist/trpc/routers/skills.d.ts +41 -0
  226. package/dist/trpc/routers/skills.d.ts.map +1 -0
  227. package/dist/trpc/routers/skills.js +194 -0
  228. package/dist/trpc/routers/skills.js.map +1 -0
  229. package/dist/trpc/routers/storage.d.ts +44 -0
  230. package/dist/trpc/routers/storage.d.ts.map +1 -0
  231. package/dist/trpc/routers/storage.js +153 -0
  232. package/dist/trpc/routers/storage.js.map +1 -0
  233. package/dist/trpc/routers/threads.d.ts +95 -0
  234. package/dist/trpc/routers/threads.d.ts.map +1 -0
  235. package/dist/trpc/routers/threads.js +150 -0
  236. package/dist/trpc/routers/threads.js.map +1 -0
  237. package/dist/trpc/routers/users.d.ts +79 -0
  238. package/dist/trpc/routers/users.d.ts.map +1 -0
  239. package/dist/trpc/routers/users.js +125 -0
  240. package/dist/trpc/routers/users.js.map +1 -0
  241. package/dist/trpc/routers/workflows.d.ts +681 -0
  242. package/dist/trpc/routers/workflows.d.ts.map +1 -0
  243. package/dist/trpc/routers/workflows.js +725 -0
  244. package/dist/trpc/routers/workflows.js.map +1 -0
  245. package/dist/trpc/routers/workspace.d.ts +127 -0
  246. package/dist/trpc/routers/workspace.d.ts.map +1 -0
  247. package/dist/trpc/routers/workspace.js +203 -0
  248. package/dist/trpc/routers/workspace.js.map +1 -0
  249. package/dist/unified-websocket-runner.d.ts +280 -0
  250. package/dist/unified-websocket-runner.d.ts.map +1 -0
  251. package/dist/unified-websocket-runner.js +3840 -0
  252. package/dist/unified-websocket-runner.js.map +1 -0
  253. package/dist/workspace-api.d.ts +14 -0
  254. package/dist/workspace-api.d.ts.map +1 -0
  255. package/dist/workspace-api.js +104 -0
  256. package/dist/workspace-api.js.map +1 -0
  257. package/dist/ws-adapter.d.ts +25 -0
  258. package/dist/ws-adapter.d.ts.map +1 -0
  259. package/dist/ws-adapter.js +63 -0
  260. package/dist/ws-adapter.js.map +1 -0
  261. package/package.json +92 -0
@@ -0,0 +1,166 @@
1
+ /**
2
+ * Collections router — migrated from REST `/api/collections*`.
3
+ *
4
+ * NOTE: the file-upload endpoint `POST /api/collections/:name/index` remains
5
+ * on REST because its multipart/form-data body is not a natural fit for tRPC's
6
+ * JSON link. The other CRUD + query endpoints move here.
7
+ */
8
+ import { Workflow } from "@nodetool-ai/models";
9
+ import { getVecStore, VecNotFoundError } from "@nodetool-ai/vectorstore";
10
+ import { ApiErrorCode } from "../../error-codes.js";
11
+ import { router } from "../index.js";
12
+ import { protectedProcedure } from "../middleware.js";
13
+ import { throwApiError } from "../error-formatter.js";
14
+ import { listOutput, getInput, collectionResponse, createInput, updateInput, deleteInput, deleteOutput, queryInput, queryOutput } from "@nodetool-ai/protocol/api-schemas/collections.js";
15
+ /**
16
+ * Normalize a CollectionMetadata (may contain `undefined` values per the
17
+ * vectorstore type) to the wire schema (string | number | boolean only).
18
+ */
19
+ function normalizeMetadata(metadata) {
20
+ const result = {};
21
+ if (!metadata)
22
+ return result;
23
+ for (const [key, value] of Object.entries(metadata)) {
24
+ if (value !== undefined)
25
+ result[key] = value;
26
+ }
27
+ return result;
28
+ }
29
+ /**
30
+ * Helper: resolve a workflow's name from an id. Returns `null` on any
31
+ * lookup failure. Mirrors the REST handler's forgiving behaviour.
32
+ */
33
+ async function resolveWorkflowName(workflowId) {
34
+ if (!workflowId)
35
+ return null;
36
+ try {
37
+ const workflow = (await Workflow.get(workflowId));
38
+ return workflow?.name ?? null;
39
+ }
40
+ catch {
41
+ return null;
42
+ }
43
+ }
44
+ /** Map VecNotFoundError → tRPC NOT_FOUND. Re-throws anything else. */
45
+ function rethrowAsTrpc(err) {
46
+ if (err instanceof VecNotFoundError) {
47
+ throwApiError(ApiErrorCode.NOT_FOUND, "Collection not found");
48
+ }
49
+ throw err;
50
+ }
51
+ export const collectionsRouter = router({
52
+ list: protectedProcedure.output(listOutput).query(async () => {
53
+ const store = await getVecStore();
54
+ const collections = await store.listCollections();
55
+ const results = await Promise.all(collections.map(async (col) => {
56
+ const count = await col.count();
57
+ const metadata = normalizeMetadata(col.metadata);
58
+ const workflowName = await resolveWorkflowName(typeof metadata.workflow === "string" ? metadata.workflow : undefined);
59
+ return {
60
+ name: col.name,
61
+ count,
62
+ metadata,
63
+ workflow_name: workflowName
64
+ };
65
+ }));
66
+ return { collections: results, count: results.length };
67
+ }),
68
+ get: protectedProcedure
69
+ .input(getInput)
70
+ .output(collectionResponse)
71
+ .query(async ({ input }) => {
72
+ const store = await getVecStore();
73
+ try {
74
+ const collection = await store.getCollection({ name: input.name });
75
+ const count = await collection.count();
76
+ return {
77
+ name: collection.name,
78
+ metadata: normalizeMetadata(collection.metadata),
79
+ count
80
+ };
81
+ }
82
+ catch (err) {
83
+ rethrowAsTrpc(err);
84
+ }
85
+ }),
86
+ create: protectedProcedure
87
+ .input(createInput)
88
+ .output(collectionResponse)
89
+ .mutation(async ({ input }) => {
90
+ const store = await getVecStore();
91
+ const metadata = {};
92
+ if (input.embedding_model) {
93
+ metadata.embedding_model = input.embedding_model;
94
+ }
95
+ if (input.embedding_provider) {
96
+ metadata.embedding_provider = input.embedding_provider;
97
+ }
98
+ const collection = await store.createCollection({
99
+ name: input.name,
100
+ metadata
101
+ });
102
+ return {
103
+ name: collection.name,
104
+ metadata: normalizeMetadata(collection.metadata),
105
+ count: 0
106
+ };
107
+ }),
108
+ update: protectedProcedure
109
+ .input(updateInput)
110
+ .output(collectionResponse)
111
+ .mutation(async ({ input }) => {
112
+ const store = await getVecStore();
113
+ let collection;
114
+ try {
115
+ collection = await store.getCollection({ name: input.name });
116
+ }
117
+ catch (err) {
118
+ rethrowAsTrpc(err);
119
+ }
120
+ const existing = normalizeMetadata(collection.metadata);
121
+ const merged = { ...existing };
122
+ if (input.metadata) {
123
+ Object.assign(merged, input.metadata);
124
+ }
125
+ const newName = input.rename ?? collection.name;
126
+ await collection.modify({ name: newName, metadata: merged });
127
+ const count = await collection.count();
128
+ return {
129
+ name: newName,
130
+ metadata: merged,
131
+ count
132
+ };
133
+ }),
134
+ delete: protectedProcedure
135
+ .input(deleteInput)
136
+ .output(deleteOutput)
137
+ .mutation(async ({ input }) => {
138
+ const store = await getVecStore();
139
+ try {
140
+ await store.deleteCollection({ name: input.name });
141
+ }
142
+ catch (err) {
143
+ rethrowAsTrpc(err);
144
+ }
145
+ return { message: `Collection ${input.name} deleted successfully` };
146
+ }),
147
+ query: protectedProcedure
148
+ .input(queryInput)
149
+ .output(queryOutput)
150
+ .query(async ({ input }) => {
151
+ const store = await getVecStore();
152
+ let collection;
153
+ try {
154
+ collection = await store.getCollection({ name: input.name });
155
+ }
156
+ catch (err) {
157
+ rethrowAsTrpc(err);
158
+ }
159
+ const results = await collection.query({
160
+ queryTexts: input.query_texts,
161
+ nResults: input.n_results
162
+ });
163
+ return results;
164
+ })
165
+ });
166
+ //# sourceMappingURL=collections.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"collections.js","sourceRoot":"","sources":["../../../src/trpc/routers/collections.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAE,QAAQ,EAAE,MAAM,qBAAqB,CAAC;AAC/C,OAAO,EACL,WAAW,EACX,gBAAgB,EAEjB,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AACpD,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,OAAO,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AACtD,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,EACL,UAAU,EACV,QAAQ,EACR,kBAAkB,EAClB,WAAW,EACX,WAAW,EACX,WAAW,EACX,YAAY,EACZ,UAAU,EACV,WAAW,EACZ,MAAM,kDAAkD,CAAC;AAE1D;;;GAGG;AACH,SAAS,iBAAiB,CACxB,QAAwC;IAExC,MAAM,MAAM,GAA8C,EAAE,CAAC;IAC7D,IAAI,CAAC,QAAQ;QAAE,OAAO,MAAM,CAAC;IAC7B,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QACpD,IAAI,KAAK,KAAK,SAAS;YAAE,MAAM,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;IAC/C,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;GAGG;AACH,KAAK,UAAU,mBAAmB,CAChC,UAA8B;IAE9B,IAAI,CAAC,UAAU;QAAE,OAAO,IAAI,CAAC;IAC7B,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,CAAC,MAAM,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,CAExC,CAAC;QACT,OAAO,QAAQ,EAAE,IAAI,IAAI,IAAI,CAAC;IAChC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,sEAAsE;AACtE,SAAS,aAAa,CAAC,GAAY;IACjC,IAAI,GAAG,YAAY,gBAAgB,EAAE,CAAC;QACpC,aAAa,CAAC,YAAY,CAAC,SAAS,EAAE,sBAAsB,CAAC,CAAC;IAChE,CAAC;IACD,MAAM,GAAG,CAAC;AACZ,CAAC;AAED,MAAM,CAAC,MAAM,iBAAiB,GAAG,MAAM,CAAC;IACtC,IAAI,EAAE,kBAAkB,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC,KAAK,IAAI,EAAE;QAC3D,MAAM,KAAK,GAAG,MAAM,WAAW,EAAE,CAAC;QAClC,MAAM,WAAW,GAAG,MAAM,KAAK,CAAC,eAAe,EAAE,CAAC;QAElD,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAC/B,WAAW,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;YAC5B,MAAM,KAAK,GAAG,MAAM,GAAG,CAAC,KAAK,EAAE,CAAC;YAChC,MAAM,QAAQ,GAAG,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACjD,MAAM,YAAY,GAAG,MAAM,mBAAmB,CAC5C,OAAO,QAAQ,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CACtE,CAAC;YACF,OAAO;gBACL,IAAI,EAAE,GAAG,CAAC,IAAI;gBACd,KAAK;gBACL,QAAQ;gBACR,aAAa,EAAE,YAAY;aAC5B,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;QAEF,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC;IACzD,CAAC,CAAC;IAEF,GAAG,EAAE,kBAAkB;SACpB,KAAK,CAAC,QAAQ,CAAC;SACf,MAAM,CAAC,kBAAkB,CAAC;SAC1B,KAAK,CAAC,KAAK,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE;QACzB,MAAM,KAAK,GAAG,MAAM,WAAW,EAAE,CAAC;QAClC,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,MAAM,KAAK,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;YACnE,MAAM,KAAK,GAAG,MAAM,UAAU,CAAC,KAAK,EAAE,CAAC;YACvC,OAAO;gBACL,IAAI,EAAE,UAAU,CAAC,IAAI;gBACrB,QAAQ,EAAE,iBAAiB,CAAC,UAAU,CAAC,QAAQ,CAAC;gBAChD,KAAK;aACN,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,aAAa,CAAC,GAAG,CAAC,CAAC;QACrB,CAAC;IACH,CAAC,CAAC;IAEJ,MAAM,EAAE,kBAAkB;SACvB,KAAK,CAAC,WAAW,CAAC;SAClB,MAAM,CAAC,kBAAkB,CAAC;SAC1B,QAAQ,CAAC,KAAK,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE;QAC5B,MAAM,KAAK,GAAG,MAAM,WAAW,EAAE,CAAC;QAClC,MAAM,QAAQ,GAA2B,EAAE,CAAC;QAC5C,IAAI,KAAK,CAAC,eAAe,EAAE,CAAC;YAC1B,QAAQ,CAAC,eAAe,GAAG,KAAK,CAAC,eAAe,CAAC;QACnD,CAAC;QACD,IAAI,KAAK,CAAC,kBAAkB,EAAE,CAAC;YAC7B,QAAQ,CAAC,kBAAkB,GAAG,KAAK,CAAC,kBAAkB,CAAC;QACzD,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,KAAK,CAAC,gBAAgB,CAAC;YAC9C,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,QAAQ;SACT,CAAC,CAAC;QAEH,OAAO;YACL,IAAI,EAAE,UAAU,CAAC,IAAI;YACrB,QAAQ,EAAE,iBAAiB,CAAC,UAAU,CAAC,QAAQ,CAAC;YAChD,KAAK,EAAE,CAAC;SACT,CAAC;IACJ,CAAC,CAAC;IAEJ,MAAM,EAAE,kBAAkB;SACvB,KAAK,CAAC,WAAW,CAAC;SAClB,MAAM,CAAC,kBAAkB,CAAC;SAC1B,QAAQ,CAAC,KAAK,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE;QAC5B,MAAM,KAAK,GAAG,MAAM,WAAW,EAAE,CAAC;QAClC,IAAI,UAAU,CAAC;QACf,IAAI,CAAC;YACH,UAAU,GAAG,MAAM,KAAK,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;QAC/D,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,aAAa,CAAC,GAAG,CAAC,CAAC;QACrB,CAAC;QAED,MAAM,QAAQ,GAAG,iBAAiB,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACxD,MAAM,MAAM,GAAG,EAAE,GAAG,QAAQ,EAAE,CAAC;QAC/B,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;YACnB,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;QACxC,CAAC;QAED,MAAM,OAAO,GAAG,KAAK,CAAC,MAAM,IAAI,UAAU,CAAC,IAAI,CAAC;QAChD,MAAM,UAAU,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;QAE7D,MAAM,KAAK,GAAG,MAAM,UAAU,CAAC,KAAK,EAAE,CAAC;QACvC,OAAO;YACL,IAAI,EAAE,OAAO;YACb,QAAQ,EAAE,MAAM;YAChB,KAAK;SACN,CAAC;IACJ,CAAC,CAAC;IAEJ,MAAM,EAAE,kBAAkB;SACvB,KAAK,CAAC,WAAW,CAAC;SAClB,MAAM,CAAC,YAAY,CAAC;SACpB,QAAQ,CAAC,KAAK,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE;QAC5B,MAAM,KAAK,GAAG,MAAM,WAAW,EAAE,CAAC;QAClC,IAAI,CAAC;YACH,MAAM,KAAK,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;QACrD,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,aAAa,CAAC,GAAG,CAAC,CAAC;QACrB,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,cAAc,KAAK,CAAC,IAAI,uBAAuB,EAAE,CAAC;IACtE,CAAC,CAAC;IAEJ,KAAK,EAAE,kBAAkB;SACtB,KAAK,CAAC,UAAU,CAAC;SACjB,MAAM,CAAC,WAAW,CAAC;SACnB,KAAK,CAAC,KAAK,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE;QACzB,MAAM,KAAK,GAAG,MAAM,WAAW,EAAE,CAAC;QAClC,IAAI,UAAU,CAAC;QACf,IAAI,CAAC;YACH,UAAU,GAAG,MAAM,KAAK,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;QAC/D,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,aAAa,CAAC,GAAG,CAAC,CAAC;QACrB,CAAC;QACD,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC;YACrC,UAAU,EAAE,KAAK,CAAC,WAAW;YAC7B,QAAQ,EAAE,KAAK,CAAC,SAAS;SAC1B,CAAC,CAAC;QACH,OAAO,OAAO,CAAC;IACjB,CAAC,CAAC;CACL,CAAC,CAAC"}
@@ -0,0 +1,129 @@
1
+ export declare const costsRouter: import("@trpc/server").TRPCBuiltRouter<{
2
+ ctx: import("../context.js").Context;
3
+ meta: object;
4
+ errorShape: import("../error-formatter.js").ApiErrorShape;
5
+ transformer: true;
6
+ }, import("@trpc/server").TRPCDecorateCreateRouterOptions<{
7
+ list: import("@trpc/server").TRPCQueryProcedure<{
8
+ input: {
9
+ provider?: string | undefined;
10
+ model?: string | undefined;
11
+ limit?: number | undefined;
12
+ startKey?: string | undefined;
13
+ };
14
+ output: {
15
+ calls: {
16
+ id: string;
17
+ user_id: string;
18
+ node_id: string;
19
+ provider: string;
20
+ model: string;
21
+ workflow_id: string | null;
22
+ cost: number | null;
23
+ input_tokens: number | null;
24
+ output_tokens: number | null;
25
+ total_tokens: number | null;
26
+ cached_tokens: number | null;
27
+ reasoning_tokens: number | null;
28
+ created_at: string | null;
29
+ metadata: Record<string, unknown> | null;
30
+ }[];
31
+ next_start_key: string | null;
32
+ };
33
+ meta: object;
34
+ }>;
35
+ aggregate: import("@trpc/server").TRPCQueryProcedure<{
36
+ input: {
37
+ provider?: string | undefined;
38
+ model?: string | undefined;
39
+ };
40
+ output: {
41
+ user_id: string;
42
+ provider: string | null;
43
+ model: string | null;
44
+ total_cost: number;
45
+ total_input_tokens: number;
46
+ total_output_tokens: number;
47
+ total_tokens: number;
48
+ call_count: number;
49
+ };
50
+ meta: object;
51
+ }>;
52
+ aggregateByProvider: import("@trpc/server").TRPCQueryProcedure<{
53
+ input: void;
54
+ output: {
55
+ provider: string;
56
+ total_cost: number;
57
+ total_input_tokens: number;
58
+ total_output_tokens: number;
59
+ total_tokens: number;
60
+ call_count: number;
61
+ }[];
62
+ meta: object;
63
+ }>;
64
+ aggregateByModel: import("@trpc/server").TRPCQueryProcedure<{
65
+ input: {
66
+ provider?: string | undefined;
67
+ };
68
+ output: {
69
+ provider: string;
70
+ model: string;
71
+ total_cost: number;
72
+ total_input_tokens: number;
73
+ total_output_tokens: number;
74
+ total_tokens: number;
75
+ call_count: number;
76
+ }[];
77
+ meta: object;
78
+ }>;
79
+ summary: import("@trpc/server").TRPCQueryProcedure<{
80
+ input: void;
81
+ output: {
82
+ overall: {
83
+ user_id: string;
84
+ provider: string | null;
85
+ model: string | null;
86
+ total_cost: number;
87
+ total_input_tokens: number;
88
+ total_output_tokens: number;
89
+ total_tokens: number;
90
+ call_count: number;
91
+ };
92
+ by_provider: {
93
+ provider: string;
94
+ total_cost: number;
95
+ total_input_tokens: number;
96
+ total_output_tokens: number;
97
+ total_tokens: number;
98
+ call_count: number;
99
+ }[];
100
+ by_model: {
101
+ provider: string;
102
+ model: string;
103
+ total_cost: number;
104
+ total_input_tokens: number;
105
+ total_output_tokens: number;
106
+ total_tokens: number;
107
+ call_count: number;
108
+ }[];
109
+ recent_calls: {
110
+ id: string;
111
+ user_id: string;
112
+ node_id: string;
113
+ provider: string;
114
+ model: string;
115
+ workflow_id: string | null;
116
+ cost: number | null;
117
+ input_tokens: number | null;
118
+ output_tokens: number | null;
119
+ total_tokens: number | null;
120
+ cached_tokens: number | null;
121
+ reasoning_tokens: number | null;
122
+ created_at: string | null;
123
+ metadata: Record<string, unknown> | null;
124
+ }[];
125
+ };
126
+ meta: object;
127
+ }>;
128
+ }>>;
129
+ //# sourceMappingURL=costs.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"costs.d.ts","sourceRoot":"","sources":["../../../src/trpc/routers/costs.ts"],"names":[],"mappings":"AAqDA,eAAO,MAAM,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA0DtB,CAAC"}
@@ -0,0 +1,95 @@
1
+ import { Prediction } from "@nodetool-ai/models";
2
+ import { router } from "../index.js";
3
+ import { protectedProcedure } from "../middleware.js";
4
+ import { listPredictionsInput, listPredictionsOutput, aggregateInput, aggregateByUserOutput, aggregateByProviderOutput, aggregateByModelInput, aggregateByModelOutput, summaryOutput } from "@nodetool-ai/protocol/api-schemas/costs.js";
5
+ function toPredictionResponse(pred) {
6
+ return {
7
+ id: pred.id,
8
+ user_id: pred.user_id,
9
+ node_id: pred.node_id ?? "",
10
+ provider: pred.provider,
11
+ model: pred.model,
12
+ workflow_id: pred.workflow_id ?? null,
13
+ cost: pred.cost ?? null,
14
+ input_tokens: pred.input_tokens ?? null,
15
+ output_tokens: pred.output_tokens ?? null,
16
+ total_tokens: pred.total_tokens ?? null,
17
+ cached_tokens: pred.cached_tokens ?? null,
18
+ reasoning_tokens: pred.reasoning_tokens ?? null,
19
+ created_at: pred.created_at,
20
+ metadata: pred.metadata ?? null
21
+ };
22
+ }
23
+ /**
24
+ * Normalize `AggregateResult` → `AggregateByUserOutput` by coercing optional
25
+ * `provider`/`model` (which can be `undefined`) to `null` so the shape matches
26
+ * the zod schema. Mocks in tests may return either shape.
27
+ */
28
+ function normalizeAggregate(agg) {
29
+ return {
30
+ user_id: agg.user_id,
31
+ provider: agg.provider ?? null,
32
+ model: agg.model ?? null,
33
+ total_cost: agg.total_cost,
34
+ total_input_tokens: agg.total_input_tokens,
35
+ total_output_tokens: agg.total_output_tokens,
36
+ total_tokens: agg.total_tokens,
37
+ call_count: agg.call_count
38
+ };
39
+ }
40
+ export const costsRouter = router({
41
+ list: protectedProcedure
42
+ .input(listPredictionsInput)
43
+ .output(listPredictionsOutput)
44
+ .query(async ({ ctx, input }) => {
45
+ const [calls, nextKey] = await Prediction.paginate(ctx.userId, {
46
+ provider: input.provider,
47
+ model: input.model,
48
+ limit: input.limit,
49
+ startKey: input.startKey
50
+ });
51
+ return {
52
+ calls: calls.map(toPredictionResponse),
53
+ next_start_key: nextKey || null
54
+ };
55
+ }),
56
+ aggregate: protectedProcedure
57
+ .input(aggregateInput)
58
+ .output(aggregateByUserOutput)
59
+ .query(async ({ ctx, input }) => {
60
+ const result = await Prediction.aggregateByUser(ctx.userId, {
61
+ provider: input.provider,
62
+ model: input.model
63
+ });
64
+ return normalizeAggregate(result);
65
+ }),
66
+ aggregateByProvider: protectedProcedure
67
+ .output(aggregateByProviderOutput)
68
+ .query(async ({ ctx }) => {
69
+ return Prediction.aggregateByProvider(ctx.userId);
70
+ }),
71
+ aggregateByModel: protectedProcedure
72
+ .input(aggregateByModelInput)
73
+ .output(aggregateByModelOutput)
74
+ .query(async ({ ctx, input }) => {
75
+ return Prediction.aggregateByModel(ctx.userId, {
76
+ provider: input.provider
77
+ });
78
+ }),
79
+ summary: protectedProcedure.output(summaryOutput).query(async ({ ctx }) => {
80
+ const [overall, byProvider, byModel, paginated] = await Promise.all([
81
+ Prediction.aggregateByUser(ctx.userId),
82
+ Prediction.aggregateByProvider(ctx.userId),
83
+ Prediction.aggregateByModel(ctx.userId),
84
+ Prediction.paginate(ctx.userId, { limit: 10 })
85
+ ]);
86
+ const [recentCalls] = paginated;
87
+ return {
88
+ overall: normalizeAggregate(overall),
89
+ by_provider: byProvider,
90
+ by_model: byModel,
91
+ recent_calls: recentCalls.map(toPredictionResponse)
92
+ };
93
+ })
94
+ });
95
+ //# sourceMappingURL=costs.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"costs.js","sourceRoot":"","sources":["../../../src/trpc/routers/costs.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAwB,MAAM,qBAAqB,CAAC;AACvE,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,OAAO,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AACtD,OAAO,EACL,oBAAoB,EACpB,qBAAqB,EACrB,cAAc,EACd,qBAAqB,EACrB,yBAAyB,EACzB,qBAAqB,EACrB,sBAAsB,EACtB,aAAa,EAGd,MAAM,4CAA4C,CAAC;AAEpD,SAAS,oBAAoB,CAAC,IAAgB;IAC5C,OAAO;QACL,EAAE,EAAE,IAAI,CAAC,EAAE;QACX,OAAO,EAAE,IAAI,CAAC,OAAO;QACrB,OAAO,EAAE,IAAI,CAAC,OAAO,IAAI,EAAE;QAC3B,QAAQ,EAAE,IAAI,CAAC,QAAQ;QACvB,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,IAAI;QACrC,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,IAAI;QACvB,YAAY,EAAE,IAAI,CAAC,YAAY,IAAI,IAAI;QACvC,aAAa,EAAE,IAAI,CAAC,aAAa,IAAI,IAAI;QACzC,YAAY,EAAE,IAAI,CAAC,YAAY,IAAI,IAAI;QACvC,aAAa,EAAE,IAAI,CAAC,aAAa,IAAI,IAAI;QACzC,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,IAAI,IAAI;QAC/C,UAAU,EAAE,IAAI,CAAC,UAAU;QAC3B,QAAQ,EAAE,IAAI,CAAC,QAAQ,IAAI,IAAI;KAChC,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,SAAS,kBAAkB,CAAC,GAAoB;IAC9C,OAAO;QACL,OAAO,EAAE,GAAG,CAAC,OAAO;QACpB,QAAQ,EAAE,GAAG,CAAC,QAAQ,IAAI,IAAI;QAC9B,KAAK,EAAE,GAAG,CAAC,KAAK,IAAI,IAAI;QACxB,UAAU,EAAE,GAAG,CAAC,UAAU;QAC1B,kBAAkB,EAAE,GAAG,CAAC,kBAAkB;QAC1C,mBAAmB,EAAE,GAAG,CAAC,mBAAmB;QAC5C,YAAY,EAAE,GAAG,CAAC,YAAY;QAC9B,UAAU,EAAE,GAAG,CAAC,UAAU;KAC3B,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,MAAM,WAAW,GAAG,MAAM,CAAC;IAChC,IAAI,EAAE,kBAAkB;SACrB,KAAK,CAAC,oBAAoB,CAAC;SAC3B,MAAM,CAAC,qBAAqB,CAAC;SAC7B,KAAK,CAAC,KAAK,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE;QAC9B,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,GAAG,MAAM,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,EAAE;YAC7D,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,QAAQ,EAAE,KAAK,CAAC,QAAQ;SACzB,CAAC,CAAC;QACH,OAAO;YACL,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,oBAAoB,CAAC;YACtC,cAAc,EAAE,OAAO,IAAI,IAAI;SAChC,CAAC;IACJ,CAAC,CAAC;IAEJ,SAAS,EAAE,kBAAkB;SAC1B,KAAK,CAAC,cAAc,CAAC;SACrB,MAAM,CAAC,qBAAqB,CAAC;SAC7B,KAAK,CAAC,KAAK,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE;QAC9B,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,eAAe,CAAC,GAAG,CAAC,MAAM,EAAE;YAC1D,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,KAAK,EAAE,KAAK,CAAC,KAAK;SACnB,CAAC,CAAC;QACH,OAAO,kBAAkB,CAAC,MAAM,CAAC,CAAC;IACpC,CAAC,CAAC;IAEJ,mBAAmB,EAAE,kBAAkB;SACpC,MAAM,CAAC,yBAAyB,CAAC;SACjC,KAAK,CAAC,KAAK,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE;QACvB,OAAO,UAAU,CAAC,mBAAmB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IACpD,CAAC,CAAC;IAEJ,gBAAgB,EAAE,kBAAkB;SACjC,KAAK,CAAC,qBAAqB,CAAC;SAC5B,MAAM,CAAC,sBAAsB,CAAC;SAC9B,KAAK,CAAC,KAAK,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE;QAC9B,OAAO,UAAU,CAAC,gBAAgB,CAAC,GAAG,CAAC,MAAM,EAAE;YAC7C,QAAQ,EAAE,KAAK,CAAC,QAAQ;SACzB,CAAC,CAAC;IACL,CAAC,CAAC;IAEJ,OAAO,EAAE,kBAAkB,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE;QACxE,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,SAAS,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YAClE,UAAU,CAAC,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC;YACtC,UAAU,CAAC,mBAAmB,CAAC,GAAG,CAAC,MAAM,CAAC;YAC1C,UAAU,CAAC,gBAAgB,CAAC,GAAG,CAAC,MAAM,CAAC;YACvC,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;SAC/C,CAAC,CAAC;QACH,MAAM,CAAC,WAAW,CAAC,GAAG,SAAS,CAAC;QAChC,OAAO;YACL,OAAO,EAAE,kBAAkB,CAAC,OAAO,CAAC;YACpC,WAAW,EAAE,UAAU;YACvB,QAAQ,EAAE,OAAO;YACjB,YAAY,EAAE,WAAW,CAAC,GAAG,CAAC,oBAAoB,CAAC;SACpD,CAAC;IACJ,CAAC,CAAC;CACH,CAAC,CAAC"}
@@ -0,0 +1,34 @@
1
+ export declare const filesRouter: import("@trpc/server").TRPCBuiltRouter<{
2
+ ctx: import("../context.js").Context;
3
+ meta: object;
4
+ errorShape: import("../error-formatter.js").ApiErrorShape;
5
+ transformer: true;
6
+ }, import("@trpc/server").TRPCDecorateCreateRouterOptions<{
7
+ list: import("@trpc/server").TRPCQueryProcedure<{
8
+ input: {
9
+ path: string;
10
+ };
11
+ output: {
12
+ name: string;
13
+ path: string;
14
+ size: number;
15
+ is_dir: boolean;
16
+ modified_at: string;
17
+ }[];
18
+ meta: object;
19
+ }>;
20
+ info: import("@trpc/server").TRPCQueryProcedure<{
21
+ input: {
22
+ path: string;
23
+ };
24
+ output: {
25
+ name: string;
26
+ path: string;
27
+ size: number;
28
+ is_dir: boolean;
29
+ modified_at: string;
30
+ };
31
+ meta: object;
32
+ }>;
33
+ }>>;
34
+ //# sourceMappingURL=files.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"files.d.ts","sourceRoot":"","sources":["../../../src/trpc/routers/files.ts"],"names":[],"mappings":"AA8CA,eAAO,MAAM,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAsEtB,CAAC"}
@@ -0,0 +1,92 @@
1
+ /**
2
+ * tRPC router for the files domain — local filesystem browser (JSON ops).
3
+ * Binary download (/api/files/download) stays as REST.
4
+ */
5
+ import * as fs from "node:fs/promises";
6
+ import * as path from "node:path";
7
+ import * as os from "node:os";
8
+ import { router } from "../index.js";
9
+ import { protectedProcedure } from "../middleware.js";
10
+ import { throwApiError } from "../error-formatter.js";
11
+ import { listFilesInput, listFilesOutput, fileInfoInput, fileInfoOutput } from "@nodetool-ai/protocol/api-schemas/files.js";
12
+ import { ApiErrorCode } from "@nodetool-ai/protocol/api-schemas/api-error-code.js";
13
+ // ── Sandbox helpers ─────────────────────────────────────────────────────────
14
+ function getRootDir() {
15
+ return os.homedir();
16
+ }
17
+ /**
18
+ * Resolve a user-provided path within the sandbox root.
19
+ * Returns the resolved absolute path, or throws FORBIDDEN if path escapes.
20
+ */
21
+ function resolveSandboxed(rootDir, userPath) {
22
+ const resolved = path.resolve(rootDir, userPath.startsWith("/") ? "." + userPath : userPath);
23
+ const normalizedRoot = path.resolve(rootDir);
24
+ if (!resolved.startsWith(normalizedRoot + path.sep) &&
25
+ resolved !== normalizedRoot) {
26
+ throwApiError(ApiErrorCode.FORBIDDEN, "Path traversal not allowed");
27
+ }
28
+ return resolved;
29
+ }
30
+ // ── Router ──────────────────────────────────────────────────────────────────
31
+ export const filesRouter = router({
32
+ list: protectedProcedure
33
+ .input(listFilesInput)
34
+ .output(listFilesOutput)
35
+ .query(async ({ input }) => {
36
+ if (process.env["NODETOOL_ENV"] === "production") {
37
+ throwApiError(ApiErrorCode.FORBIDDEN, "File browser is disabled in production");
38
+ }
39
+ const rootDir = getRootDir();
40
+ const resolved = resolveSandboxed(rootDir, input.path);
41
+ try {
42
+ const entries = await fs.readdir(resolved, { withFileTypes: true });
43
+ return Promise.all(entries.map(async (entry) => {
44
+ const fullPath = path.join(resolved, entry.name);
45
+ let size = 0;
46
+ let modifiedAt = "";
47
+ try {
48
+ const stat = await fs.stat(fullPath);
49
+ size = stat.size;
50
+ modifiedAt = stat.mtime.toISOString();
51
+ }
52
+ catch {
53
+ // stat may fail for broken symlinks — return defaults
54
+ }
55
+ return {
56
+ name: entry.name,
57
+ path: fullPath,
58
+ size,
59
+ is_dir: entry.isDirectory(),
60
+ modified_at: modifiedAt
61
+ };
62
+ }));
63
+ }
64
+ catch {
65
+ throwApiError(ApiErrorCode.NOT_FOUND, "Directory not found");
66
+ }
67
+ }),
68
+ info: protectedProcedure
69
+ .input(fileInfoInput)
70
+ .output(fileInfoOutput)
71
+ .query(async ({ input }) => {
72
+ if (process.env["NODETOOL_ENV"] === "production") {
73
+ throwApiError(ApiErrorCode.FORBIDDEN, "File browser is disabled in production");
74
+ }
75
+ const rootDir = getRootDir();
76
+ const resolved = resolveSandboxed(rootDir, input.path);
77
+ try {
78
+ const stat = await fs.stat(resolved);
79
+ return {
80
+ name: path.basename(resolved),
81
+ path: resolved,
82
+ size: stat.size,
83
+ is_dir: stat.isDirectory(),
84
+ modified_at: stat.mtime.toISOString()
85
+ };
86
+ }
87
+ catch {
88
+ throwApiError(ApiErrorCode.NOT_FOUND, "File not found");
89
+ }
90
+ })
91
+ });
92
+ //# sourceMappingURL=files.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"files.js","sourceRoot":"","sources":["../../../src/trpc/routers/files.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,OAAO,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACvC,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAE9B,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,OAAO,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AACtD,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,EACL,cAAc,EACd,eAAe,EACf,aAAa,EACb,cAAc,EACf,MAAM,4CAA4C,CAAC;AACpD,OAAO,EAAE,YAAY,EAAE,MAAM,qDAAqD,CAAC;AAEnF,+EAA+E;AAE/E,SAAS,UAAU;IACjB,OAAO,EAAE,CAAC,OAAO,EAAE,CAAC;AACtB,CAAC;AAED;;;GAGG;AACH,SAAS,gBAAgB,CAAC,OAAe,EAAE,QAAgB;IACzD,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAC3B,OAAO,EACP,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,GAAG,QAAQ,CAAC,CAAC,CAAC,QAAQ,CACrD,CAAC;IACF,MAAM,cAAc,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAC7C,IACE,CAAC,QAAQ,CAAC,UAAU,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC;QAC/C,QAAQ,KAAK,cAAc,EAC3B,CAAC;QACD,aAAa,CAAC,YAAY,CAAC,SAAS,EAAE,4BAA4B,CAAC,CAAC;IACtE,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,+EAA+E;AAE/E,MAAM,CAAC,MAAM,WAAW,GAAG,MAAM,CAAC;IAChC,IAAI,EAAE,kBAAkB;SACrB,KAAK,CAAC,cAAc,CAAC;SACrB,MAAM,CAAC,eAAe,CAAC;SACvB,KAAK,CAAC,KAAK,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE;QACzB,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,KAAK,YAAY,EAAE,CAAC;YACjD,aAAa,CACX,YAAY,CAAC,SAAS,EACtB,wCAAwC,CACzC,CAAC;QACJ,CAAC;QAED,MAAM,OAAO,GAAG,UAAU,EAAE,CAAC;QAC7B,MAAM,QAAQ,GAAG,gBAAgB,CAAC,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;QAEvD,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;YACpE,OAAO,OAAO,CAAC,GAAG,CAChB,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;gBAC1B,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;gBACjD,IAAI,IAAI,GAAG,CAAC,CAAC;gBACb,IAAI,UAAU,GAAG,EAAE,CAAC;gBACpB,IAAI,CAAC;oBACH,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;oBACrC,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;oBACjB,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;gBACxC,CAAC;gBAAC,MAAM,CAAC;oBACP,sDAAsD;gBACxD,CAAC;gBACD,OAAO;oBACL,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,IAAI,EAAE,QAAQ;oBACd,IAAI;oBACJ,MAAM,EAAE,KAAK,CAAC,WAAW,EAAE;oBAC3B,WAAW,EAAE,UAAU;iBACxB,CAAC;YACJ,CAAC,CAAC,CACH,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,aAAa,CAAC,YAAY,CAAC,SAAS,EAAE,qBAAqB,CAAC,CAAC;QAC/D,CAAC;IACH,CAAC,CAAC;IAEJ,IAAI,EAAE,kBAAkB;SACrB,KAAK,CAAC,aAAa,CAAC;SACpB,MAAM,CAAC,cAAc,CAAC;SACtB,KAAK,CAAC,KAAK,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE;QACzB,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,KAAK,YAAY,EAAE,CAAC;YACjD,aAAa,CACX,YAAY,CAAC,SAAS,EACtB,wCAAwC,CACzC,CAAC;QACJ,CAAC;QAED,MAAM,OAAO,GAAG,UAAU,EAAE,CAAC;QAC7B,MAAM,QAAQ,GAAG,gBAAgB,CAAC,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;QAEvD,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACrC,OAAO;gBACL,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC;gBAC7B,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,MAAM,EAAE,IAAI,CAAC,WAAW,EAAE;gBAC1B,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE;aACtC,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,aAAa,CAAC,YAAY,CAAC,SAAS,EAAE,gBAAgB,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC,CAAC;CACL,CAAC,CAAC"}
@@ -0,0 +1,92 @@
1
+ /**
2
+ * Jobs router — migrated from REST `/api/jobs*`.
3
+ *
4
+ * Ported verbatim from the legacy http-api.ts handlers (`handleJobsRoot`,
5
+ * `handleJobById`, `handleJobCancel`, and the inline `/running/all` handler).
6
+ * User ownership is enforced on every procedure — a job whose `user_id`
7
+ * doesn't match `ctx.userId` is indistinguishable from a missing one.
8
+ *
9
+ * The trigger stubs (`/api/jobs/triggers/*`) stay on REST — they're
10
+ * 501-placeholders for a feature that isn't available in standalone mode.
11
+ */
12
+ export declare const jobsRouter: import("@trpc/server").TRPCBuiltRouter<{
13
+ ctx: import("../context.js").Context;
14
+ meta: object;
15
+ errorShape: import("../error-formatter.js").ApiErrorShape;
16
+ transformer: true;
17
+ }, import("@trpc/server").TRPCDecorateCreateRouterOptions<{
18
+ list: import("@trpc/server").TRPCQueryProcedure<{
19
+ input: {
20
+ limit?: number | undefined;
21
+ workflow_id?: string | undefined;
22
+ };
23
+ output: {
24
+ jobs: {
25
+ id: string;
26
+ user_id: string;
27
+ job_type: "workflow";
28
+ status: string;
29
+ workflow_id: string;
30
+ started_at: string | null;
31
+ finished_at: string | null;
32
+ error: string | null;
33
+ cost: number | null;
34
+ }[];
35
+ next_start_key: string | null;
36
+ };
37
+ meta: object;
38
+ }>;
39
+ runningAll: import("@trpc/server").TRPCQueryProcedure<{
40
+ input: void;
41
+ output: {
42
+ job_id: string;
43
+ status: string;
44
+ workflow_id: string;
45
+ created_at: string | null;
46
+ is_running: boolean;
47
+ is_completed: boolean;
48
+ }[];
49
+ meta: object;
50
+ }>;
51
+ get: import("@trpc/server").TRPCQueryProcedure<{
52
+ input: {
53
+ id: string;
54
+ };
55
+ output: {
56
+ id: string;
57
+ user_id: string;
58
+ job_type: "workflow";
59
+ status: string;
60
+ workflow_id: string;
61
+ started_at: string | null;
62
+ finished_at: string | null;
63
+ error: string | null;
64
+ cost: number | null;
65
+ };
66
+ meta: object;
67
+ }>;
68
+ delete: import("@trpc/server").TRPCMutationProcedure<{
69
+ input: {
70
+ id: string;
71
+ };
72
+ output: {
73
+ ok: true;
74
+ };
75
+ meta: object;
76
+ }>;
77
+ cancel: import("@trpc/server").TRPCMutationProcedure<{
78
+ input: {
79
+ id: string;
80
+ };
81
+ output: {
82
+ job_id: string;
83
+ status: string;
84
+ workflow_id: string;
85
+ created_at: string | null;
86
+ is_running: boolean;
87
+ is_completed: boolean;
88
+ };
89
+ meta: object;
90
+ }>;
91
+ }>>;
92
+ //# sourceMappingURL=jobs.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"jobs.d.ts","sourceRoot":"","sources":["../../../src/trpc/routers/jobs.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAkDH,eAAO,MAAM,UAAU;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8DrB,CAAC"}