Make both system and user prompt optional in customPrompt return

arshad-yaseen · Sep 10, 2024 · d04cef2 · d04cef2
1 parent 6769cb4
commit d04cef2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/src/helpers/provider.ts b/src/helpers/provider.ts
@@ -26,9 +26,11 @@ export const createRequestBody = (
   provider: CompletionProvider,
   customPrompt: CustomPrompt | undefined,
 ): PickChatCompletionCreateParams<CompletionProvider> => {
-  const {system: systemPrompt, user: userPrompt} = customPrompt
-    ? customPrompt(completionMetadata)
-    : generatePrompt(completionMetadata);
+  const defaultPrompts = generatePrompt(completionMetadata);
+  const customPrompts = customPrompt ? customPrompt(completionMetadata) : {};
+
+  const systemPrompt = customPrompts.system ?? defaultPrompts.system;
+  const userPrompt = customPrompts.user ?? defaultPrompts.user;
 
   const commonParams = {
     model: getModelId(model),

diff --git a/src/types/completion.ts b/src/types/completion.ts
@@ -95,8 +95,8 @@ export interface CompletionRequestOptions {
 }
 
 export type CustomPrompt = (completionMetadata: CompletionMetadata) => {
-  system: string;
-  user: string;
+  system?: string;
+  user?: string;
 };
 
 export interface CompletionResponse {