Merge pull request #43 from askorama/feat/add-custom-prompts-prop

Feat: add custom prompt configuration

src/answerSession.ts

@@ -192,6 +192,11 @@ export class AnswerSession {
       requestBody.append('interactionId', interactionId)
       requestBody.append('alias', this.oramaClient.getAlias() ?? '')
 
+      const systemPromptConfiguration = this.oramaClient.getSystemPromptConfiguration()
+      if (systemPromptConfiguration) {
+        requestBody.append('systemPrompts', JSON.stringify(systemPromptConfiguration))
+      }
+
       if (this.userContext) {
         requestBody.append('userContext', serializeUserContext(this.userContext))
       }

src/client.ts

@@ -69,6 +69,7 @@ export class OramaClient {
   private readonly collector?: Collector
   private readonly cache?: Cache<Results<AnyDocument>>
   private readonly profile: Profile
+  private systemPrompts?: string[]
   private searchDebounceTimer?: any // NodeJS.Timer
   private searchRequestCounter = 0
   private blockSearchTillAuth = false
@@ -430,4 +431,21 @@ export class OramaClient {
   public reset(): void {
     this.profile.reset()
   }
+
+  /**
+   * Methods associated with custom system prompts
+   */
+  public setSystemPromptConfiguration(config: { systemPrompts: string[] }): void {
+    if (Array.isArray(config.systemPrompts)) {
+      if (!config.systemPrompts.every((prompt) => typeof prompt === 'string')) {
+        throw new Error('Invalid system prompt configuration')
+      }
+
+      this.systemPrompts = config.systemPrompts
+    }
+  }
+
+  public getSystemPromptConfiguration(): string[] | undefined {
+    return this.systemPrompts
+  }
 }