FastAPI-based working frontend (#10)
This commit is contained in:
20
playground/http_client.py
Normal file
20
playground/http_client.py
Normal file
@@ -0,0 +1,20 @@
|
||||
import requests
|
||||
import json
|
||||
|
||||
def http_bot():
|
||||
prompt = "How are you? I'm fine."
|
||||
|
||||
headers = {"User-Agent": "Test Client"}
|
||||
pload = {
|
||||
"prompt": prompt,
|
||||
}
|
||||
response = requests.post("http://localhost:10002", headers=headers, json=pload, stream=True)
|
||||
|
||||
for chunk in response.iter_lines(chunk_size=8192, decode_unicode=False, delimiter=b"\0"):
|
||||
if chunk:
|
||||
data = json.loads(chunk.decode("utf-8"))
|
||||
output = data["text"]
|
||||
yield output
|
||||
|
||||
for h in http_bot():
|
||||
print(h, end="", flush=True)
|
||||
Reference in New Issue
Block a user