alex/nova-api
1
0
Fork 0
mirror of https://github.com/NovaOSS/nova-api.git synced 2024-11-25 16:13:58 +01:00
Code Issues Packages Projects Releases Wiki Activity

idk

Browse source
This commit is contained in:
nsde 2023-08-06 12:46:41 +02:00
parent ea35674853
commit bc5cd7122e
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
api/streaming.py
View file

@ -86,6 +86,7 @@ async def stream(
'Sorry, the API has no working keys anymore.',
'The admins have been messaged automatically.'
)
return
for k, v in target_request.get('headers', {}).items():
headers[k] = v
@ -138,6 +139,7 @@ async def stream(
chunk = f'{chunk.decode("utf8")}\n\n'
chunk = chunk.replace(os.getenv('MAGIC_WORD', 'novaOSScheckKeyword'), payload['model'])
# chunk = chunk.replace(os.getenv('MAGIC_USER_WORD', 'novaOSSuserKeyword'), user['_id'])
print(chunk)
if not chunk.strip():
send = False
@ -145,7 +147,6 @@ async def stream(
if is_chat and '{' in chunk:
data = json.loads(chunk.split('data: ')[1])
send = True
print(data)
if target_request['module'] == 'twa' and data.get('text'):
chunk = chat.create_chat_chunk(