wip

packages/ai/src/utils/utils.ts

@@ -0,0 +1,93 @@
+import handleStreamData from './handleStreamData.ts';
+
+export class Utils {
+  static getSeason(date: string): string {
+    const hemispheres = {
+      Northern: ['Winter', 'Spring', 'Summer', 'Autumn'],
+      Southern: ['Summer', 'Autumn', 'Winter', 'Spring'],
+    };
+    const d = new Date(date);
+    const month = d.getMonth();
+    const day = d.getDate();
+    const hemisphere = 'Northern';
+
+    if (month < 2 || (month === 2 && day <= 20) || month === 11) return hemispheres[hemisphere][0];
+    if (month < 5 || (month === 5 && day <= 21)) return hemispheres[hemisphere][1];
+    if (month < 8 || (month === 8 && day <= 22)) return hemispheres[hemisphere][2];
+    return hemispheres[hemisphere][3];
+  }
+  static getTimezone(timezone) {
+    if (timezone) {
+      return timezone;
+    }
+    return Intl.DateTimeFormat().resolvedOptions().timeZone;
+  }
+
+  static getCurrentDate() {
+    return new Date().toISOString();
+  }
+
+  static isAssetUrl(url) {
+    const { pathname } = new URL(url);
+    return pathname.startsWith('/assets/');
+  }
+
+  static selectEquitably({ a, b, c, d }, itemCount = 9) {
+    const sources = [a, b, c, d];
+    const result = {};
+
+    let combinedItems: any[] = [];
+    sources.forEach((source, index) => {
+      combinedItems.push(...Object.keys(source).map(key => ({ source: index, key })));
+    });
+
+    combinedItems = combinedItems.sort(() => Math.random() - 0.5);
+
+    let selectedCount = 0;
+    while (selectedCount < itemCount && combinedItems.length > 0) {
+      const { source, key } = combinedItems.shift();
+      const sourceObject = sources[source];
+
+      if (!result[key]) {
+        result[key] = sourceObject[key];
+        selectedCount++;
+      }
+    }
+
+    return result;
+  }
+
+  static normalizeWithBlanks<T extends NormalizeChatMessage>(msgs: T[]): T[] {
+    const out: T[] = [];
+
+    // In local mode first turn expected to be user.
+    let expected: NormalizeRole = 'user';
+
+    for (const m of msgs) {
+      while (m.role !== expected) {
+        // Insert blanks to match expected sequence user/assistant/user...
+        out.push(makeNormalizeBlank(expected) as T);
+        expected = expected === 'user' ? 'assistant' : 'user';
+      }
+
+      out.push(m);
+      expected = expected === 'user' ? 'assistant' : 'user';
+    }
+
+    return out;
+  }
+
+  static handleStreamData = handleStreamData;
+}
+
+// Normalize module exports
+export type NormalizeRole = 'user' | 'assistant';
+
+export interface NormalizeChatMessage extends Record<any, any> {
+  role: NormalizeRole;
+}
+
+export const makeNormalizeBlank = (role: NormalizeRole): NormalizeChatMessage => ({
+  role,
+  content: '',
+});