app service extends embeddingService

api/index.ts

@@ -3,6 +3,7 @@ import { DataBase } from "./database";
 import express, { Express } from "express";
 import "dotenv/config";
 import { getValue } from "./utils";
+import { CONSTANTS } from "./core/constants";
 const app: Express = express();
 
 export const database: DataBase = new DataBase();
@@ -31,7 +32,7 @@ const filePath: string = getValue("PDF_ABSOLUTE_PATH");
 const apiKey: string = getValue("API_KEY");
 
 const createEmbedding = async () => {
-  const appService = new AppService(apiKey, filePath);
+  const appService = new AppService(apiKey, filePath, CONSTANTS.generativeAIModel);
   return await appService.createEmbeddings();
 };
 

api/services/app.service.ts

@@ -1,26 +1,27 @@
-import { CONSTANTS } from "../core/constants";
 import { IDocumentService } from "../interfaces/document-service.interface";
-import { IEmbeddingService } from "../interfaces/embedding-service.interface";
 import { DocumentService } from "./document-service";
 import { EmbeddingService } from "./embed-service";
 import { database } from "..";
 
-export class AppService {
-  constructor(private readonly APIkey: string, private readonly documentPath: string) {}
+export class AppService extends EmbeddingService {
+  constructor(apikey: string, private readonly documentPath: string, AIModel: string) {
+    super(apikey, AIModel);
+  }
   async createEmbeddings(): Promise<void> {
     const documentService: IDocumentService = new DocumentService();
-    const embeddingService: IEmbeddingService = new EmbeddingService(this.APIkey, CONSTANTS.generativeAIModel);
     let text: string;
     if (this.documentPath.length) {
       text = await documentService.convertPDFToText(this.documentPath);
     }
     if (text) {
       console.log("...generating embeddings");
-      const embeddings = await embeddingService.generateEmbeddings(text);
+      const embeddings = await this.generateEmbeddings(text);
       if (embeddings) {
         console.log("...embeddings generated");
         return await database.createDocument(text, embeddings);
       }
     }
   }
+
+  async similaritySearch() {}
 }

api/services/embed-service.ts

@@ -3,7 +3,7 @@ import { IEmbeddingService } from "../interfaces/embedding-service.interface";
 
 export class EmbeddingService implements IEmbeddingService {
   genAI: GoogleGenerativeAI;
-  constructor(private readonly apiKey: string, private readonly AIModel: string) {
+  constructor(protected readonly apiKey: string, protected readonly AIModel: string) {
     this.genAI = new GoogleGenerativeAI(this.apiKey);
   }
   /**