From 633f2312dc6b19c6f8b510c9ca89a8defe1a0f13 Mon Sep 17 00:00:00 2001 From: Bagus Indrayana Date: Wed, 12 Jul 2023 01:44:20 +0800 Subject: add testing --- testing/aiservice/testing.py | 30 ++++++++++++++++++++++++++++++ 1 file changed, 30 insertions(+) create mode 100644 testing/aiservice/testing.py (limited to 'testing/aiservice') 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 -- cgit v1.2.3