@zodic/shared 0.0.293 → 0.0.295

package/app/services/ArchetypeService.ts

@@ -1,5 +1,8 @@
+import { eq } from 'drizzle-orm';
 import { inject, injectable } from 'inversify';
-import { KVArchetype } from '../../types/scopes/legacy';
+import { ChatMessages, schema } from '../..';
+import { KVArchetype, ZodiacSignSlug } from '../../types/scopes/legacy';
+import { generateArchetypePrompt } from '../../utils/archetypeNamesMessages';
 import { buildCosmicMirrorArchetypeKVKey } from '../../utils/KVKeysBuilders';
 import { AppContext } from '../base';
 
@@ -281,4 +284,240 @@ export class ArchetypeService {
       })
     );
   }
+
+  async generateArchetypeNames(
+    combination: string,
+    gender: 'male' | 'female' | 'non-binary',
+    overrideExisting: boolean = false
+  ) {
+    const [sun, ascendant, moon] = combination.split('-') as ZodiacSignSlug[];
+
+    const prompt = generateArchetypePrompt([{ sun, ascendant, moon }]);
+    const messages: ChatMessages = [
+      {
+        role: 'user',
+        content: prompt,
+      },
+    ];
+    const response = await this.context.api().callTogether.single(messages, {});
+
+    if (!response) {
+      throw new Error(
+        `No response when generating archetype names for ${combination}`
+      );
+    }
+
+    const blocks = response
+      .split('-EN')[1]
+      .split('-PT')[0]
+      .trim()
+      .split(/\n\d\.\s*\n?/)
+      .filter(Boolean);
+    const ptBlocks = response
+      .split('-PT')[1]
+      .trim()
+      .split(/\n\d\.\s*\n?/)
+      .filter(Boolean);
+
+    const englishNames = blocks.map((block) => {
+      const nameMatch = block.match(/• Name:\s*(.+)/);
+      const essenceMatch = block.match(/• Essence:\s*(.+)/);
+      return {
+        name: nameMatch?.[1]?.trim() || '',
+        essenceLine: essenceMatch?.[1]?.trim() || '',
+      };
+    });
+
+    const portugueseVariants = ptBlocks.map((block) => {
+      const masc = block.match(/• Masculino:\s*(.+)/)?.[1]?.trim() || '';
+      const fem = block.match(/• Feminino:\s*(.+)/)?.[1]?.trim() || '';
+      const essence = block.match(/• Essência:\s*(.+)/)?.[1]?.trim() || '';
+      return { masc, fem, essenceLine: essence };
+    });
+
+    const db = this.context.drizzle();
+
+    await Promise.all(
+      englishNames.map(async (entry, i) => {
+        const index = (i + 1).toString();
+        const existingEntry = await db
+          .select({ id: schema.archetypesData.id })
+          .from(schema.archetypesData)
+          .where(
+            eq(schema.archetypesData.id, `${combination}:${gender}:${index}`)
+          )
+          .execute();
+
+        if (overrideExisting || existingEntry.length === 0) {
+          await db
+            .insert(schema.archetypesData)
+            .values({
+              id: `${combination}:${gender}:${index}`,
+              combination,
+              gender,
+              archetypeIndex: index,
+              language: 'en-us',
+              name: entry.name,
+              essenceLine: entry.essenceLine,
+              status: 'idle',
+            })
+            .onConflictDoUpdate({
+              target: [schema.archetypesData.id],
+              set: {
+                name: entry.name,
+                essenceLine: entry.essenceLine,
+                updatedAt: new Date().getTime(),
+              },
+            });
+        }
+
+        const existingEntryPt = await db
+          .select({ id: schema.archetypesData.id })
+          .from(schema.archetypesData)
+          .where(
+            eq(schema.archetypesData.id, `${combination}:${gender}:${index}:pt`)
+          )
+          .execute();
+
+        if (overrideExisting || existingEntryPt.length === 0) {
+          await db
+            .insert(schema.archetypesData)
+            .values({
+              id: `${combination}:${gender}:${index}:pt`,
+              combination,
+              gender,
+              archetypeIndex: index,
+              language: 'pt-br',
+              name: portugueseVariants[i].masc,
+              essenceLine: portugueseVariants[i].essenceLine,
+              status: 'idle',
+            })
+            .onConflictDoUpdate({
+              target: [schema.archetypesData.id],
+              set: {
+                name: portugueseVariants[i].masc,
+                essenceLine: portugueseVariants[i].essenceLine,
+                updatedAt: new Date().getTime(),
+              },
+            });
+        }
+      })
+    );
+  }
+
+  async generateArchetypeNamesBatch(
+    entries: Array<{
+      combination: string;
+      gender: 'male' | 'female' | 'non-binary';
+    }>,
+    overrideExisting: boolean = false
+  ) {
+    const prompts = entries.map(({ combination }) => {
+      const [sun, ascendant, moon] = combination.split('-') as ZodiacSignSlug[];
+      return { sun, ascendant, moon };
+    });
+
+    const prompt = generateArchetypePrompt(prompts);
+    const messages: ChatMessages = [{ role: 'user', content: prompt }];
+    const response = await this.context.api().callTogether.single(messages, {});
+
+    if (!response)
+      throw new Error('No response when generating batch archetype names');
+
+    const db = this.context.drizzle();
+
+    const blocks = response
+      .split(/Composition \d+/)
+      .slice(1)
+      .map((b) => b.trim());
+
+    for (let i = 0; i < entries.length; i++) {
+      const { combination, gender } = entries[i];
+      const block = blocks[i];
+      const en = block.split('-EN')[1].split('-PT')[0].trim();
+      const pt = block.split('-PT')[1].trim();
+
+      const english = en
+        .split(/\n\d\.\s*\n?/)
+        .filter(Boolean)
+        .map((line) => ({
+          name: line.match(/• Name:\s*(.+)/)?.[1]?.trim() || '',
+          essenceLine: line.match(/• Essence:\s*(.+)/)?.[1]?.trim() || '',
+        }));
+
+      const portuguese = pt
+        .split(/\n\d\.\s*\n?/)
+        .filter(Boolean)
+        .map((line) => ({
+          masc: line.match(/• Masculino:\s*(.+)/)?.[1]?.trim() || '',
+          essenceLine: line.match(/• Essência:\s*(.+)/)?.[1]?.trim() || '',
+        }));
+
+      for (let j = 0; j < 3; j++) {
+        const index = (j + 1).toString();
+
+        const existingEntry = await db
+          .select({ id: schema.archetypesData.id })
+          .from(schema.archetypesData)
+          .where(
+            eq(schema.archetypesData.id, `${combination}:${gender}:${index}`)
+          )
+          .execute();
+
+        if (overrideExisting || existingEntry.length === 0) {
+          await db
+            .insert(schema.archetypesData)
+            .values({
+              id: `${combination}:${gender}:${index}`,
+              combination,
+              gender,
+              archetypeIndex: index,
+              language: 'en-us',
+              name: english[j].name,
+              essenceLine: english[j].essenceLine,
+              status: 'idle',
+            })
+            .onConflictDoUpdate({
+              target: [schema.archetypesData.id],
+              set: {
+                name: english[j].name,
+                essenceLine: english[j].essenceLine,
+                updatedAt: new Date().getTime(),
+              },
+            });
+        }
+
+        const existingEntryPt = await db
+          .select({ id: schema.archetypesData.id })
+          .from(schema.archetypesData)
+          .where(
+            eq(schema.archetypesData.id, `${combination}:${gender}:${index}:pt`)
+          )
+          .execute();
+
+        if (overrideExisting || existingEntryPt.length === 0) {
+          await db
+            .insert(schema.archetypesData)
+            .values({
+              id: `${combination}:${gender}:${index}:pt`,
+              combination,
+              gender,
+              archetypeIndex: index,
+              language: 'pt-br',
+              name: portuguese[j].masc,
+              essenceLine: portuguese[j].essenceLine,
+              status: 'idle',
+            })
+            .onConflictDoUpdate({
+              target: [schema.archetypesData.id],
+              set: {
+                name: portuguese[j].masc,
+                essenceLine: portuguese[j].essenceLine,
+                updatedAt: new Date().getTime(),
+              },
+            });
+        }
+      }
+    }
+  }
 }

package/app/workflow/ArchetypeWorkflow.ts

@@ -87,6 +87,33 @@ export class ArchetypeWorkflow {
     };
   }
 
+  async generateNames({
+    combinations,
+    gender,
+    overrideExisting,
+  }: {
+    combinations: string[];
+    gender: Gender;
+    overrideExisting?: boolean;
+  }): Promise<void> {
+    if (combinations.length === 0) return;
+
+    if (combinations.length === 1) {
+      await this.archetypeService.generateArchetypeNames(
+        combinations[0],
+        gender,
+        overrideExisting
+      );
+    } else {
+      const entries = combinations.map((combination) => ({
+        combination,
+        gender,
+      }));
+
+      await this.archetypeService.generateArchetypeNamesBatch(entries, overrideExisting);
+    }
+  }
+
   /**
    * Fetch all three archetypes from KV for a given combination and gender.
    */

package/db/migrations/0012_sudden_doctor_spectrum.sql

@@ -0,0 +1,27 @@
+PRAGMA foreign_keys=OFF;--> statement-breakpoint
+CREATE TABLE `__new_archetypes_data` (
+	`id` text PRIMARY KEY NOT NULL,
+	`language` text NOT NULL,
+	`gender` text,
+	`combination` text NOT NULL,
+	`archetype_index` text NOT NULL,
+	`name` text NOT NULL,
+	`essence_line` text NOT NULL,
+	`description` text,
+	`content` text DEFAULT '[]' NOT NULL,
+	`virtues` text DEFAULT '[]' NOT NULL,
+	`leonardo_prompt` text,
+	`status` text DEFAULT 'idle',
+	`created_at` integer DEFAULT CURRENT_TIMESTAMP,
+	`images` text DEFAULT '[]' NOT NULL,
+	`updated_at` integer DEFAULT CURRENT_TIMESTAMP
+);
+--> statement-breakpoint
+INSERT INTO `__new_archetypes_data`("id", "language", "gender", "combination", "archetype_index", "name", "essence_line", "description", "content", "virtues", "leonardo_prompt", "status", "created_at", "images", "updated_at") SELECT "id", "language", "gender", "combination", "archetype_index", "name", "essence_line", "description", "content", "virtues", "leonardo_prompt", "status", "created_at", "images", "updated_at" FROM `archetypes_data`;--> statement-breakpoint
+DROP TABLE `archetypes_data`;--> statement-breakpoint
+ALTER TABLE `__new_archetypes_data` RENAME TO `archetypes_data`;--> statement-breakpoint
+PRAGMA foreign_keys=ON;--> statement-breakpoint
+CREATE INDEX `archetypes_data_language_idx` ON `archetypes_data` (`language`);--> statement-breakpoint
+CREATE INDEX `archetypes_data_combination_idx` ON `archetypes_data` (`combination`);--> statement-breakpoint
+CREATE INDEX `archetypes_data_archetype_index_idx` ON `archetypes_data` (`archetype_index`);--> statement-breakpoint
+CREATE INDEX `archetypes_data_gender_idx` ON `archetypes_data` (`gender`);