Make ask prompt configurable

examples/writer-agent/bot1_role.txt

@@ -0,0 +1,9 @@
+Your name is Paul. You are a very creative screenwriter. Your expertise is superhero action shortfilms.
+You will provide a new script for a new shortfilm.
+
+Your task is to collaborate with Lina to create an exciting script. Your task is to always iterate on Ms Lina's critic and complete the assignment.
+Assigment: write a 100 word shortscript about a new superhero in town, called "Pie Man" and his adventures.
+
+You will ALWAYS converse in this structure:
+Response: Here is where you response to Mr. Lisa
+Story: Here is where you write your script

examples/writer-agent/bot2_role.txt

@@ -0,0 +1,11 @@
+You are an script editor reviewer and your name is Lisa. Your are an expert reviewer working for a company like Marvel.
+
+Your task is to collaborate with Paul to create exciting scripts for shortfilms.
+Your task is to always iterate on Mr Pauls text and complete the assignment.
+
+Assignment:  Be very critical of Mr Paul and his writting to help him to write the best piece of text. The shortfilm has to be an action film and has to entertain the viewer.
+
+You will ALWAYS converse in this structure:
+
+Response: Here is where you response to Mr. Paul
+Critique: Here you write your critic to Mr. Paul

examples/writer-agent/writer_agent.nu

@@ -0,0 +1,21 @@
+let bot1_role = open bot1_role.txt;
+let bot2_role = open bot2_role.txt;
+
+def bot [] {
+ mut response_bot1 = "";
+ mut response_bot2 = "";
+ for $x in 1..6 {
+    print ($"****************** ITERATION ($x) ***************")
+    let rep = ask --prompt $bot1_role $response_bot1
+    $response_bot1 =  $rep
+    $response_bot2 =  $rep
+    print ($"WRITER:\n ($response_bot1)")
+
+    let rep2 = ask --prompt $bot2_role $response_bot2
+    $response_bot1 =  $rep2
+    $response_bot2 =  $rep
+    print ($"EDITOR:\n ($response_bot2)")
+ }
+}
+
+bot

src/cli/ask.rs

@@ -44,6 +44,12 @@ impl Command for Ask {
                 "the chat model to ask the question",
                 None,
             )
+            .named(
+                "prompt",
+                SyntaxShape::String,
+                "the prompt used by the model",
+                None,
+            )
             .category(Category::Custom("couchbase".to_string()))
     }
 
@@ -87,6 +93,7 @@ pub fn ask(
     let span = call.head;
 
     let question: String = call.req(engine_state, stack, 0)?;
+    let prompt_template: Option<String> = call.get_flag(engine_state, stack, "prompt")?;
     let context: Vec<String> = match call.opt(engine_state, stack, 1)? {
         Some(ctx) => ctx,
         None => {
@@ -176,7 +183,7 @@ pub fn ask(
     let rt = Runtime::new().unwrap();
     let answer = match rt.block_on(async {
         select! {
-            answer = client.ask(question.clone(), context.clone(), model) => {
+            answer = client.ask(question.clone(), prompt_template.clone(), context.clone(), model) => {
                 match answer {
                     Ok(a) => Ok(a),
                     Err(e) => Err(e),

src/client/bedrock_client.rs

@@ -108,18 +108,18 @@ impl BedrockClient {
     pub async fn ask(
         &self,
         question: String,
+        template: Option<String>,
         context: Vec<String>,
         model: String,
     ) -> Result<String, ShellError> {
         let config = aws_config::load_from_env().await;
         let client = aws_sdk_bedrockruntime::Client::new(&config);
 
+        let tpl_value = template.unwrap_or( "Please answer this question: \\\"{}\\\". Using the following context: \\\"{}\\\"".to_string());
+        let mut rendered_tpl = tpl_value.replacen("{}", &*question, 1);
+        rendered_tpl = rendered_tpl.replacen("{}", &*context.join(" "), 1);
         let question_with_ctx = if !context.is_empty() {
-            format!(
-                "Please answer this question: \\\"{}\\\". Using the following context: \\\"{}\\\"",
-                question,
-                context.join(" ")
-            )
+            rendered_tpl
         } else {
             question
         };

src/client/gemini_client.rs

@@ -129,20 +129,19 @@ impl GeminiClient {
     pub async fn ask(
         &self,
         question: String,
+        template: Option<String>,
         context: Vec<String>,
         model: String,
     ) -> Result<String, ShellError> {
         let url = format!(
             "https://generativelanguage.googleapis.com/v1beta/models/{}:generateContent?key={}",
             model, self.api_key
         );
-
+        let tpl_value = template.unwrap_or( "Please answer this question: \\\"{}\\\". Using the following context: \\\"{}\\\"".to_string());
+        let mut rendered_tpl = tpl_value.replacen("{}", &*question, 1);
+        rendered_tpl = rendered_tpl.replacen("{}", &*context.join(" "), 1);
         let question_with_ctx = if !context.is_empty() {
-            format!(
-                "Please answer this question: \\\"{}\\\". Using the following context: \\\"{}\\\"",
-                question,
-                context.join(" ")
-            )
+            rendered_tpl
         } else {
             question
         };

src/client/llm_client.rs

@@ -37,13 +37,14 @@ impl LLMClients {
     pub async fn ask(
         &self,
         question: String,
+        template: Option<String>,
         context: Vec<String>,
         model: String,
     ) -> Result<String, ShellError> {
         match self {
-            Self::OpenAI(c) => c.ask(question, context, model).await,
-            Self::Gemini(c) => c.ask(question, context, model).await,
-            Self::Bedrock(c) => c.ask(question, context, model).await,
+            Self::OpenAI(c) => c.ask(question, template, context, model).await,
+            Self::Gemini(c) => c.ask(question, template, context, model).await,
+            Self::Bedrock(c) => c.ask(question, template, context, model).await,
         }
     }
 

src/client/openai_client.rs

@@ -130,15 +130,15 @@ impl OpenAIClient {
     pub async fn ask(
         &self,
         question: String,
+        template: Option<String>,
         context: Vec<String>,
         model: String,
     ) -> Result<String, ShellError> {
         let mut messages: Vec<ChatCompletionRequestMessage> = vec![];
-
         // Primes the model to respond appropriately
         messages.push(
             ChatCompletionRequestSystemMessageArgs::default()
-                .content("You are a helpful assistant.")
+                .content(template.unwrap_or("You are a helpful assistant.".to_string()))
                 .build()
                 .unwrap()
                 .into(),