Skip to content
This repository was archived by the owner on Nov 13, 2023. It is now read-only.

Commit 1ea113c

Browse files
authored
Merge pull request #10 from thoraxe/wrap-response
use the happy response generator to wrap the response to the user
2 parents 5bf31ef + dd96a67 commit 1ea113c

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

ols.py

+5-2
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
from modules.task_processor import TaskProcessor
77
from modules.question_validator import QuestionValidator
88
from modules.yaml_generator import YamlGenerator
9+
from modules.happy_response_generator import HappyResponseGenerator
910
import logging, sys, os
1011

1112
logging.basicConfig(
@@ -153,10 +154,12 @@ def ols2_request(llm_request: LLMRequest):
153154
# RAG for supporting documentation
154155

155156
# generate a user-friendly response to wrap the YAML and/or the supporting information
156-
llm_response.response = generated_yaml
157-
157+
response_wrapper = HappyResponseGenerator()
158+
wrapper = response_wrapper.generate(conversation, llm_request.query)
158159

160+
llm_response.response = wrapper + "\n" + generated_yaml
159161
return llm_response
162+
160163
else:
161164
# something weird happened, so generate an internal error
162165
# something bad happened with the validation

0 commit comments

Comments
 (0)