Przeglądaj źródła

api_like_OAI.py : compat with Microsoft Guidance (#2746)

Check for None in addition to empty string check in all request params

Co-authored-by: Georgi Gerganov <ggerganov@gmail.com>
Ryder Wishart 2 lat temu
rodzic
commit
8e6716a102
1 zmienionych plików z 7 dodań i 1 usunięć
  1. 7 1
      examples/server/api_like_OAI.py

+ 7 - 1
examples/server/api_like_OAI.py

@@ -23,7 +23,13 @@ parser.add_argument("--port", type=int, help="Set the port to listen.(default: 8
 args = parser.parse_args()
 
 def is_present(json, key):
-    return key in json
+    try:
+        buf = json[key]
+    except KeyError:
+        return False
+    if json[key] == None:
+        return False
+    return True
 
 #convert chat to prompt
 def convert_chat(messages):