From fc3c105d4281bbc037f45bd36ab8335963e8909e Mon Sep 17 00:00:00 2001
From: Dave Lee <dave@gray101.com>
Date: Thu, 25 May 2023 02:47:13 -0400
Subject: [PATCH] fix generation!

---
 openai-openapi/localai_model_patches.yaml | 45 +++++++++++++++--------
 1 file changed, 29 insertions(+), 16 deletions(-)

diff --git a/openai-openapi/localai_model_patches.yaml b/openai-openapi/localai_model_patches.yaml
index 2045b1f7..106ecaf8 100644
--- a/openai-openapi/localai_model_patches.yaml
+++ b/openai-openapi/localai_model_patches.yaml
@@ -8,16 +8,21 @@
 ---
 components:
   schemas:
-    CreateChatCompletionRequest:
+    #@overlay/match missing_ok=True
+    LocalAIBaseRequestExtension:
+      type: object
+      nullable: true
       properties:
-        #@overlay/match missing_ok=True
-        x-localai-extensions:
-          type: object
+        seed:
+          type: integer
+          nullable: true
+    #@overlay/match missing_ok=True
+    LocalAITextRequestExtension:
+      allOf:
+        - $ref: "#/components/schemas/LocalAIBaseRequestExtension"
+        - type: object
           nullable: true
           properties:
-            seed:
-              type: integer
-              nullable: true
             top_k:
               type: integer
               nullable: true
@@ -46,19 +51,27 @@ components:
               type: number
               nullable: true
               x-go-type: float64
-    CreateImageRequest:
-      properties:
-        #@overlay/match missing_ok=True
-        x-localai-extensions:
-          type: object
+    #@overlay/match missing_ok=True
+    LocalAIImageRequestExtension:
+      allOf:
+        - $ref: "#/components/schemas/LocalAIBaseRequestExtension"
+        - type: object
           nullable: true
           properties:
-            seed:
-              type: integer
-              nullable: true
             mode:
               type: integer
               nullable: true
             step:
               type: integer
-              nullable: true
\ No newline at end of file
+              nullable: true
+    CreateChatCompletionRequest:
+      properties:
+        #@overlay/match missing_ok=True
+        x-localai-extensions:
+           $ref: "#/components/schemas/LocalAITextRequestExtension"
+    CreateImageRequest:
+      properties:
+        #@overlay/match missing_ok=True
+        x-localai-extensions:
+           $ref: "#/components/schemas/LocalAIImageRequestExtension"
+            
\ No newline at end of file