From d4a12133e773d0ae15be2fffdd0f58ca085b5394 Mon Sep 17 00:00:00 2001
From: Concedo <39025047+LostRuins@users.noreply.github.com>
Date: Mon, 4 Mar 2024 14:26:49 +0800
Subject: [PATCH] added SD samplers endpoint

---
 kcpp_docs.embd | 22 ++++++++++++++++++++--
 koboldcpp.py   |  6 ++++++
 2 files changed, 26 insertions(+), 2 deletions(-)

diff --git a/kcpp_docs.embd b/kcpp_docs.embd
index 365cf36f4..5b0d2d457 100644
--- a/kcpp_docs.embd
+++ b/kcpp_docs.embd
@@ -894,6 +894,25 @@
                       ]
                    }
                 },
+                "/sdapi/v1/samplers": {
+                   "get": {
+                      "description": "Gets a list of supported samplers.",
+                      "responses": {
+                         "200": {
+                            "content": {
+                               "application/json": {
+                                  "example": [{"name":"Euler a","aliases":["k_euler_a","k_euler_ancestral"],"options":{}},{"name":"Euler","aliases":["k_euler"],"options":{}},{"name":"Heun","aliases":["k_heun"],"options":{}},{"name":"DPM2","aliases":["k_dpm_2"],"options":{}},{"name":"DPM++ 2M","aliases":["k_dpmpp_2m"],"options":{}}]
+                               }
+                            },
+                            "description": "Successful request"
+                         }
+                      },
+                      "summary": "Gets a list of supported samplers",
+                      "tags": [
+                         "sdapi/v1"
+                      ]
+                   }
+                },
                 "/sdapi/v1/txt2img": {
                    "post": {
                       "description": "Generates an image from a text prompt, and returns a base64 encoded png.",
@@ -1030,8 +1049,7 @@
                 }
              ]
           };
-   </script>
-
+    </script>
 
    <script>
       //self destruct into json if requested
diff --git a/koboldcpp.py b/koboldcpp.py
index 9940dcd14..1b0f6c42e 100644
--- a/koboldcpp.py
+++ b/koboldcpp.py
@@ -978,6 +978,12 @@ Enter Prompt:<br>
                 response_body = (json.dumps([{"title":friendlysdmodelname,"model_name":friendlysdmodelname,"hash":"8888888888","sha256":"8888888888888888888888888888888888888888888888888888888888888888","filename":fullsdmodelpath,"config": None}]).encode())
         elif self.path.endswith('/sdapi/v1/options'):
            response_body = (json.dumps({"samples_format":"png","sd_model_checkpoint":friendlysdmodelname}).encode())
+        elif self.path.endswith('/sdapi/v1/samplers'):
+            if friendlysdmodelname=="inactive" or fullsdmodelpath=="":
+                response_body = (json.dumps([]).encode())
+            else:
+                response_body = (json.dumps([{"name":"Euler a","aliases":["k_euler_a","k_euler_ancestral"],"options":{}},{"name":"Euler","aliases":["k_euler"],"options":{}},{"name":"Heun","aliases":["k_heun"],"options":{}},{"name":"DPM2","aliases":["k_dpm_2"],"options":{}},{"name":"DPM++ 2M","aliases":["k_dpmpp_2m"],"options":{}}]).encode())
+
 
         elif self.path=="/api":
             content_type = 'text/html'