summaryrefslogtreecommitdiffstats
path: root/unfinished/easyai
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--unfinished/easyai/main.py24
1 files changed, 15 insertions, 9 deletions
diff --git a/unfinished/easyai/main.py b/unfinished/easyai/main.py
index 0607d58a..3b246977 100644
--- a/unfinished/easyai/main.py
+++ b/unfinished/easyai/main.py
@@ -1,4 +1,8 @@
-import requests
+from requests import get
+from os import urandom
+from json import loads
+
+sessionId = urandom(10).hex()
headers = {
'Accept': 'text/event-stream',
@@ -12,15 +16,17 @@ headers = {
}
while True:
+ prompt = input('you: ')
+
params = {
- 'message': 'what is my name',
- 'sessionId': '2eacb8ad826056587598',
+ 'message': prompt,
+ 'sessionId': sessionId
}
- for chunk in requests.get('http://easy-ai.ink/easyapi/v1/chat/completions', params=params,
- headers=headers, verify=False, stream=True).iter_lines():
-
- if b'data:' in chunk:
- print(chunk)
+ for chunk in get('http://easy-ai.ink/easyapi/v1/chat/completions', params = params,
+ headers = headers, verify = False, stream = True).iter_lines():
- print(chunk) \ No newline at end of file
+ if b'content' in chunk:
+ data = loads(chunk.decode('utf-8').split('data:')[1])
+ print(data['content'], end='')
+ \ No newline at end of file