Added a test for the runGPTAnalysis function. Needs MAJOR prompr revision

Author: ConnerWithAnE

src/gpt-controller.ts

@@ -20,7 +20,7 @@ export class GPTController {
     this.model = model;
   }
 
-  async runGPTAnalysis(files: Express.Multer.File[]): Promise<GPTResponse[]> {
+  async runGPTAnalysis(filePaths: string[]): Promise<GPTResponse[]> {
     const assistantParams: AssistantBody = {
       name: "Radiation Effects Researcher",
       instructions:
@@ -34,14 +34,15 @@ export class GPTController {
     const results: GPTResponse[] = [];
 
     // Upload files and create threads concurrently
-    const fileThreads = files.map(async (file: Express.Multer.File) => {
+    const fileThreads = filePaths.map(async (filePath: string) => {
       // Pretty sure we need an assistant for each thread to keep it separated.
-      const fileID = await this.uploadFile(file.path);
+      const fileID = await this.uploadFile(filePath);
       const threadMessage: ThreadMessage = {
         role: "assistant",
         content: prompt + questions,
         attachments: [{ file_id: fileID, tools: [{ type: "file_search" }] }],
       };
+      console.log(`Thread Message: ${threadMessage}`)
       // Create the three threads for each paper
       let threadResults: GPTData[] = []
       for (let i = 0; i < 3; i++) {
@@ -74,13 +75,14 @@ export class GPTController {
           console.log(run.status);
         }
       } 
+      /*
       const threadFinal: GPTResponse = {
         pass_1: threadResults[0],
         pass_2: threadResults[1],
         pass_3: threadResults[2]
       }
       results.push(threadFinal);
-
+      
       // TODO: Need to add the stream and and return it, not working yet.
       // Will be uncommented to implement
 

src/index.ts

@@ -3,6 +3,7 @@ import sqlite3 from "sqlite3";
 import cors from "cors";
 import bodyParser from "body-parser";
 import { open, Database } from "sqlite";
+import dotenv from "dotenv"
 
 // Import routers
 
@@ -16,7 +17,7 @@ import { GPTModel } from "./enums";
 
 const app = express();
 const PORT = process.env.PORT || 3000; // Use environment variable if available, otherwise default to 3000
-
+dotenv.config();
 /* In the future this will be used to ensure that only requests from certain domains are accepted
 const corsOptions = {
   origin: (origin: string | undefined, callback: (err: Error | null, allowed: boolean) => void) => {

src/prompts.data.ts

@@ -1,5 +1,5 @@
 export const questions = [
-  "What is the first author's name, in the format (J. Doe)",
+  "What are all of the author's names, in the format (J. Doe) in a list like this ['J. Doe', 'R. Austin']",
   "What is the Part No. or name if that is not available",
   'What is the type of part (eg, switching regulator), if there are multiple part numbers listed, list them all and seperate them with a "¶"',
   "Who is the manufacturer",

src/routes/admin-router.ts

@@ -50,7 +50,8 @@ async function insertRows(
 }
 
 async function parsePapers(files: any, gptController: GPTController): Promise<GPTResponse[]> {
-  gptController.runGPTAnalysis(files)
+  const fileList: string[] = files.forEach((file: any) => file.path)
+  gptController.runGPTAnalysis(fileList)
   const temp: GPTResponse[] = [];
   return temp;
 }

test/gpt-controller.test.ts

@@ -0,0 +1,32 @@
+import { GetQuery, InsertData, RadData } from '../src/types';
+
+import { GPTController } from '../src/gpt-controller';
+import { GPTModel } from '../src/enums';
+import fs from 'fs';
+import dotenv from 'dotenv';
+import path, { resolve } from 'path';
+
+async function initializeGPT(): Promise<GPTController> {
+  return new GPTController(GPTModel.GPT3_5Turbo);
+}
+
+let testGPT: GPTController;
+
+// Initialize the database once before all tests
+beforeAll(async () => {
+  dotenv.config();
+  testGPT = await initializeGPT();
+});
+
+/**********
+ *  TODO
+ * Finish unit testing (using jest)
+ ***********/
+
+test('Insertion of a paper with one author', async () => {
+  const paper: string = path.resolve(__dirname, 
+    './testfiles/SEE_in-flight_data_for_two_static_32KB_memories_on_high_earth_orbit.pdf');
+  fs.readFileSync(paper);
+
+  await expect(testGPT.runGPTAnalysis([paper])).resolves.not.toThrow();
+}, 60000);