feat: add json-schema-to-ts

lucgagan · Jun 27, 2023 · 536013c · 536013c
1 parent 837e786
commit 536013c
Show file tree

Hide file tree

Showing 3 changed files with 43 additions and 1 deletion.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -1,6 +1,7 @@
 {
   "dependencies": {
     "ajv": "^8.12.0",
+    "json-schema-to-ts": "^2.9.1",
     "ts-custom-error": "^3.3.1",
     "zod": "^3.21.4"
   },

diff --git a/src/createChat.ts b/src/createChat.ts
@@ -8,6 +8,7 @@ import { retry } from "./retry";
 import { omit } from "./omit";
 import { createUserFunction, type UserFunction } from "./createUserFunction";
 import Ajv, { AnySchemaObject } from "ajv";
+import { type FromSchema } from "json-schema-to-ts";
 
 type JsonValue =
   | JsonObject
@@ -164,7 +165,7 @@ export const createChat = (
 
   type SendMessageReturn<T> = T extends undefined
     ? Choice
-    : StructuredChoice<JsonObject>;
+    : StructuredChoice<FromSchema<T["expect"]["schema"]>>;
 
   function sendMessage<T extends MessageOptions>(
     prompt: string,