Add support for image endpoint

Author: tomukmatthews

src/providers/nscale/api.ts

@@ -9,6 +9,8 @@ const NscaleAPIConfig: ProviderAPIConfig = {
     switch (fn) {
       case 'chatComplete':
         return '/chat/completions';
+      case 'imageGenerate':
+        return '/images/generations';
       default:
         return '';
     }

src/providers/nscale/imageGenerate.ts

@@ -0,0 +1,28 @@
+import { ParameterConfig } from '../types';
+
+export const NscaleImageGenerateConfig: { [key: string]: ParameterConfig } = {
+  prompt: {
+    param: 'prompt',
+    required: true,
+  },
+  model: {
+    param: 'model',
+    required: true,
+  },
+  n: {
+    param: 'n',
+  },
+  size: {
+    param: 'size',
+  },
+};
+
+export const NscaleImageGenerateResponseTransform = (response: any) => {
+  return {
+    created: Date.now(),
+    data: response.data.map((item: any) => ({
+      url: item.url,
+      b64_json: item.b64_json,
+    })),
+  };
+};

src/providers/nscale/index.ts

@@ -5,13 +5,19 @@ import {
   NscaleChatCompleteResponseTransform,
   NscaleChatCompleteStreamChunkTransform,
 } from './chatComplete';
+import {
+  NscaleImageGenerateConfig,
+  NscaleImageGenerateResponseTransform,
+} from './imageGenerate';
 
 const NscaleConfig: ProviderConfigs = {
   chatComplete: NscaleChatCompleteConfig,
+  imageGenerate: NscaleImageGenerateConfig,
   api: NscaleAPIConfig,
   responseTransforms: {
     chatComplete: NscaleChatCompleteResponseTransform,
     'stream-chatComplete': NscaleChatCompleteStreamChunkTransform,
+    imageGenerate: NscaleImageGenerateResponseTransform,
   },
 };