summaryrefslogtreecommitdiffstats
path: root/unfinished/easyai/main.py
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--unfinished/easyai/main.py22
1 files changed, 15 insertions, 7 deletions
diff --git a/unfinished/easyai/main.py b/unfinished/easyai/main.py
index 3b246977..475430ee 100644
--- a/unfinished/easyai/main.py
+++ b/unfinished/easyai/main.py
@@ -1,9 +1,12 @@
+# Import necessary libraries
from requests import get
-from os import urandom
-from json import loads
+from os import urandom
+from json import loads
+# Generate a random session ID
sessionId = urandom(10).hex()
+# Set up headers for the API request
headers = {
'Accept': 'text/event-stream',
'Accept-Language': 'en,fr-FR;q=0.9,fr;q=0.8,es-ES;q=0.7,es;q=0.6,en-US;q=0.5,am;q=0.4,de;q=0.3',
@@ -15,18 +18,23 @@ headers = {
'token': 'null',
}
+# Main loop to interact with the AI
while True:
+ # Get user input
prompt = input('you: ')
-
+
+ # Set up parameters for the API request
params = {
'message': prompt,
'sessionId': sessionId
}
- for chunk in get('http://easy-ai.ink/easyapi/v1/chat/completions', params = params,
- headers = headers, verify = False, stream = True).iter_lines():
-
+ # Send request to the API and process the response
+ for chunk in get('http://easy-ai.ink/easyapi/v1/chat/completions', params=params,
+ headers=headers, verify=False, stream=True).iter_lines():
+
+ # Check if the chunk contains the 'content' field
if b'content' in chunk:
+ # Parse the JSON data and print the content
data = loads(chunk.decode('utf-8').split('data:')[1])
print(data['content'], end='')
- \ No newline at end of file