@arcote.tech/arc-ai 0.4.6 → 0.4.9

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@arcote.tech/arc-ai",
   "type": "module",
-  "version": "0.4.6",
+  "version": "0.4.9",
   "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.4.6",
-    "@arcote.tech/arc-auth": "^0.4.6",
+    "@arcote.tech/arc": "^0.4.9",
+    "@arcote.tech/arc-auth": "^0.4.9",
     "typescript": "^5.0.0"
   },
   "devDependencies": {

package/src/aggregates/budget.ts

@@ -5,7 +5,7 @@ import {
   id,
   number,
   string,
-  type AggregateConstructorAny,
+  type ArcAggregateElement,
   type ArcId,
 } from "@arcote.tech/arc";
 import type { Token } from "@arcote.tech/arc-auth";
@@ -27,7 +27,7 @@ export type BudgetAggregateData = {
   budgetId: ArcId<any>;
   accountId: ArcId<any>;
   userToken: Token;
-  CompletionAggregate: AggregateConstructorAny;
+  CompletionAggregate: ArcAggregateElement;
 };
 
 export const createBudgetAggregate = <
@@ -133,7 +133,7 @@ export const createBudgetAggregate = <
     )
 
     .protectBy(userToken, (p) => ({ accountId: p.accountId }))
-    .build();
+;
 };
 
 export type BudgetAggregate = ReturnType<typeof createBudgetAggregate>;

package/src/aggregates/completion.ts

@@ -231,7 +231,7 @@ export const createCompletionAggregate = <
     )
 
     .protectBy(userToken, (p) => ({ accountId: p.accountId }))
-    .build();
+;
 };
 
 export type CompletionAggregate = ReturnType<typeof createCompletionAggregate>;

package/src/ai-builder.ts

@@ -1,7 +1,6 @@
 import {
-  aggregateContextElement,
   context,
-  type AggregateConstructorAny,
+  type ArcAggregateElement,
   type ArcContextElement,
 } from "@arcote.tech/arc";
 import type { AccountId, Token } from "@arcote.tech/arc-auth";
@@ -13,8 +12,8 @@ import type { LLMProvider, PricingConfig } from "./types";
 export class AIBuilder<
   CompId,
   BudId,
-  Completion extends AggregateConstructorAny,
-  Budget extends AggregateConstructorAny,
+  Completion extends ArcAggregateElement,
+  Budget extends ArcAggregateElement,
   Providers extends LLMProvider[],
   Elements extends ArcContextElement<any>[],
 > {
@@ -64,7 +63,7 @@ export function ai<
     providers: config.providers,
     pricing: config.pricing,
   });
-  const completionElement = aggregateContextElement(Completion);
+  const completionElement = Completion;
 
   const Budget = createBudgetAggregate({
     name: config.name,
@@ -73,7 +72,7 @@ export function ai<
     userToken: config.userToken,
     CompletionAggregate: Completion,
   });
-  const budgetElement = aggregateContextElement(Budget);
+  const budgetElement = Budget;
 
   const streamRoute = createStreamCompletionRoute({
     name: config.name,

package/src/routes/stream-completion.ts

@@ -1,11 +1,11 @@
-import { route, type ArcAggregateElement, type AggregateConstructorAny } from "@arcote.tech/arc";
+import { route, type ArcAggregateElement } from "@arcote.tech/arc";
 import type { Token } from "@arcote.tech/arc-auth";
 import type { LLMProvider, PricingConfig, StreamChunk } from "../types";
 import type { ArcToolAny } from "../tool/tool";
 
 export type StreamCompletionRouteData = {
   name: string;
-  completionElement: ArcAggregateElement<AggregateConstructorAny>;
+  completionElement: ArcAggregateElement;
   userToken: Token;
   providers: LLMProvider[];
   pricing?: Record<string, PricingConfig>;