Liyan06 commited on
Commit
e9622b9
·
1 Parent(s): 9695d05

add server-sent events (SSE) function

Browse files
Files changed (1) hide show
  1. 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
- return generate_sse()
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)