diff --git a/examples/llamarine/app.py b/examples/llamarine/app.py index 9eae43b2a..a314e6541 100644 --- a/examples/llamarine/app.py +++ b/examples/llamarine/app.py @@ -71,13 +71,15 @@ def main(use_domain_lm: bool = False): logger.level('DEBUG') st.session_state.agent_solutions[st.session_state.typed_problem]: str = \ - get_or_create_agent(use_domain_lm=use_domain_lm).solve( + get_or_create_agent(use_domain_lm).solve( problem=st.session_state.typed_problem, allow_reject=True) if (solution := st.session_state.agent_solutions[st.session_state.typed_problem]): if use_domain_lm: solution = OpenAILM.from_defaults().get_response( - prompt=solution, + prompt=f"""Please respond the following text: + {solution} + """, history=[ {"role": "system", "content": LLAMARINE_SYSTEM_PROMPT}, {"role": "user", "content": LLAMARINE_USER_PROMPT},