diff options
Diffstat (limited to 'testing')
| -rw-r--r-- | testing/aiservice/testing.py | 30 |
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 |