From fcc84b61e4f58ec490b7fdfc6aab050457e7181e Mon Sep 17 00:00:00 2001 From: Travis Fischer Date: Wed, 15 Mar 2023 18:16:29 -0500 Subject: [PATCH] feat: add gpt-4 demo --- demos/demo-gpt-4.ts | 35 +++++++++++++++++++++++++++++++++++ 1 file changed, 35 insertions(+) create mode 100644 demos/demo-gpt-4.ts diff --git a/demos/demo-gpt-4.ts b/demos/demo-gpt-4.ts new file mode 100644 index 0000000..5f31251 --- /dev/null +++ b/demos/demo-gpt-4.ts @@ -0,0 +1,35 @@ +import dotenv from 'dotenv-safe' +import { oraPromise } from 'ora' + +import { ChatGPTAPI } from '../src' + +dotenv.config() + +/** + * Demo CLI for testing the GPT-4 model. + * + * ``` + * npx tsx demos/demo-gpt-4.ts + * ``` + */ +async function main() { + const api = new ChatGPTAPI({ + apiKey: process.env.OPENAI_API_KEY, + debug: true, + completionParams: { + model: 'gpt-4' + } + }) + + const prompt = 'When should you use Python vs TypeScript?' + + const res = await oraPromise(api.sendMessage(prompt), { + text: prompt + }) + console.log(res.text) +} + +main().catch((err) => { + console.error(err) + process.exit(1) +})