fix(agents): allow override granite bee prompts (#186)

Signed-off-by: MICHAEL DESMOND <[email protected]>
i-am-bee · Nov 19, 2024 · d7c4060 · d7c4060
1 parent 4c933c4
commit d7c4060
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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,
       },
     });
   }