From 73618a91618a4f7e39acf0f890ba0fe510175022 Mon Sep 17 00:00:00 2001
From: "t.me/xtekky" <98614666+xtekky@users.noreply.github.com>
Date: Wed, 26 Apr 2023 12:12:05 +0100
Subject: http://easy-ai.ink/

---
 unfinished/easyai/main.py | 24 +++++++++++++++---------
 1 file changed, 15 insertions(+), 9 deletions(-)

(limited to 'unfinished/easyai')

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
-- 
cgit v1.2.3