From a56bdb077af4f144871151c0ca6f08c88e5df6b1 Mon Sep 17 00:00:00 2001
From: Jason <jason@jxnl.co>
Date: Mon, 24 Jul 2023 15:24:25 +0800
Subject: [PATCH] fix json

---
 examples/citation_with_extraction/main.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/examples/citation_with_extraction/main.py b/examples/citation_with_extraction/main.py
index 618dd9c..7d01588 100644
--- a/examples/citation_with_extraction/main.py
+++ b/examples/citation_with_extraction/main.py
@@ -1,3 +1,4 @@
+import json
 from typing import Iterable, List
 from fastapi import FastAPI, Request, HTTPException
 from fastapi.params import Depends
@@ -121,6 +122,7 @@ async def extract(question: Question, openai_key=Depends(get_api_key)):
                 "spans": spans,
                 "citation": [question.context[a:b] for (a, b) in spans],
             }
-            yield f"data: {resp}"
+            resp_json = json.dumps(resp)
+            yield f"data: {resp_json}"
 
     return StreamingResponse(generate(), media_type="text/event-stream")