m

1 files changed, 61 insertions, 0 deletions

diff --git a/tests/index.test.ts b/tests/index.test.ts
new file mode 100644
index 0000000..1c328fa
--- /dev/null
+++ b/tests/index.test.ts
@@ -0,0 +1,61 @@
+import { describe, expect, test } from "bun:test";
+import selectModel from "../index";
+
+describe("Model selector smoke tests", () => {
+  // test("chatgpt model instantiates correctly and basic send works", async () => {
+  //   const api = selectModel({ chatgpt: "" });
+  //   expect(api).toBeDefined();
+  //   expect(typeof api.send).toBe("function");
+  //   expect(typeof api.stream).toBe("function");
+  //   expect(typeof api.setModel).toBe("function");
+  //   expect(typeof api.tokenizer).toBe("function");
+  //   expect(typeof api.maxTokens).toBe("number");
+  //   const res = await api.send("hello");
+  //   expect(res).toBeTruthy();
+  //   expect(typeof (res.ok ?? res.error)).toBe("string");
+  // });
+  // test("gemini model instantiates correctly and basic send works", async () => {
+  //   const api = selectModel({ gemini: "" });
+  //   expect(api).toBeDefined();
+  //   expect(typeof api.send).toBe("function");
+  //   expect(typeof api.stream).toBe("function");
+  //   const res = await api.send("hello");
+  //   console.log({ res });
+  // });
+  // test("claude model instantiates correctly and basic send works", async () => {
+  //   const api = selectModel({ claude: "" });
+  //   expect(api).toBeDefined();
+  //   expect(typeof api.send).toBe("function");
+  //   expect(typeof api.stream).toBe("function");
+  //   const res = await api.send("hello");
+  //   expect(res).toBeTruthy();
+  //   expect(typeof (res.ok ?? res.error)).toBe("string");
+  // });
+  // test("deepseek model instantiates correctly and basic send works", async () => {
+  //   const api = selectModel({ deepseek: "" });
+  //   expect(api).toBeDefined();
+  //   expect(typeof api.send).toBe("function");
+  //   expect(typeof api.stream).toBe("function");
+  //   const res = await api.send("hello");
+  //   expect(res).toBeTruthy();
+  //   expect(typeof (res.ok ?? res.error)).toBe("string");
+  // });
+  // test("kimi model instantiates correctly and basic send works", async () => {
+  //   const api = selectModel({ kimi: "" });
+  //   expect(api).toBeDefined();
+  //   expect(typeof api.send).toBe("function");
+  //   expect(typeof api.stream).toBe("function");
+  //   const res = await api.send("hello");
+  //   expect(res).toBeTruthy();
+  //   expect(typeof (res.ok ?? res.error)).toBe("string");
+  // });
+  // test("grok model instantiates correctly and basic send works", async () => {
+  //   const api = selectModel({ grok: "" });
+  //   expect(api).toBeDefined();
+  //   expect(typeof api.send).toBe("function");
+  //   expect(typeof api.stream).toBe("function");
+  //   const res = await api.send("hello");
+  //   expect(res).toBeTruthy();
+  //   expect(typeof (res.ok ?? res.error)).toBe("string");
+  // });
+});