Update openai types (#15)

Fixes #14 Ref: openai/openai-node#266
hmarr · Sep 28, 2023 · 75891fc · 75891fc
1 parent 1ab24fe
commit 75891fc
Show file tree

Hide file tree

Showing 5 changed files with 13 additions and 13 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -26,7 +26,7 @@
   "devDependencies": {
     "@types/jest": "^29.5.3",
     "jest": "^29.6.1",
-    "openai": "^4.2.0",
+    "openai": "^4.10.0",
     "prettier": "^3.0.2",
     "ts-jest": "^29.1.1",
     "typescript": "^5.1.6"
@@ -37,4 +37,4 @@
   "prettier": {
     "trailingComma": "all"
   }
-}
+}
diff --git a/src/functions.ts b/src/functions.ts
@@ -1,6 +1,6 @@
 import OpenAI from "openai";
 
-type OpenAIFunction = OpenAI.Chat.CompletionCreateParams.Function;
+type OpenAIFunction = OpenAI.Chat.ChatCompletionCreateParams.Function;
 
 // Types representing the OpenAI function definitions. While the OpenAI client library
 // does have types for function definitions, the properties are just Record<string, unknown>,

diff --git a/src/index.ts b/src/index.ts
@@ -2,9 +2,9 @@ import OpenAI from "openai";
 import { Tiktoken, getEncoding } from "js-tiktoken";
 import { FunctionDef, formatFunctionDefinitions } from "./functions";
 
-type Message = OpenAI.Chat.CreateChatCompletionRequestMessage;
-type Function = OpenAI.Chat.CompletionCreateParams.Function;
-type FunctionCall = OpenAI.Chat.CompletionCreateParams.FunctionCallOption;
+type Message = OpenAI.Chat.ChatCompletionMessageParam;
+type Function = OpenAI.Chat.ChatCompletionCreateParams.Function;
+type FunctionCall = OpenAI.Chat.ChatCompletionCreateParams.FunctionCallOption;
 
 let encoder: Tiktoken | undefined;
 

diff --git a/tests/token-counts.test.ts b/tests/token-counts.test.ts
@@ -1,9 +1,9 @@
 import OpenAI from "openai";
 import { promptTokensEstimate } from "../src";
 
-type Message = OpenAI.Chat.CreateChatCompletionRequestMessage;
-type Function = OpenAI.Chat.CompletionCreateParams.Function;
-type FunctionCall = OpenAI.Chat.CompletionCreateParams.FunctionCallOption;
+type Message = OpenAI.Chat.ChatCompletionMessageParam;
+type Function = OpenAI.Chat.ChatCompletionCreateParams.Function;
+type FunctionCall = OpenAI.Chat.ChatCompletionCreateParams.FunctionCallOption;
 type Example = {
   messages: Message[];
   functions?: Function[];