kopia lustrzana https://github.com/transitive-bullshit/chatgpt-api
73 wiersze
1.3 KiB
TypeScript
73 wiersze
1.3 KiB
TypeScript
import dotenv from 'dotenv-safe'
|
|
import { oraPromise } from 'ora'
|
|
|
|
import { ChatGPTAPI } from '../src'
|
|
|
|
dotenv.config()
|
|
|
|
/**
|
|
* Demo CLI for testing conversation support.
|
|
*
|
|
* ```
|
|
* npx tsx demos/demo-conversation.ts
|
|
* ```
|
|
*/
|
|
async function main() {
|
|
const api = new ChatGPTAPI({
|
|
apiKey: process.env.OPENAI_API_KEY,
|
|
debug: false
|
|
})
|
|
|
|
const prompt = 'Write a poem about cats.'
|
|
|
|
let res = await oraPromise(api.sendMessage(prompt), {
|
|
text: prompt
|
|
})
|
|
|
|
console.log('\n' + res.text + '\n')
|
|
|
|
const prompt2 = 'Can you make it cuter and shorter?'
|
|
|
|
res = await oraPromise(
|
|
api.sendMessage(prompt2, {
|
|
conversationId: res.conversationId,
|
|
parentMessageId: res.id
|
|
}),
|
|
{
|
|
text: prompt2
|
|
}
|
|
)
|
|
console.log('\n' + res.text + '\n')
|
|
|
|
const prompt3 = 'Now write it in French.'
|
|
|
|
res = await oraPromise(
|
|
api.sendMessage(prompt3, {
|
|
conversationId: res.conversationId,
|
|
parentMessageId: res.id
|
|
}),
|
|
{
|
|
text: prompt3
|
|
}
|
|
)
|
|
console.log('\n' + res.text + '\n')
|
|
|
|
const prompt4 = 'What were we talking about again?'
|
|
|
|
res = await oraPromise(
|
|
api.sendMessage(prompt4, {
|
|
conversationId: res.conversationId,
|
|
parentMessageId: res.id
|
|
}),
|
|
{
|
|
text: prompt4
|
|
}
|
|
)
|
|
console.log('\n' + res.text + '\n')
|
|
}
|
|
|
|
main().catch((err) => {
|
|
console.error(err)
|
|
process.exit(1)
|
|
})
|