chatgpt-api/test/anthropic.test.ts

36 wiersze
866 B
TypeScript

import test from 'ava'
import { expectTypeOf } from 'expect-type'
import { AnthropicChatModel } from '@/llms/anthropic'
import { createTestAgenticRuntime } from './_utils'
test('AnthropicChatModel ⇒ string output', async (t) => {
t.timeout(2 * 60 * 1000)
const agentic = createTestAgenticRuntime()
const builder = new AnthropicChatModel({
agentic,
modelParams: {
temperature: 0,
max_tokens_to_sample: 30
},
messages: [
{
role: 'system',
content:
'You are a human, not a large language model. You are not an AI. Respond like a human would. Your name is Leeroy Jenkins.'
},
{
role: 'user',
content: 'what is your name?'
}
]
})
const result = await builder.call()
t.truthy(typeof result === 'string')
expectTypeOf(result).toMatchTypeOf<string>()
})