feat(llm): add fromTemplate factory to the JSON driver

i-am-bee · Sep 23, 2024 · 6d3b356 · 6d3b356
1 parent 05d6dbb
commit 6d3b356
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/llms/drivers/json.ts b/src/llms/drivers/json.ts
@@ -42,6 +42,15 @@ IMPORTANT: Every message must be a parsable JSON string without additional outpu
     this.register();
   }
 
+  static fromTemplate<T extends JsonDriver["template"]>(
+    template: T,
+    ...parameters: ConstructorParameters<typeof this>
+  ) {
+    const driver = new JsonDriver(...parameters);
+    driver.template = template;
+    return driver;
+  }
+
   protected parseResponse(textResponse: string): unknown {
     return parseBrokenJson(textResponse);
   }