fix passing of the assistant options to the invoke

src/LangtailAssistants.spec.ts

@@ -26,15 +26,15 @@ class MockLangtailPrompts implements ILangtailPrompts {
 }
 
 describe("LangtailAssistants", () => {
-  describe("public API", () => {
-    const createLt = () => {
-      const promptsMock = new MockLangtailPrompts()
-      return {
-        lt: new LangtailAssistants(promptsMock),
-        promptsMock
-      }
+  const createLt = () => {
+    const promptsMock = new MockLangtailPrompts()
+    return {
+      lt: new LangtailAssistants(promptsMock),
+      promptsMock
     }
+  }
 
+  describe("public API", () => {
     it("should translate assistant to prompt option in langtailPrompts.invoke call", async () => {
       const { lt, promptsMock } = createLt();
       const assistant = "test-assistant";
@@ -43,6 +43,7 @@ describe("LangtailAssistants", () => {
 
       expect(promptsMock.invoke).toHaveBeenCalledWith(
         expect.objectContaining({
+          assistant: true,
           prompt: assistant,
         })
       );
@@ -90,6 +91,7 @@ describe("LangtailAssistants", () => {
       });
 
       expect(promptsMock.invoke).toHaveBeenCalledWith({
+        assistant: true,
         prompt: assistant,
         messages,
       });
@@ -106,4 +108,35 @@ describe("LangtailAssistants", () => {
       expect(result).toBe(mockResponse);
     });
   })
+
+  it("should pass assistant: true by default to langtailPrompts.invoke call", async () => {
+    const { lt, promptsMock } = createLt();
+    const assistant = "test-assistant";
+
+    await lt.invoke({ assistant });
+
+    expect(promptsMock.invoke).toHaveBeenCalledWith(
+      expect.objectContaining({
+        assistant: true,
+        prompt: "test-assistant",
+      })
+    );
+  });
+
+  it("should pass threadId to langtailPrompts.invoke call when provided", async () => {
+    const { lt, promptsMock } = createLt();
+    const assistant = "test-assistant";
+    const threadId = "test-thread-id";
+
+    await lt.invoke({ assistant, threadId });
+
+    expect(promptsMock.invoke).toHaveBeenCalledWith(
+      expect.objectContaining({
+        assistant: true,
+        prompt: assistant,
+        threadId: threadId,
+      })
+    );
+  });
+
 })

src/LangtailAssistants.ts

@@ -12,13 +12,14 @@ export class LangtailAssistants {
     this.langtailPrompts = langtailPrompts
   }
 
-  invoke<P extends PromptSlug, E extends Environment<P> = undefined, V extends Version<P, E> = undefined, S extends boolean = false>(options: Omit<IRequestParamsStream<P, E, V, S>, "prompt"> & {
+  invoke<P extends PromptSlug, E extends Environment<P> = undefined, V extends Version<P, E> = undefined, S extends boolean = false>(options: Omit<IRequestParamsStream<P, E, V, S>, "prompt" | "assistant"> & {
     assistant: P
   }): Promise<S extends true ? StreamResponseType : OpenAIResponseWithHttp> {
     const { assistant, ...rest } = options
     return this.langtailPrompts.invoke<P, E, V, S>({
       ...rest,
-      prompt: assistant
+      prompt: assistant,
+      assistant: true,
     })
   }
 }

src/schemas.ts

@@ -7,6 +7,8 @@ import type {
 
 export interface ILangtailExtraProps {
   doNotRecord?: boolean
+  threadId?: string
+  assistant?: boolean
   metadata?: Record<string, any>
 }