Liyan06
commited on
Commit
·
e9622b9
1
Parent(s):
9695d05
add server-sent events (SSE) function
Browse files- handler.py +3 -2
handler.py
CHANGED
|
@@ -35,6 +35,8 @@ class EndpointHandler():
|
|
| 35 |
'ranked_docs': ranked_docs,
|
| 36 |
'scores': scores
|
| 37 |
}
|
|
|
|
|
|
|
| 38 |
|
| 39 |
else:
|
| 40 |
assert len(data['inputs']['claims']) == 1, "Only one claim is allowed for web retrieval for the current version."
|
|
@@ -53,10 +55,9 @@ class EndpointHandler():
|
|
| 53 |
}
|
| 54 |
yield f"data: {json.dumps(outputs)}\n"
|
| 55 |
|
| 56 |
-
|
| 57 |
|
| 58 |
|
| 59 |
-
|
| 60 |
def search_relevant_docs(self, claim, timeout=10, max_search_results_per_query=5, allow_duplicated_urls=False):
|
| 61 |
|
| 62 |
search_results = search_google(claim, timeout=timeout)
|
|
|
|
| 35 |
'ranked_docs': ranked_docs,
|
| 36 |
'scores': scores
|
| 37 |
}
|
| 38 |
+
|
| 39 |
+
return outputs
|
| 40 |
|
| 41 |
else:
|
| 42 |
assert len(data['inputs']['claims']) == 1, "Only one claim is allowed for web retrieval for the current version."
|
|
|
|
| 55 |
}
|
| 56 |
yield f"data: {json.dumps(outputs)}\n"
|
| 57 |
|
| 58 |
+
return generate_sse()
|
| 59 |
|
| 60 |
|
|
|
|
| 61 |
def search_relevant_docs(self, claim, timeout=10, max_search_results_per_query=5, allow_duplicated_urls=False):
|
| 62 |
|
| 63 |
search_results = search_google(claim, timeout=timeout)
|