add openaihosted testing

hippo306 · Apr 28, 2023 · f95ace3 · f95ace3
1 parent 8fe493a
commit f95ace3
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/testing/openaihosted_test.py b/testing/openaihosted_test.py
@@ -0,0 +1,14 @@
+import openaihosted
+
+messages = [{"role": "system", "content": "You are a helpful assistant."}]
+while True:
+    question = input("Question: ")
+    if question == "!stop":
+        break
+
+    messages.append({"role": "user", "content": question})
+    request = openaihosted.Completion.create(messages=messages)
+
+    response = request["responses"]
+    messages.append({"role": "assistant", "content": response})
+    print(f"Answer: {response}")