Class GoogleModel

Implements the Gemini language model integration for Theia

Implements

LanguageModel

Index

Constructors

Properties

apiKey enableStreaming id model retrySettings status tokenUsageService?

Methods

getSettings handleNonStreamingRequest handleStreamingRequest initializeGemini request

Constructors

constructor

new GoogleModel(
    id: string,
    model: string,
    status: LanguageModelStatus,
    enableStreaming: boolean,
    apiKey: () => undefined | string,
    retrySettings: () => GoogleLanguageModelRetrySettings,
    tokenUsageService?: TokenUsageService,
): GoogleModel
Parameters
- id: string
- model: string
- status: LanguageModelStatus
- enableStreaming: boolean
- apiKey: () => undefined | string
- retrySettings: () => GoogleLanguageModelRetrySettings
- OptionaltokenUsageService: TokenUsageService
Returns GoogleModel
- Defined in node/google-language-model.ts:141

Properties

apiKey

apiKey: () => undefined | string

enableStreaming

enableStreaming: boolean

`Readonly`id

id: string

model

model: string

retrySettings

retrySettings: () => GoogleLanguageModelRetrySettings

status

status: LanguageModelStatus

`Protected` `Optional` `Readonly`tokenUsageService

tokenUsageService?: TokenUsageService

Methods

`Protected`getSettings

getSettings(request: LanguageModelRequest): Readonly<Record<string, unknown>>
Parameters
- request: LanguageModelRequest
Returns Readonly<Record<string, unknown>>
- Defined in node/google-language-model.ts:151

`Protected`handleNonStreamingRequest

handleNonStreamingRequest(
genAI: GoogleGenAI,
request: UserRequest,
): Promise<LanguageModelTextResponse>
Parameters
- genAI: GoogleGenAI
- request: UserRequest
Returns Promise<LanguageModelTextResponse>
- Defined in node/google-language-model.ts:408

`Protected`handleStreamingRequest

handleStreamingRequest(
    genAI: GoogleGenAI,
    request: UserRequest,
    cancellationToken?: CancellationToken,
    toolMessages?: Content[],
): Promise<LanguageModelStreamResponse>
Parameters
- genAI: GoogleGenAI
- request: UserRequest
- OptionalcancellationToken: CancellationToken
- OptionaltoolMessages: Content[]
Returns Promise<LanguageModelStreamResponse>
- Defined in node/google-language-model.ts:172

`Protected`initializeGemini

initializeGemini(): GoogleGenAI
Returns GoogleGenAI
- Defined in node/google-language-model.ts:465

request

request(
request: UserRequest,
cancellationToken?: CancellationToken,
): Promise<LanguageModelResponse>
Parameters
- request: UserRequest
- OptionalcancellationToken: CancellationToken
Returns Promise<LanguageModelResponse>
Implementation of LanguageModel.request
- Defined in node/google-language-model.ts:155