diff options
Diffstat (limited to 'testing')
-rw-r--r-- | testing/aiassistest.py | 13 | ||||
-rw-r--r-- | testing/aicolors_test.py | 6 | ||||
-rw-r--r-- | testing/deepai_test.py | 18 | ||||
-rw-r--r-- | testing/gptworldai_test.py | 18 | ||||
-rw-r--r-- | testing/hpgptai_test.py | 41 | ||||
-rw-r--r-- | testing/italygpt2_test.py | 4 |
6 files changed, 100 insertions, 0 deletions
diff --git a/testing/aiassistest.py b/testing/aiassistest.py new file mode 100644 index 00000000..57a34f15 --- /dev/null +++ b/testing/aiassistest.py @@ -0,0 +1,13 @@ +import aiassist + +question1 = "Who won the world series in 2020?" +req = aiassist.Completion.create(prompt=question1) +answer = req["text"] +message_id = req["parentMessageId"] + +question2 = "Where was it played?" +req2 = aiassist.Completion.create(prompt=question2, parentMessageId=message_id) +answer2 = req2["text"] + +print(answer) +print(answer2) diff --git a/testing/aicolors_test.py b/testing/aicolors_test.py new file mode 100644 index 00000000..853f7e45 --- /dev/null +++ b/testing/aicolors_test.py @@ -0,0 +1,6 @@ +from gpt4free import aicolors + +prompt = "Light green color" +req = aicolors.Completion.create(prompt=prompt) + +print(req) 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 diff --git a/testing/gptworldai_test.py b/testing/gptworldai_test.py new file mode 100644 index 00000000..3dfb32ce --- /dev/null +++ b/testing/gptworldai_test.py @@ -0,0 +1,18 @@ +import gptworldAi + +# single completion +for chunk in gptworldAi.Completion.create("你是谁", "127.0.0.1:7890"): + print(chunk, end="", flush=True) +print() + +# chat completion +message = [] +while True: + prompt = input("请输入问题:") + message.append({"role": "user", "content": prompt}) + text = "" + for chunk in gptworldAi.ChatCompletion.create(message, '127.0.0.1:7890'): + text = text + chunk + print(chunk, end="", flush=True) + print() + message.append({"role": "assistant", "content": text}) diff --git a/testing/hpgptai_test.py b/testing/hpgptai_test.py new file mode 100644 index 00000000..cdd146dd --- /dev/null +++ b/testing/hpgptai_test.py @@ -0,0 +1,41 @@ +import hpgptai + +#single completion +res = hpgptai.Completion.create("你是谁","127.0.0.1:7890") +print(res["reply"]) + + +#chat completion +messages = [ + { + "content": "你是谁", + "html": "你是谁", + "id": hpgptai.ChatCompletion.randomStr(), + "role": "user", + "who": "User: ", + }, + { + "content": "我是一位AI助手,专门为您提供各种服务和支持。我可以回答您的问题,帮助您解决问题,提供相关信息,并执行一些任务。请随时告诉我您需要什么帮助。", + "html": "我是一位AI助手,专门为您提供各种服务和支持。我可以回答您的问题,帮助您解决问题,提供相关信息,并执行一些任务。请随时告诉我您需要什么帮助。", + "id": hpgptai.ChatCompletion.randomStr(), + "role": "assistant", + "who": "AI: ", + }, + { + "content": "我上一句问的是什么?", + "html": "我上一句问的是什么?", + "id": hpgptai.ChatCompletion.randomStr(), + "role": "user", + "who": "User: ", + }, +] +res = hpgptai.ChatCompletion.create(messages,proxy="127.0.0.1:7890") +print(res["reply"]) + + + + + + + + diff --git a/testing/italygpt2_test.py b/testing/italygpt2_test.py new file mode 100644 index 00000000..0494c8a2 --- /dev/null +++ b/testing/italygpt2_test.py @@ -0,0 +1,4 @@ +from gpt4free import italygpt2 +account_data=italygpt2.Account.create() +for chunk in italygpt2.Completion.create(account_data=account_data,prompt="Who are you?"): + print(chunk, end="", flush=True)
\ No newline at end of file |