@arcote.tech/arc-ai 0.7.7 → 0.7.8

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@arcote.tech/arc-ai",
   "type": "module",
-  "version": "0.7.7",
+  "version": "0.7.8",
   "private": false,
   "description": "AI provider abstraction, completion tracking, and budget management for Arc framework",
   "main": "./src/index.ts",
@@ -10,8 +10,8 @@
     "type-check": "tsc --noEmit"
   },
   "peerDependencies": {
-    "@arcote.tech/arc": "^0.7.7",
-    "@arcote.tech/arc-auth": "^0.7.7",
+    "@arcote.tech/arc": "^0.7.8",
+    "@arcote.tech/arc-auth": "^0.7.8",
     "typescript": "^5.0.0"
   },
   "devDependencies": {

package/src/aggregates/credit-ledger.ts

@@ -96,63 +96,73 @@ export function creditLedger(config: {
       },
     )
 
-    // ─── topUp — add credits ────────────────────────────────────
+    // ─── topUp — add credits (server-internal: listeners, cron, ai-orchestration)
     .mutateMethod(
       "topUp",
-      (fn) => fn.withParams({
-        scopeId,
-        amount: number(),
-        reason: string(),
-        metadata: string().optional(),
-      }).handle(
-        ONLY_SERVER &&
-        (async (ctx, params) => {
-          const entryId = ledgerId.generate();
-          await ctx.credited.emit({
-            ledgerId: entryId,
-            scopeId: params.scopeId,
-            amount: params.amount,
-            reason: params.reason,
-            metadata: params.metadata,
-          });
-          return { ok: true };
-        }),
-      ),
+      (fn) => fn
+        .private()
+        .withParams({
+          scopeId,
+          amount: number(),
+          reason: string(),
+          metadata: string().optional(),
+        })
+        .handle(
+          ONLY_SERVER &&
+          (async (ctx, params) => {
+            const entryId = ledgerId.generate();
+            await ctx.credited.emit({
+              ledgerId: entryId,
+              scopeId: params.scopeId,
+              amount: params.amount,
+              reason: params.reason,
+              metadata: params.metadata,
+            });
+            return { ok: true };
+          }),
+        ),
     )
 
-    // ─── deduct — remove credits ────────────────────────────────
+    // ─── deduct — remove credits (server-internal: ai consumers)
     .mutateMethod(
       "deduct",
-      (fn) => fn.withParams({
-        scopeId,
-        amount: number(),
-        reason: string(),
-        model: string().optional(),
-        metadata: string().optional(),
-      }).handle(
-        ONLY_SERVER &&
-        (async (ctx, params) => {
-          const entryId = ledgerId.generate();
-          await ctx.debited.emit({
-            ledgerId: entryId,
-            scopeId: params.scopeId,
-            amount: params.amount,
-            reason: params.reason,
-            model: params.model,
-            metadata: params.metadata,
-          });
-          return { ok: true };
-        }),
-      ),
+      (fn) => fn
+        .private()
+        .withParams({
+          scopeId,
+          amount: number(),
+          reason: string(),
+          model: string().optional(),
+          metadata: string().optional(),
+        })
+        .handle(
+          ONLY_SERVER &&
+          (async (ctx, params) => {
+            const entryId = ledgerId.generate();
+            await ctx.debited.emit({
+              ledgerId: entryId,
+              scopeId: params.scopeId,
+              amount: params.amount,
+              reason: params.reason,
+              model: params.model,
+              metadata: params.metadata,
+            });
+            return { ok: true };
+          }),
+        ),
     )
 
-    // ─── checkBalance — query current balance ───────────────────
+    // ─── checkBalance — query current scope's balance ───────────
     .clientQuery(
       "checkBalance",
       (fn) => fn
-        .withParams({ scopeId: string() })
-        .handle(async (ctx, params) => {
-          const entry = await ctx.$query.findOne({ scopeId: params.scopeId });
+        .protectedBy(protectBy, (p: any) => ({
+          scopeId: p.workspaceId ?? p.accountId,
+        }))
+        .handle(async (ctx) => {
+          // Scope filter (z protectedBy) jest aplikowany automatycznie do
+          // ctx.$query — findOne({}) zwraca tylko wiersz aktualnego scope'u.
+          const entry = await ctx.$query.findOne({});
           const balance = entry?.balance ?? 0;
           return {
             balance,
@@ -161,9 +171,7 @@ export function creditLedger(config: {
             totalDebited: entry?.totalDebited ?? 0,
           };
         }),
-    )
-
-    .protectBy(protectBy, (p: any) => ({ scopeId: p.workspaceId ?? p.accountId }));
+    );
 
   const elements = [Ledger];
 

package/src/tool/tool.ts

@@ -122,14 +122,14 @@ export class ArcTool<
       Data["handler"] extends Function
         ? ServerToolViewProps<
             Data["params"] extends ArcObjectAny ? $type<Data["params"]> : {},
-            Data["result"] extends ArcObjectAny
-              ? $type<Data["result"]>
+            Data extends { result: infer R extends ArcObjectAny }
+              ? $type<R>
               : unknown
           >
         : InteractiveToolViewProps<
             Data["params"] extends ArcObjectAny ? $type<Data["params"]> : {},
-            Data["result"] extends ArcObjectAny
-              ? $type<Data["result"]>
+            Data extends { result: infer R extends ArcObjectAny }
+              ? $type<R>
               : unknown
           >
     >,