2
0

Add OpenRouter block

Closes #1254
This commit is contained in:
Baptiste Arnaud
2024-03-05 10:56:54 +01:00
parent 77bc138c38
commit 84d6c594af
14 changed files with 173 additions and 3 deletions

View File

@ -0,0 +1,50 @@
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 },
}),
},
},
})