Skip to content

Commit

Permalink
sfc service and sfp formatter reverted to working state
Browse files Browse the repository at this point in the history
  • Loading branch information
moon-strider committed Jan 15, 2024
1 parent 37952fd commit 195eb6e
Show file tree
Hide file tree
Showing 3 changed files with 9 additions and 6 deletions.
12 changes: 7 additions & 5 deletions annotators/speech_function_classifier/server.py
Original file line number Diff line number Diff line change
Expand Up @@ -51,8 +51,8 @@ def handler(payload: List[Dict]):
def answer():
st_time = time.time()
phrases = request.json.get("phrase", [])
prev_phrases = request.json.get("prev_phrase", [])
prev_speech_funcs = request.json.get("prev_speech_function", [])
prev_phrases = [request.json.get("prev_phrase", [])]
prev_speech_funcs = [request.json.get("prev_speech_function", [])]
payloads = []
for phr, prev_phr, prev_speech_func in zip(phrases, prev_phrases, prev_speech_funcs):
payloads.append(
Expand All @@ -67,9 +67,11 @@ def answer():
@app.route("/respond_batch", methods=["POST"]) # candidate annotator
def annotation():
st_time = time.time()
phrases = request.json.get("phrase", [])
prev_phrases = request.json.get("prev_phrase", [])
prev_speech_funcs = request.json.get("prev_speech_function", [])
responses, phrases, prev_phrases, prev_speech_funcs = [], [], [], []
for payload in request.json:
phrases.append(payload["phrase"])
prev_phrases.append(payload["prev_phrase"])
prev_speech_funcs.append(payload["prev_speech_function"])
payloads = []
for phr, prev_phr, prev_speech_func in zip(phrases, prev_phrases, prev_speech_funcs):
payloads.append(
Expand Down
1 change: 1 addition & 0 deletions annotators/speech_function_predictor/server.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,6 +64,7 @@ def answer():
def annotation():
st_time = time.time()
payload = request.json.get("funcs", [])
logger.info(f"PAYLOAD IN BATCH RESPOND SFP --------------------------------: {payload}")
responses = handler(payload)
total_time = time.time() - st_time
logger.info(f"speech_function_predictor batch exec time: {total_time:.3f}s")
Expand Down
2 changes: 1 addition & 1 deletion state_formatters/dp_formatters.py
Original file line number Diff line number Diff line change
Expand Up @@ -1183,7 +1183,7 @@ def speech_function_predictor_formatter(dialog: Dict):
res = None
res = dialog["human_utterances"][-1]["annotations"].get("speech_function_classifier", [""])
logger.info(f"formatter log sfp/sfc: {res}")
return [{"funcs": [res]}]
return [{"funcs": res}]


def speech_function_hypotheses_predictor_formatter(dialog: Dict):
Expand Down

0 comments on commit 195eb6e

Please sign in to comment.