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