add OpenAIBaseInput interface

This commit is contained in:
chungyau97 2023-05-10 20:30:44 +07:00
parent 6ca0c364b9
commit e1667067be
1 changed files with 2 additions and 1 deletions

View File

@ -1,3 +1,4 @@
import { OpenAIBaseInput } from 'langchain/dist/types/openai-types'
import { INode, INodeData, INodeParams } from '../../../src/Interface'
import { getBaseClasses } from '../../../src/utils'
import { AzureOpenAIInput, ChatOpenAI } from 'langchain/chat_models/openai'
@ -79,7 +80,7 @@ class AzureChatOpenAI_ChatModels implements INode {
const azureOpenAIApiEmbeddingsDeploymentName = nodeData.inputs?.azureOpenAIApiEmbeddingsDeploymentName as string
const azureOpenAIApiCompletionsDeploymentName = nodeData.inputs?.azureOpenAIApiCompletionsDeploymentName as string
const obj: Partial<AzureOpenAIInput> & { temperature?: number } = {
const obj: Partial<AzureOpenAIInput> & Partial<OpenAIBaseInput> = {
temperature: parseInt(temperature, 10),
azureOpenAIApiKey,
azureOpenAIApiInstanceName,