From 516e226290429c9192ed3ef3d53a25ae51aaa482 Mon Sep 17 00:00:00 2001 From: hiro Date: Tue, 5 Dec 2023 00:29:53 +0700 Subject: [PATCH] chore: Move interface definition location --- .../src/@types/global.d.ts | 27 ++++++++++++++++++- 1 file changed, 26 insertions(+), 1 deletion(-) diff --git a/extensions/inference-openai-extension/src/@types/global.d.ts b/extensions/inference-openai-extension/src/@types/global.d.ts index 7267f0940..988c3c7db 100644 --- a/extensions/inference-openai-extension/src/@types/global.d.ts +++ b/extensions/inference-openai-extension/src/@types/global.d.ts @@ -1,2 +1,27 @@ +import { Model } from "@janhq/core"; + declare const MODULE: string; -declare const INFERENCE_URL: string; + +declare interface EngineSettings { + base_url?: string; + api_key?: string; +} + +enum OpenAIChatCompletionModelName { + 'gpt-3.5-turbo-instruct' = 'gpt-3.5-turbo-instruct', + 'gpt-3.5-turbo-instruct-0914' = 'gpt-3.5-turbo-instruct-0914', + 'gpt-4-1106-preview' = 'gpt-4-1106-preview', + 'gpt-3.5-turbo-0613' = 'gpt-3.5-turbo-0613', + 'gpt-3.5-turbo-0301' = 'gpt-3.5-turbo-0301', + 'gpt-3.5-turbo' = 'gpt-3.5-turbo', + 'gpt-3.5-turbo-16k-0613' = 'gpt-3.5-turbo-16k-0613', + 'gpt-3.5-turbo-1106' = 'gpt-3.5-turbo-1106', + 'gpt-4-vision-preview' = 'gpt-4-vision-preview', + 'gpt-4' = 'gpt-4', + 'gpt-4-0314' = 'gpt-4-0314', + 'gpt-4-0613' = 'gpt-4-0613', +} + +declare type OpenAIModel = Omit & { + id: OpenAIChatCompletionModelName; +}; \ No newline at end of file