feat: simplify sendMessage options

README.md

@@ -133,8 +133,10 @@ const chat = createChat({
 });
 
 try {
-  await chat.sendMessage("continue sequence: a b c", ({ cancel }) => {
-    cancel();
+  await chat.sendMessage("continue sequence: a b c", {
+    onUpdate: ({ cancel }) => {
+      cancel();
+    },
   });
 } catch (error) {
   if (error instanceof CancelledCompletion) {
@@ -172,11 +174,9 @@ const chat = createChat({
 const response = await chat.sendMessage(
   "what is the next token in this sequence: a b c",
   {
-    options: {
-      maxTokens: 1,
-      // token 34093 is "boo"
-      logitBias: { "34093": 100 },
-    },
+    maxTokens: 1,
+    // token 34093 is "boo"
+    logitBias: { "34093": 100 },
   }
 );
 

src/createChat.test.ts

@@ -100,7 +100,7 @@ test("cancel response", async () => {
     chat.sendMessage("continue sequence: a b c", {
       onUpdate: ({ cancel }) => {
         cancel();
-      }
+      },
     })
   );
 });
@@ -144,7 +144,7 @@ test("calls user defined function", async () => {
   );
 
   assert.equal(response.role, "assistant");
-  assert.match(response.content, /the current weather in Albuquerque/i);
+  assert.match(response.content, /(the current weather in Albuquerque)|(weather in Albuquerque is currently)/i);
 });
 
 test("overrides function call", async () => {
@@ -181,9 +181,7 @@ test("overrides function call", async () => {
   });
 
   await chat.sendMessage("What is the weather in Chicago?", {
-    options: {
-      functionCall: "none",
-    },
+    functionCall: "none",
   });
 
   assert.equal(getCurrentWeather.mock.calls.length, 0);
@@ -198,11 +196,9 @@ test("overrides message options", async () => {
   const response = await chat.sendMessage(
     "what is the next token in this sequence: a b c",
     {
-      options: {
-        maxTokens: 1,
-        // token 34093 is "boo"
-        logitBias: { "34093": 100 },
-      },
+      maxTokens: 1,
+      // token 34093 is "boo"
+      logitBias: { "34093": 100 },
     }
   );
 

src/createChat.ts

@@ -7,13 +7,9 @@ import { retry } from "./retry";
 import { omit } from "./omit";
 import { createUserFunction, type UserFunction } from "./createUserFunction";
 
-type MessageOptions = {
-  functionName?: string;
-  onUpdate?: CompletionsOptions["onUpdate"];
-  options?: Partial<
-    Omit<Omit<Omit<CompletionsOptions, "messages">, "n">, "functions">
-  >;
-};
+type MessageOptions = Partial<
+  Omit<Omit<Omit<CompletionsOptions, "messages">, "n">, "functions">
+>;
 
 /**
  * @property apiKey - OpenAI API key.
@@ -81,7 +77,7 @@ export const createChat = (
         messages,
         onUpdate: messageOptions?.onUpdate,
         ...options,
-        ...messageOptions?.options,
+        ...messageOptions,
       });
     });