add util

HarryGuo2999 · HarryGuo2999 · commit 62d87d7cea46 · 2024-09-10T16:36:35.000+08:00
diff --git a/llama_cpp/server/util.py b/llama_cpp/server/util.py
@@ -9,7 +9,10 @@ def version():
     return rust_lib.version()
 
 def device_info(key):
-    key_bytes = key.encode('utf-8')
+    if key:
+        key_bytes = key.encode('utf-8')
+    else:
+        key_bytes = b""
     return rust_lib.device_info(key_bytes)
 
 def get_device_info(key="123"):
