image edits route for openai

Author: narengogi

src/handlers/imageEditsHandler.ts

@@ -15,7 +15,7 @@ import { Context } from 'hono';
  */
 export async function imageEditsHandler(c: Context): Promise<Response> {
   try {
-    let request = await c.req.json();
+    let request = await c.req.raw.formData();
     let requestHeaders = Object.fromEntries(c.req.raw.headers);
     const camelCaseConfig = constructConfigFromRequestHeaders(requestHeaders);
 

src/providers/azure-openai/api.ts

@@ -96,6 +96,9 @@ const AzureOpenAIAPIConfig: ProviderAPIConfig = {
       case 'imageGenerate': {
         return `/deployments/${deploymentId}/images/generations?api-version=${apiVersion}`;
       }
+      case 'imageEdit': {
+        return `/deployments/${deploymentId}/images/edits?api-version=${apiVersion}`;
+      }
       case 'createSpeech': {
         return `/deployments/${deploymentId}/audio/speech?api-version=${apiVersion}`;
       }

src/providers/openai/api.ts

@@ -38,6 +38,8 @@ const OpenAIAPIConfig: ProviderAPIConfig = {
         return '/embeddings';
       case 'imageGenerate':
         return '/images/generations';
+      case 'imageEdit':
+        return '/images/edits';
       case 'createSpeech':
         return '/audio/speech';
       case 'createTranscription':

src/providers/openai/imageEdits.ts

@@ -0,0 +1,81 @@
+import { OPEN_AI } from '../../globals';
+import { ErrorResponse, ImageGenerateResponse, ProviderConfig } from '../types';
+import { OpenAIErrorResponseTransform } from './utils';
+
+export const OpenAIImageEditConfig: ProviderConfig = {
+  image: {
+    param: 'image',
+    required: true,
+  },
+  prompt: {
+    param: 'prompt',
+    required: true,
+  },
+  background: {
+    param: 'background',
+  },
+  input_fidelity: {
+    param: 'input_fidelity',
+  },
+  mask: {
+    param: 'mask',
+  },
+  model: {
+    param: 'model',
+    default: 'dall-e-2',
+  },
+  n: {
+    param: 'n',
+    min: 1,
+    max: 10,
+  },
+  output_compression: {
+    param: 'output_compression',
+    min: 0,
+    max: 100,
+  },
+  output_format: {
+    param: 'output_format',
+  },
+  partial_images: {
+    param: 'partial_images',
+    min: 0,
+    max: 3,
+  },
+  quality: {
+    param: 'quality',
+  },
+  response_format: {
+    param: 'response_format',
+  },
+  size: {
+    param: 'size',
+  },
+  stream: {
+    param: 'stream',
+  },
+  user: {
+    param: 'user',
+  },
+};
+
+interface OpenAIImageObject {
+  b64_json?: string; // The base64-encoded JSON of the generated image, if response_format is b64_json.
+  url?: string; // The URL of the generated image, if response_format is url (default).
+  revised_prompt?: string; // The prompt that was used to generate the image, if there was any revision to the prompt.
+}
+
+interface OpenAIImageGenerateResponse extends ImageGenerateResponse {
+  data: OpenAIImageObject[];
+}
+
+export const OpenAIImageEditResponseTransform: (
+  response: OpenAIImageGenerateResponse | ErrorResponse,
+  responseStatus: number
+) => ImageGenerateResponse | ErrorResponse = (response, responseStatus) => {
+  if (responseStatus !== 200 && 'error' in response) {
+    return OpenAIErrorResponseTransform(response, OPEN_AI);
+  }
+
+  return response;
+};

src/providers/openai/index.ts

@@ -46,13 +46,18 @@ import {
   OpenAIListInputItemsResponseTransformer,
 } from '../open-ai-base';
 import { OPEN_AI } from '../../globals';
+import {
+  OpenAIImageEditConfig,
+  OpenAIImageEditResponseTransform,
+} from './imageEdits';
 
 const OpenAIConfig: ProviderConfigs = {
   complete: OpenAICompleteConfig,
   embed: OpenAIEmbedConfig,
   api: OpenAIAPIConfig,
   chatComplete: OpenAIChatCompleteConfig,
   imageGenerate: OpenAIImageGenerateConfig,
+  imageEdit: OpenAIImageEditConfig,
   createSpeech: OpenAICreateSpeechConfig,
   createTranscription: {},
   createTranslation: {},
@@ -77,6 +82,7 @@ const OpenAIConfig: ProviderConfigs = {
     chatComplete: OpenAIChatCompleteResponseTransform,
     // 'stream-chatComplete': OpenAIChatCompleteResponseTransform,
     imageGenerate: OpenAIImageGenerateResponseTransform,
+    imageEdit: OpenAIImageEditResponseTransform,
     createSpeech: OpenAICreateSpeechResponseTransform,
     createTranscription: OpenAICreateTranscriptionResponseTransform,
     createTranslation: OpenAICreateTranslationResponseTransform,