Fix a few build issues

braintrustdata · Nov 24, 2023 · 5c836de · 5c836de
1 parent bb24b8a
commit 5c836de
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 9 deletions.
diff --git a/.gitignore b/.gitignore
@@ -15,3 +15,4 @@ autoevals-*.tar.gz
 autoevals-*.tgz
 typedoc.json
 build
+.turbo
diff --git a/js/llm.ts b/js/llm.ts
@@ -7,6 +7,7 @@ import { templates } from "./templates.js";
 import {
   ChatCompletionCreateParams,
   ChatCompletionMessage,
+  ChatCompletionMessageParam,
 } from "openai/resources/index.mjs";
 import { currentSpan } from "./util.js";
 
@@ -63,7 +64,7 @@ export function buildClassificationFunctions(useCoT: boolean) {
 export type OpenAIClassifierArgs<RenderArgs> = {
   name: string;
   model: string;
-  messages: ChatCompletionMessage[];
+  messages: ChatCompletionMessageParam[];
   choiceScores: Record<string, number>;
   classificationFunctions: ChatCompletionCreateParams.Function[];
   cache?: ChatCache;
@@ -117,9 +118,9 @@ export async function OpenAIClassifier<RenderArgs, Output>(
     ...remainingRenderArgs,
   };
 
-  const messages: ChatCompletionMessage[] = messagesArg.map((m) => ({
+  const messages: ChatCompletionMessageParam[] = messagesArg.map((m) => ({
     ...m,
-    content: m.content && mustache.render(m.content, renderArgs),
+    content: m.content && mustache.render(m.content as string, renderArgs),
   }));
 
   let ret = null;
@@ -220,7 +221,7 @@ export function LLMClassifierFromTemplate<RenderArgs>({
       promptTemplate + "\n" + (useCoT ? COT_SUFFIX : NO_COT_SUFFIX);
 
     let maxTokens = 512;
-    const messages: ChatCompletionMessage[] = [
+    const messages: ChatCompletionMessageParam[] = [
       {
         role: "user",
         content: prompt,

diff --git a/js/oai.ts b/js/oai.ts
@@ -1,7 +1,7 @@
 import {
   ChatCompletion,
   ChatCompletionCreateParams,
-  ChatCompletionMessage,
+  ChatCompletionMessageParam,
 } from "openai/resources/index.mjs";
 import { OpenAI } from "openai";
 
@@ -10,9 +10,9 @@ import { currentSpan } from "./util.js";
 
 export interface CachedLLMParams {
   model: string;
-  messages: ChatCompletionMessage[];
+  messages: ChatCompletionMessageParam[];
   functions?: ChatCompletionCreateParams.Function[];
-  function_call?: ChatCompletionCreateParams.FunctionCallOption;
+  function_call?: ChatCompletionCreateParams["function_call"];
   temperature?: number;
   max_tokens?: number;
 }

diff --git a/tsconfig.json b/tsconfig.json
@@ -4,12 +4,12 @@
     "declaration": true,
     "outDir": "./jsdist",
     "lib": ["es6"],
-    "module": "commonjs",
+    "module": "NodeNext",
     "target": "es6",
     "moduleResolution": "nodenext",
     "strict": true,
     "esModuleInterop": true,
-    "skipLibCheck": true,
+    "skipLibCheck": true
   },
   "include": ["js"],
   "exclude": ["node_modules/**"]
-Original file line number
+Diff line change
@@ Expand Up / @@ -15,3 +15,4 @@ autoevals-*.tar.gz @@
     autoevals-*.tgz
     typedoc.json
     build
+    .turbo