@zodic/shared 0.0.70 → 0.0.72

package/app/api/index.ts

@@ -1,9 +1,13 @@
+import OpenAI from 'openai';
 import {
   BackendBindings,
   BatchInputItem,
   ChatGPTBatchResponse,
   ChatGPTFileUploadResponse,
   ChatMessages,
+  DeepSeekBatchInputItem,
+  DeepSeekBatchResponse,
+  DeepSeekOptions,
   NatalChartInterpretation,
 } from '../../types';
 import {
@@ -20,7 +24,70 @@ import {
 } from '../../types/scopes/legacy';
 import { makeAstrologyApiCall } from './astrology';
 
+const deepseek = (env: BackendBindings) =>
+  new OpenAI({
+    apiKey: env.DEEPSEEK_API_KEY,
+    baseURL: 'https://api.deepseek.com/v1', 
+  });
+
 export const Api = (env: BackendBindings) => ({
+  callDeepSeek: {
+    single: async (
+      messages: ChatMessages,
+      { model = 'deepseek-v3', options = {} }: DeepSeekOptions
+    ): Promise<string> => {
+      try {
+        const response = await deepseek(env).chat.completions.create({
+          model,
+          messages,
+          ...options,
+        });
+
+        if (response.choices[0].message.content) {
+          return response.choices[0].message.content.trim();
+        } else {
+          throw new Error('Content is null');
+        }
+      } catch (err: any) {
+        console.error('Error calling DeepSeek API:', err.message);
+        throw err;
+      }
+    },
+
+    batch: async (
+      batchItems: DeepSeekBatchInputItem[]
+    ): Promise<DeepSeekBatchResponse> => {
+      console.log('Preparing to send DeepSeek batch request.');
+
+      try {
+        const responses: DeepSeekBatchResponse = [];
+
+        for (const item of batchItems) {
+          const { messages, options } = item;
+          const response = await deepseek(env).chat.completions.create({
+            model: options?.model || 'deepseek-v3',
+            messages,
+            ...options,
+          });
+
+          if (response.choices[0].message.content) {
+            responses.push({
+              input: item,
+              output: response.choices[0].message.content.trim(),
+            });
+          } else {
+            console.error('Content is null');
+          }
+        }
+
+        console.log('Batch processing completed successfully.');
+        return responses;
+      } catch (error) {
+        console.error('Error in callDeepSeekBatch:', error);
+        throw error;
+      }
+    },
+  },
   callChatGPT: {
     single: async (
       messages: ChatMessages,

package/app/base/AppContext.ts

@@ -1,7 +1,7 @@
 import { drizzle } from 'drizzle-orm/d1';
 import * as schema from '../../db/schema';
 import { BackendBindings } from '../../types';
-import { buildChatGPTMessages } from '../../utils/buildMessages';
+import { buildLLMMessages } from '../../utils/buildMessages';
 import { Api } from '../api';
 
 export class AppContext {
@@ -9,8 +9,8 @@ export class AppContext {
 
   constructor(public env: BackendBindings) {}
 
-  buildChatGPTMessages() {
-    return buildChatGPTMessages(this.env);
+  buildLLMMessages() {
+    return buildLLMMessages(this.env);
   }
 
   kvCosmicMirrorArchetypesStore() {

package/app/services/ConceptService.ts

@@ -19,12 +19,10 @@ export class ConceptService {
       `Generating basic info for concept: ${conceptSlug}, combination: ${combinationString}`
     );
 
-    const messages = this.context
-      .buildChatGPTMessages()
-      .generateConceptBasicInfo({
-        combination: combinationString,
-        conceptSlug,
-      });
+    const messages = this.context.buildLLMMessages().generateConceptBasicInfo({
+      combination: combinationString,
+      conceptSlug,
+    });
 
     const response = await this.context.api().callChatGPT.single(messages, {});
     if (!response) {
@@ -86,7 +84,7 @@ export class ConceptService {
     }
 
     const messages = this.context
-      .buildChatGPTMessages()
+      .buildLLMMessages()
       .generateConceptLeonardoPrompt({
         conceptSlug,
         combination: combinationString,
@@ -125,15 +123,13 @@ export class ConceptService {
       );
     }
 
-    const messages = this.context
-      .buildChatGPTMessages()
-      .generateConceptContent({
-        conceptSlug,
-        combination: combinationString,
-        name: concept.name,
-        description: concept.description,
-        poem: concept.poem,
-      });
+    const messages = this.context.buildLLMMessages().generateConceptContent({
+      conceptSlug,
+      combination: combinationString,
+      name: concept.name,
+      description: concept.description,
+      poem: concept.poem,
+    });
 
     const response = await this.context.api().callChatGPT.single(messages, {});
     if (!response) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zodic/shared",
-  "version": "0.0.70",
+  "version": "0.0.72",
   "module": "index.ts",
   "type": "module",
   "publishConfig": {
@@ -31,6 +31,7 @@
     "hono": "^4.6.13",
     "inversify": "^6.2.1",
     "jose": "^5.9.6",
+    "openai": "^4.81.0",
     "reflect-metadata": "^0.2.2"
   }
 }

package/types/scopes/cloudflare.ts

@@ -54,6 +54,7 @@ export type CentralBindings = {
   ASTROLOGY_PDF_USER_ID: string;
   ASTROLOGY_PDF_API_KEY: string;
   PIAPI_API_KEY: string;
+  DEEPSEEK_API_KEY: string;
 
   API_BASE_URL: string;
   PUBLIC_GOOGLE_CLIENT_ID: string;
@@ -113,6 +114,7 @@ export type BackendBindings = Env &
     | 'PIAPI_API_KEY'
     | 'CONCEPT_GENERATION_QUEUE'
     | 'KV_CONCEPTS_MANAGEMENT'
+    | 'DEEPSEEK_API_KEY'
   >;
 
 export type BackendCtx = Context<

package/types/scopes/generic.ts

@@ -17,7 +17,6 @@ export const ProductType = {
   REPORT: 'report',
 };
 
-
 export const VALID_GENDERS_ARRAY = ['male', 'female', 'non-binary'] as const;
 
 export type Gender = (typeof VALID_GENDERS_ARRAY)[number];
@@ -241,3 +240,13 @@ export const ConceptPhases: Record<string, ConceptPhase> = {
   CONTENT: 'content',
   IMAGES: 'images',
 };
+
+export type DeepSeekOptions = { model?: string; options?: Record<string, any> };
+export type DeepSeekBatchInputItem = {
+  messages: ChatMessages;
+  options?: DeepSeekOptions;
+};
+export type DeepSeekBatchResponse = Array<{
+  input: DeepSeekBatchInputItem;
+  output: string;
+}>;

package/utils/buildMessages.ts

@@ -1,7 +1,7 @@
 import { BackendBindings, ChatMessages, Crown } from '../types';
 import { conceptPrompts } from './conceptPrompts';
 
-export const buildChatGPTMessages = (env: BackendBindings) => ({
+export const buildLLMMessages = (env: BackendBindings) => ({
   generateCosmicMirrorArchetypeBasicInfo: ({
     sun,
     ascendant,
@@ -106,7 +106,7 @@ export const buildChatGPTMessages = (env: BackendBindings) => ({
     description,
     poem,
     combination,
-    conceptSlug
+    conceptSlug,
   }: {
     name: string;
     description: string;