add testing

2024-12-25 20:22:47 +03:00 · 2023-07-12 01:44:20 +08:00 · 2023-07-12 01:44:20 +08:00 · 633f2312dc
commit 633f2312dc
parent 5705931506
1 changed files with 30 additions and 0 deletions
--- a/testing/aiservice/testing.py
+++ 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)