Allow post and get

jina-ai · Oct 20, 2023 · 64fc314 · 64fc314
1 parent d4da6e9
commit 64fc314
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/jina/clients/base/helper.py b/jina/clients/base/helper.py
@@ -197,13 +197,12 @@ async def send_streaming_message(self, doc: 'Document', on: str):
         :param on: Request endpoint
         :yields: responses
         """
-        req_dict = doc.json().encode()
+        req_dict = doc.json()
 
         request_kwargs = {
             'url': self.url,
             'headers': {'Accept': 'text/event-stream'},
         }
-        # request_kwargs['data'] = JinaJsonPayload(value=req_dict)
         request_kwargs['data'] = req_dict
 
         async with self.session.post(**request_kwargs) as response:

diff --git a/jina/serve/runtimes/gateway/http_fastapi_app_docarrayv2.py b/jina/serve/runtimes/gateway/http_fastapi_app_docarrayv2.py
@@ -260,8 +260,8 @@ async def event_generator():
                 ):
                     if error:
                         raise HTTPException(status_code=499, detail=str(error))
-                    yield {'data': doc.dict()}
-                yield {'data': 'DONE'}
+                    yield {'event': 'update', 'data': doc.dict()}
+                yield {'event': 'end'}
 
             return EventSourceResponse(event_generator())