summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorading2210 <ading2019@gmail.com>2023-05-18 09:05:09 +0200
committerading2210 <ading2019@gmail.com>2023-05-18 09:07:29 +0200
commit34e042bb3388e261968b2279ac4268cbd89ec193 (patch)
tree7289075609fc1cdffdde7907b3e592abf7472ba2
parentMerge pull request #543 from mitchcapper/analytics_warning_pr (diff)
downloadgpt4free-34e042bb3388e261968b2279ac4268cbd89ec193.tar
gpt4free-34e042bb3388e261968b2279ac4268cbd89ec193.tar.gz
gpt4free-34e042bb3388e261968b2279ac4268cbd89ec193.tar.bz2
gpt4free-34e042bb3388e261968b2279ac4268cbd89ec193.tar.lz
gpt4free-34e042bb3388e261968b2279ac4268cbd89ec193.tar.xz
gpt4free-34e042bb3388e261968b2279ac4268cbd89ec193.tar.zst
gpt4free-34e042bb3388e261968b2279ac4268cbd89ec193.zip
Diffstat (limited to '')
-rw-r--r--gpt4free/__init__.py8
-rw-r--r--gpt4free/deepai/README.md26
-rw-r--r--gpt4free/deepai/__init__.py46
-rw-r--r--testing/deepai_test.py18
4 files changed, 97 insertions, 1 deletions
diff --git a/gpt4free/__init__.py b/gpt4free/__init__.py
index 1e652897..739cf763 100644
--- a/gpt4free/__init__.py
+++ b/gpt4free/__init__.py
@@ -15,6 +15,7 @@ class Provider(Enum):
ForeFront = 'fore_front'
Theb = 'theb'
UseLess = 'useless'
+ DeepAI = 'deepai'
class Completion:
@@ -40,6 +41,8 @@ class Completion:
return Completion.__theb_service(prompt, **kwargs)
elif provider == Provider.UseLess:
return Completion.__useless_service(prompt, **kwargs)
+ elif provider == Provider.DeepAI:
+ return Completion.__deepai_service(prompt, **kwargs)
else:
raise Exception('Provider not exist, Please try again')
@@ -62,4 +65,7 @@ class Completion:
@staticmethod
def __theb_service(prompt: str, **kwargs):
return ''.join(theb.Completion.create(prompt=prompt))
-
+
+ @staticmethod
+ def __deepai_service(prompt: str, **kwargs):
+ return ''.join(deepai.Completion.create(prompt=prompt))
diff --git a/gpt4free/deepai/README.md b/gpt4free/deepai/README.md
new file mode 100644
index 00000000..a287cdb7
--- /dev/null
+++ b/gpt4free/deepai/README.md
@@ -0,0 +1,26 @@
+# DeepAI Wrapper
+Written by [ading2210](https://github.com/ading2210/).
+
+## Examples:
+These functions are generators which yield strings containing the newly generated text.
+
+### Completion:
+```python
+for chunk in deepai.Completion.create("Who are you?"):
+ print(chunk, end="", flush=True)
+print()
+```
+
+### Chat Completion:
+Use the same format for the messages as you would for the [official OpenAI API](https://platform.openai.com/docs/guides/chat/introduction).
+```python
+messages = [
+ {"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/gpt4free/deepai/__init__.py b/gpt4free/deepai/__init__.py
new file mode 100644
index 00000000..a2fc6f5a
--- /dev/null
+++ b/gpt4free/deepai/__init__.py
@@ -0,0 +1,46 @@
+import requests
+import json
+import hashlib
+import random
+import string
+from fake_useragent import UserAgent
+
+class ChatCompletion:
+ @classmethod
+ def md5(self, text):
+ return hashlib.md5(text.encode()).hexdigest()[::-1]
+
+ @classmethod
+ def get_api_key(self, user_agent):
+ part1 = str(random.randint(0, 10**11))
+ part2 = self.md5(user_agent+self.md5(user_agent+self.md5(user_agent+part1+"x")))
+ return f"tryit-{part1}-{part2}"
+
+ @classmethod
+ def create(self, messages):
+ user_agent = UserAgent().random
+ api_key = self.get_api_key(user_agent)
+ headers = {
+ "api-key": api_key,
+ "user-agent": user_agent
+ }
+ files = {
+ "chat_style": (None, "chat"),
+ "chatHistory": (None, json.dumps(messages))
+ }
+
+ r = requests.post("https://api.deepai.org/chat_response", headers=headers, files=files, stream=True)
+
+ for chunk in r.iter_content(chunk_size=None):
+ r.raise_for_status()
+ yield chunk.decode()
+
+class Completion:
+ @classmethod
+ def create(self, prompt):
+ return ChatCompletion.create([
+ {
+ "role": "user",
+ "content": prompt
+ }
+ ]) \ No newline at end of file
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