souljoy commited on
Commit
faff1f9
1 Parent(s): 5a73c01

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1,4 +1,4 @@
1
- from fastapi import FastAPI
2
  from fastapi.responses import HTMLResponse
3
  from pydantic import BaseModel
4
  from fastapi.responses import JSONResponse, StreamingResponse
@@ -150,12 +150,12 @@ async def chat_gpt_stream_fun(content: Text = None):
150
  print(f"Full conversation received: {full_reply_content}")
151
  content = {'content': full_reply_content}
152
  print('content:', content)
153
- yield full_reply_content
154
 
155
 
156
- @app.post("/chatgptstream")
157
  async def get_random_numbers(content: Text = None):
158
- return StreamingResponse(chat_gpt_stream_fun(content))
159
 
160
 
161
  @app.post("/embeddings")
 
1
+ from fastapi import FastAPI, status
2
  from fastapi.responses import HTMLResponse
3
  from pydantic import BaseModel
4
  from fastapi.responses import JSONResponse, StreamingResponse
 
150
  print(f"Full conversation received: {full_reply_content}")
151
  content = {'content': full_reply_content}
152
  print('content:', content)
153
+ yield json.dumps(content) + '\n'
154
 
155
 
156
+ @app.post("/chatgptstream", status_code=status.HTTP_200_OK)
157
  async def get_random_numbers(content: Text = None):
158
+ return StreamingResponse(chat_gpt_stream_fun(content), media_type='application/json')
159
 
160
 
161
  @app.post("/embeddings")