Tests

Closes #5

.github/workflows/ci.yml

@@ -0,0 +1,51 @@
+name: Tests
+
+on:
+  push:
+    branches:
+      - main
+  pull_request:
+
+jobs:
+  setup-node-modules:
+    runs-on: ubuntu-latest
+    timeout-minutes: 15
+    steps:
+      - name: Git Checkout
+        uses: actions/checkout@b4ffde65f46336ab88eb53be808477a3936bae11
+
+      - name: Cache Dependencies
+        uses: actions/cache@0c45773b623bea8c8e75f6c82b208c3cf94ea4f9
+        with:
+          path: |
+            ~/.npm
+            node_modules/.cache
+          key: ${{ runner.os }}-npm-${{ hashFiles('**/workflows/test.yml') }}
+          restore-keys: ${{ runner.os }}-npm-
+
+      - name: Install Dependencies
+        run: npm install
+
+  lint:
+    name: Linting
+    runs-on: ubuntu-latest
+    needs: setup-node-modules
+    steps:
+      - name: Run Linting
+        run: npm run lint
+
+  unit-tests:
+    name: Unit Tests
+    runs-on: ubuntu-latest
+    needs: setup-node-modules
+    steps:      
+      - name: Run Tests
+        run: npm run test:unit
+
+  e2e-tests:
+    name: E2E Tests
+    runs-on: ubuntu-latest
+    needs: setup-node-modules
+    steps:
+      - name: Run Tests
+        run: npm run test:e2e

ai-providers/open-ai.ts

@@ -2,6 +2,7 @@ import { ReadableStream, UnderlyingByteSource, ReadableByteStreamController } fr
 import OpenAI from 'openai'
 import { AiProvider, NoContentError, StreamChunkCallback } from './provider'
 import { ReadableStream as ReadableStreamPolyfill } from 'web-streams-polyfill'
+import { fetch } from 'undici'
 import { ChatCompletionChunk } from 'openai/resources/index.mjs'
 import { AiStreamEvent, encodeEvent } from './event'
 import createError from '@fastify/error'
@@ -86,7 +87,8 @@ export class OpenAiProvider implements AiProvider {
 
   constructor (model: string, apiKey: string) {
     this.model = model
-    this.client = new OpenAI({ apiKey })
+    // @ts-expect-error
+    this.client = new OpenAI({ apiKey, fetch })
   }
 
   async ask (prompt: string): Promise<string> {

index.ts

@@ -10,6 +10,11 @@ import authPlugin from './plugins/auth'
 import apiPlugin from './plugins/api'
 import createError from '@fastify/error'
 
+let dontRegisterPlatformaticService = false
+export function setDontRegisterPlatformaticService (value: boolean): void {
+  dontRegisterPlatformaticService = value
+}
+
 const stackable: Stackable<AiWarpConfig> = async function (fastify, opts) {
   const { config } = fastify.platformatic
   await fastify.register(fastifyUser, config.auth)
@@ -71,7 +76,9 @@ const stackable: Stackable<AiWarpConfig> = async function (fastify, opts) {
   })
   await fastify.register(apiPlugin, opts)
 
-  await fastify.register(platformaticService, opts)
+  if (!dontRegisterPlatformaticService) {
+    await fastify.register(platformaticService, opts)
+  }
 }
 
 stackable.configType = 'ai-warp-app'

lib/generator.ts

@@ -18,7 +18,11 @@ class AiWarpGenerator extends ServiceGenerator {
     const defaultBaseConfig = super.getDefaultConfig()
     const defaultConfig = {
       aiProvider: 'openai',
-      aiModel: 'gpt-3.5-turbo'
+      aiModel: 'gpt-3.5-turbo',
+      // TODO: temporary fix, when running the typescript files directly
+      //  (in tests) this goes a directory above the actual project. Exposing
+      //  temporarily until I come up with something better
+      aiWarpPackageJsonPath: join(__dirname, '..', '..', 'package.json')
     }
     return Object.assign({}, defaultBaseConfig, defaultConfig)
   }
@@ -132,7 +136,8 @@ class AiWarpGenerator extends ServiceGenerator {
 
   async getStackablePackageJson (): Promise<PackageJson> {
     if (this._packageJson == null) {
-      const packageJsonPath = join(__dirname, '..', '..', 'package.json')
+      // const packageJsonPath = join(__dirname, '..', '..', 'package.json')
+      const packageJsonPath = this.config.aiWarpPackageJsonPath
       const packageJsonFile = await readFile(packageJsonPath, 'utf8')
       const packageJson: Partial<PackageJson> = JSON.parse(packageJsonFile)