From b2037302e984a214014e32b00b2453a69adb8ef7 Mon Sep 17 00:00:00 2001 From: Andrew Tsegaye Date: Thu, 27 Apr 2023 09:17:07 -0700 Subject: well updated --- unfinished/easyai/main.py | 22 +++++++++++++++------- 1 file changed, 15 insertions(+), 7 deletions(-) (limited to 'unfinished/easyai/main.py') 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 -- cgit v1.2.3