TypoX / content /conn.py
rexrex9
init a
4396f99
raw
history blame contribute delete
830 Bytes
# coding=UTF-8
import os
import requests
BASE_URL = os.getenv('TYPO_URL')
API_KEY = os.getenv('API_KEY')
def sub(text):
url = BASE_URL.format(text)
headers = {
'x-api-key': API_KEY
}
res = requests.get(url,headers=headers,stream=True)
n_count = -1
for chunk in res.iter_lines():
#print(n_count)
s = chunk.decode('utf-8')
if 'data:' in s:
n_count = 0
#print(s.split('data: ')[-1])
yield s.split('data: ')[-1]
else:
#print(s)
if s.strip() =='':
n_count += 1
if n_count >= 2:
yield '\n'
def stream_function(input):
res = sub(input)
s=''
for r in res:
s+=r
print(s)
if __name__ == '__main__':
stream_function('what is DeFi?')