dtamind-components/nodes/chatmodels/ChatGoogleGenerativeAI/utils/types.ts

DTAmindai Components

import {
    CodeExecutionTool,
    FunctionDeclarationsTool as GoogleGenerativeAIFunctionDeclarationsTool,
    GoogleSearchRetrievalTool
} from '@google/generative-ai'
import { BindToolsInput } from '@langchain/core/language_models/chat_models'

export type GoogleGenerativeAIToolType =
    | BindToolsInput
    | GoogleGenerativeAIFunctionDeclarationsTool
    | CodeExecutionTool
    | GoogleSearchRetrievalTool