Add llm prompt to generated code (#4981)

DARPA-ASKEM · Sep 30, 2024 · 486d751 · 486d751
1 parent bf99671
commit 486d751
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/packages/client/hmi-client/src/components/llm/tera-notebook-jupyter-input.vue b/packages/client/hmi-client/src/components/llm/tera-notebook-jupyter-input.vue
@@ -92,6 +92,8 @@ const hideAutoComplete = ref(false);
 
 const contextLanguageOptions = ref<string[]>(['python3', 'julia-1.10']);
 
+const addQuestionToCode = (question, code) => `\n# Prompt: ${question}\n\n${code}\n# End: =================`;
+
 const submitQuestion = () => {
 	const message = props.kernelManager.sendMessage('llm_request', {
 		request: questionString.value
@@ -103,6 +105,7 @@ const submitQuestion = () => {
 		kernelStatus.value = data.content.execution_state;
 	});
 	message.register('code_cell', (data) => {
+		data.content.code = addQuestionToCode(questionString.value, data.content.code);
 		emit('llm-output', data);
 	});
 	message.register('llm_thought', (data) => {