Skip to content

Commit abd9d12

Browse files
Fix non stream case (#1115)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1 parent a7353bb commit abd9d12

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

ChatQnA/chatqna.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -148,6 +148,8 @@ def align_outputs(self, data, cur_node, inputs, runtime_graph, llm_parameters_di
148148

149149
next_data["inputs"] = prompt
150150

151+
elif self.services[cur_node].service_type == ServiceType.LLM and not llm_parameters_dict["streaming"]:
152+
next_data["text"] = data["choices"][0]["message"]["content"]
151153
else:
152154
next_data = data
153155

0 commit comments

Comments
 (0)