feat: rome is rising

Author: dosco

src/ax/ai/balance.ts

@@ -18,7 +18,6 @@ export class AxBalancer implements AxAIService {
   private services: AxAIService[];
   private currentServiceIndex: number = 0;
   private currentService: AxAIService;
-  private serviceMap: Record<string, AxAIService> = {};
 
   constructor(services: readonly AxAIService[]) {
     if (services.length === 0) {

src/ax/db/cloudflare.ts

@@ -62,7 +62,7 @@ export class AxDBCloudflare extends AxDBBase {
 
   override _upsert = async (
     req: Readonly<AxDBUpsertRequest>,
-    update?: boolean,
+    _update?: boolean,
     options?: Readonly<AxDBCloudflareOpOptions>
   ): Promise<AxDBUpsertResponse> => {
     const res = (await apiCall(

src/ax/db/weaviate.ts

@@ -196,7 +196,7 @@ export class AxDBWeaviate extends AxDBBase {
 
     const matches = resMatches.map((match) => {
       return {
-        id: match.id as string,
+        id: match['id'] as string,
         score: 1,
         metadata: match
       };

src/ax/docs/manager.ts

@@ -94,7 +94,9 @@ export class AxDBManager {
             values: embedding,
             metadata: { text: batch[index] ?? '' }
           }))
-          .filter((v) => v.metadata?.text && v.metadata?.text.length > 0);
+          .filter(
+            (v) => v.metadata?.['text'] && v.metadata?.['text'].length > 0
+          );
 
         // Batch upsert embeddings
         await this.db.batchUpsert(embeddings);
@@ -135,8 +137,11 @@ export class AxDBManager {
 
     for (const { matches } of queryResults) {
       const m = matches
-        .filter((v) => v.metadata?.text && v.metadata?.text.length > 0)
-        .map(({ score, metadata }) => ({ score, text: metadata?.text ?? '' }));
+        .filter((v) => v.metadata?.['text'] && v.metadata?.['text'].length > 0)
+        .map(({ score, metadata }) => ({
+          score,
+          text: metadata?.['text'] ?? ''
+        }));
 
       const tp = topPercent && topPercent > 1 ? topPercent / 100 : topPercent;
       const resultItems = tp ? getTopInPercent(m, tp) : m;

src/ax/dsp/generate.ts

@@ -354,9 +354,9 @@ export class AxGenerate<
     span?: AxSpan
   ): Promise<OUT> {
     const ai = options?.ai ?? this.ai;
-    const maxRetries = this.options?.maxRetries ?? options?.maxRetries ?? 5;
-    const maxSteps = this.options?.maxSteps ?? options?.maxSteps ?? 10;
-    const mem = this.options?.mem ?? options?.mem ?? new AxMemory();
+    const maxRetries = options?.maxRetries ?? this.options?.maxRetries ?? 5;
+    const maxSteps = options?.maxSteps ?? this.options?.maxSteps ?? 10;
+    const mem = options?.mem ?? this.options?.mem ?? new AxMemory();
     const canStream = ai.getFeatures().streaming;
 
     let err: ValidationError | AxAssertionError | undefined;
@@ -521,14 +521,14 @@ function parseFunctions(
     //     typeof f.args === 'object' ? JSON.stringify(f.args) : f.args;
     // }
     return funcs;
-  } else if (values.functionName) {
+  } else if (values['functionName']) {
     const { functionName, functionArguments } = values as {
       functionName: string;
       functionArguments: string;
       other: object;
     };
-    delete values.functionName;
-    delete values.functionArguments;
+    delete values['functionName'];
+    delete values['functionArguments'];
 
     return [
       {

src/ax/dsp/router.ts

@@ -28,7 +28,6 @@ export class AxRoute {
 
 export class AxRouter {
   private readonly ai: AxAIService;
-  private routes: AxRoute[] = [];
 
   private db: AxDBMemory;
   private debug?: boolean;

src/ax/funcs/code.ts

@@ -16,6 +16,16 @@ export enum AxJSInterpreterPermission {
   PROCESS = 'process'
 }
 
+type Context = {
+  console: Console;
+  fs: unknown;
+  http: unknown;
+  https: unknown;
+  os: unknown;
+  crypto: unknown;
+  process: unknown;
+};
+
 export class AxJSInterpreter {
   private permissions: readonly AxJSInterpreterPermission[];
 
@@ -28,7 +38,7 @@ export class AxJSInterpreter {
   }
 
   private codeInterpreterJavascript(code: string): unknown {
-    const context: { [key: string]: unknown } = { console };
+    const context: Partial<Context> = { console };
 
     if (this.permissions.includes(AxJSInterpreterPermission.FS)) {
       context.fs = _fs;

src/ax/mem/memory.ts

@@ -1,12 +1,10 @@
 import type { AxChatRequest, AxChatResponseResult } from '../ai/types.js';
 
-import type { AxAIMemory } from './types.js';
-
-type Writeable<T> = { -readonly [P in keyof T]: T[P] };
+import type { AxAIMemory, AxWritableChatPrompt } from './types.js';
 
 export class AxMemory implements AxAIMemory {
-  private data: AxChatRequest['chatPrompt'] = [];
-  private sdata = new Map<string, AxChatRequest['chatPrompt']>();
+  private data: AxWritableChatPrompt = [];
+  private sdata = new Map<string, AxWritableChatPrompt>();
   private limit: number;
 
   constructor(limit = 50) {
@@ -53,9 +51,7 @@ export class AxMemory implements AxAIMemory {
   ): void {
     const items = this.get(sessionId);
 
-    const lastItem = items.at(-1) as unknown as Writeable<
-      AxChatRequest['chatPrompt'][0]
-    >;
+    const lastItem = items.at(-1);
 
     if (!lastItem || lastItem.role !== 'assistant') {
       this.addResult({ content, name, functionCalls }, sessionId);
@@ -77,10 +73,6 @@ export class AxMemory implements AxAIMemory {
     return this.get(sessionId);
   }
 
-  peek(sessionId?: string): AxChatRequest['chatPrompt'] {
-    return this.get(sessionId);
-  }
-
   getLast(sessionId?: string): AxChatRequest['chatPrompt'][0] | undefined {
     const d = this.get(sessionId);
     return d.at(-1);
@@ -94,7 +86,7 @@ export class AxMemory implements AxAIMemory {
     }
   }
 
-  private get(sessionId?: string): AxChatRequest['chatPrompt'] {
+  private get(sessionId?: string): AxWritableChatPrompt {
     if (!sessionId) {
       return this.data;
     }

src/ax/mem/types.ts

@@ -1,5 +1,8 @@
 import type { AxChatRequest, AxChatResponseResult } from '../ai/index.js';
 
+type Writeable<T> = { -readonly [P in keyof T]: T[P] };
+export type AxWritableChatPrompt = Writeable<AxChatRequest['chatPrompt'][0]>[];
+
 export interface AxAIMemory {
   add(
     result: Readonly<
@@ -14,7 +17,6 @@ export interface AxAIMemory {
   ): void;
 
   history(sessionId?: string): AxChatRequest['chatPrompt'];
-  peek(sessionId?: string): AxChatRequest['chatPrompt'];
   reset(sessionId?: string): void;
 
   getLast(sessionId?: string): AxChatRequest['chatPrompt'][0] | undefined;

src/ax/util/apicall.ts

@@ -35,7 +35,7 @@ export const apiCall = async <TRequest = unknown, TResponse = unknown>(
   >,
   json: TRequest
 ): Promise<TResponse | ReadableStream<TResponse>> => {
-  const baseUrl = new URL(process.env.PROXY ?? api.url);
+  const baseUrl = new URL(process.env['PROXY'] ?? api.url);
   const apiPath = path.join(baseUrl.pathname, api.name ?? '/', baseUrl.search);
   const apiUrl = new URL(apiPath, baseUrl);