51 lines
1.6 KiB
TypeScript
51 lines
1.6 KiB
TypeScript
|
import { createAction } from '@typebot.io/forge'
|
||
|
|
import { auth } from '../auth'
|
||
|
|
import { parseChatCompletionOptions } from '@typebot.io/openai-block/shared/parseChatCompletionOptions'
|
||
|
|
import { getChatCompletionSetVarIds } from '@typebot.io/openai-block/shared/getChatCompletionSetVarIds'
|
||
|
|
import { getChatCompletionStreamVarId } from '@typebot.io/openai-block/shared/getChatCompletionStreamVarId'
|
||
|
|
import { runChatCompletion } from '@typebot.io/openai-block/shared/runChatCompletion'
|
||
|
|
import { runChatCompletionStream } from '@typebot.io/openai-block/shared/runChatCompletionStream'
|
||
|
|
import { defaultOpenRouterOptions } from '../constants'
|
||
|
|
import { got } from 'got'
|
||
|
|
import { ModelsResponse } from '../types'
|
||
|
|
|
||
|
|
export const createChatCompletion = createAction({
|
||
|
|
name: 'Create chat completion',
|
||
|
|
auth,
|
||
|
|
options: parseChatCompletionOptions({
|
||
|
|
modelFetchId: 'fetchModels',
|
||
|
|
}),
|
||
|
|
getSetVariableIds: getChatCompletionSetVarIds,
|
||
|
|
fetchers: [
|
||
|
|
{
|
||
|
|
id: 'fetchModels',
|
||
|
|
dependencies: [],
|
||
|
|
fetch: async () => {
|
||
|
|
const response = await got
|
||
|
|
.get(defaultOpenRouterOptions.baseUrl + '/models')
|
||
|
|
.json<ModelsResponse>()
|
||
|
|
|
||
|
|
return response.data.map((model) => ({
|
||
|
|
value: model.id,
|
||
|
|
label: model.name,
|
||
|
|
}))
|
||
|
|
},
|
||
|
|
},
|
||
|
|
],
|
||
|
|
run: {
|
||
|
|
server: (params) =>
|
||
|
|
runChatCompletion({
|
||
|
|
...params,
|
||
|
|
config: { baseUrl: defaultOpenRouterOptions.baseUrl },
|
||
|
|
}),
|
||
|
|
stream: {
|
||
|
|
getStreamVariableId: getChatCompletionStreamVarId,
|
||
|
|
run: (params) =>
|
||
|
|
runChatCompletionStream({
|
||
|
|
...params,
|
||
|
|
config: { baseUrl: defaultOpenRouterOptions.baseUrl },
|
||
|
|
}),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
})
|