hueso/Alpaca
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed connection handler

Browse Source
This commit is contained in:
jeffser
2024-07-07 21:49:04 -06:00
parent 215f7251d1
commit a723deaa00
2 changed files with 15 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/connection_handler.py
View File

@@ -22,4 +22,9 @@ def simple_delete(connection_url:str, data) -> dict:
return requests.delete(connection_url, headers=get_headers(False), json=data)
def stream_post(connection_url:str, data, callback:callable) -> dict:
return requests.post(connection_url, headers=get_headers(True), data=data, stream=True)
response = requests.post(connection_url, headers=get_headers(True), data=data, stream=True)
if response.status_code == 200:
for line in response.iter_lines():
if line:
callback(json.loads(line.decode("utf-8")))
return response