From d7c4060c27f407ec98a7edd3b354c541739a3381 Mon Sep 17 00:00:00 2001
From: Michael Desmond <mdesmond@us.ibm.com>
Date: Tue, 19 Nov 2024 16:42:27 -0500
Subject: [PATCH] fix(agents): allow override granite bee prompts (#186)

Signed-off-by: MICHAEL DESMOND <mdesmond@us.ibm.com>
---
 src/agents/granite/agent.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/agents/granite/agent.ts b/src/agents/granite/agent.ts
index 56843f1c..4059a309 100644
--- a/src/agents/granite/agent.ts
+++ b/src/agents/granite/agent.ts
@@ -29,8 +29,8 @@ export class GraniteBeeAgent extends BeeAgent {
       ...input,
       templates: {
         ...input.templates,
-        system: GraniteBeeSystemPrompt,
-        assistant: GraniteBeeAssistantPrompt,
+        system: input.templates?.system ?? GraniteBeeSystemPrompt,
+        assistant: input.templates?.assistant ?? GraniteBeeAssistantPrompt,
       },
     });
   }