summaryrefslogtreecommitdiffstats
path: root/testing/aiservice/testing.py
diff options
context:
space:
mode:
Diffstat (limited to 'testing/aiservice/testing.py')
-rw-r--r--testing/aiservice/testing.py30
1 files changed, 30 insertions, 0 deletions
diff --git a/testing/aiservice/testing.py b/testing/aiservice/testing.py
new file mode 100644
index 00000000..5cb6c5ef
--- /dev/null
+++ b/testing/aiservice/testing.py
@@ -0,0 +1,30 @@
+from AiService import ChatCompletion
+
+# Test 1
+response = ChatCompletion.create(model="gpt-3.5-turbo",
+ provider="AiService",
+ stream=False,
+ messages=[{'role': 'user', 'content': 'who are you?'}])
+
+print(response)
+
+# Test 2
+response = ChatCompletion.create(model="gpt-3.5-turbo",
+ provider="AiService",
+ stream=False,
+ messages=[{'role': 'user', 'content': 'what you can do?'}])
+
+print(response)
+
+
+# Test 3
+response = ChatCompletion.create(model="gpt-3.5-turbo",
+ provider="AiService",
+ stream=False,
+ messages=[
+ {'role': 'user', 'content': 'now your name is Bob'},
+ {'role': 'assistant', 'content': 'Hello Im Bob, you asistant'},
+ {'role': 'user', 'content': 'what your name again?'},
+ ])
+
+print(response) \ No newline at end of file