From 34e042bb3388e261968b2279ac4268cbd89ec193 Mon Sep 17 00:00:00 2001 From: ading2210 Date: Thu, 18 May 2023 00:05:09 -0700 Subject: add deepai wrapper --- testing/deepai_test.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) create mode 100644 testing/deepai_test.py (limited to 'testing/deepai_test.py') diff --git a/testing/deepai_test.py b/testing/deepai_test.py new file mode 100644 index 00000000..474f663e --- /dev/null +++ b/testing/deepai_test.py @@ -0,0 +1,18 @@ +from gpt4free import deepai + +#single completion +for chunk in deepai.Completion.create("Write a list of possible vacation destinations:"): + print(chunk, end="", flush=True) +print() + +#chat completion +print("==============") +messages = [ #taken from the openai docs + {"role": "system", "content": "You are a helpful assistant."}, + {"role": "user", "content": "Who won the world series in 2020?"}, + {"role": "assistant", "content": "The Los Angeles Dodgers won the World Series in 2020."}, + {"role": "user", "content": "Where was it played?"} +] +for chunk in deepai.ChatCompletion.create(messages): + print(chunk, end="", flush=True) +print() \ No newline at end of file -- cgit v1.2.3