From 1f4a69c051d15022dc870f34620f2a6a3db18f1c Mon Sep 17 00:00:00 2001 From: Concedo <39025047+LostRuins@users.noreply.github.com> Date: Mon, 17 Apr 2023 19:31:15 +0800 Subject: [PATCH] version number api --- koboldcpp.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/koboldcpp.py b/koboldcpp.py index 2bf1e6bbc..8fb7fdb1a 100644 --- a/koboldcpp.py +++ b/koboldcpp.py @@ -119,6 +119,7 @@ maxctx = 2048 maxlen = 128 modelbusy = False defaultport = 5001 +KcppVersion = "1.10" class ServerRequestHandler(http.server.SimpleHTTPRequestHandler): sys_version = "" @@ -133,7 +134,7 @@ class ServerRequestHandler(http.server.SimpleHTTPRequestHandler): super().__init__(*args, **kwargs) def do_GET(self): - global maxctx, maxlen, friendlymodelname + global maxctx, maxlen, friendlymodelname, KcppVersion if self.path in ["/", "/?"] or self.path.startswith(('/?','?')): #it's possible for the root url to have ?params without / response_body = "" if self.embedded_kailite is None: @@ -185,6 +186,12 @@ class ServerRequestHandler(http.server.SimpleHTTPRequestHandler): self.wfile.write(json.dumps({"result":"1.2.2"}).encode()) return + if self.path.endswith(('/api/extra/version')): + self.send_response(200) + self.end_headers() + self.wfile.write(json.dumps({"result":"KoboldCpp","version":KcppVersion}).encode()) + return + self.send_response(404) self.end_headers() rp = 'Error: HTTP Server is running, but this endpoint does not exist. Please check the URL.' @@ -435,7 +442,7 @@ def main(args): RunServerMultiThreaded(args.host, args.port, embedded_kailite) if __name__ == '__main__': - print("Welcome to KoboldCpp - Version 1.10") # just update version manually + print("Welcome to KoboldCpp - Version " + KcppVersion) # just update version manually parser = argparse.ArgumentParser(description='Kobold llama.cpp server') modelgroup = parser.add_mutually_exclusive_group() #we want to be backwards compatible with the unnamed positional args modelgroup.add_argument("--model", help="Model file to load", nargs="?")