Class GoogleModel

Implements the Gemini language model integration for Theia

Implements

LanguageModel

Index

Constructors

Properties

apiKey enableStreaming id model retrySettings tokenUsageService?

Methods

createFunctionDeclarations formatToolCallResult getSettings handleNonStreamingRequest handleStreamingRequest initializeGemini request withRetry

Constructors

constructor

new GoogleModel(id, model, enableStreaming, apiKey, retrySettings, tokenUsageService?): GoogleModel
Parameters
- id: string
- model: string
- enableStreaming: boolean
- apiKey: (() => undefined | string)
  - - (): undefined | string
    - Returns undefined | string
- retrySettings: (() => GoogleLanguageModelRetrySettings)
  - - (): GoogleLanguageModelRetrySettings
    - Returns GoogleLanguageModelRetrySettings
- Optional tokenUsageService: TokenUsageService
Returns GoogleModel
- Defined in packages/ai-google/src/node/google-language-model.ts:124

Properties

apiKey

apiKey: (() => undefined | string)

Type declaration

- (): undefined | string
- Returns undefined | string

enableStreaming

enableStreaming: boolean

`Readonly` id

id: string

model

model: string

retrySettings

retrySettings: (() => GoogleLanguageModelRetrySettings)

Type declaration

- (): GoogleLanguageModelRetrySettings
- Returns GoogleLanguageModelRetrySettings

`Protected` `Optional` `Readonly` tokenUsageService

tokenUsageService?: TokenUsageService

Methods

`Private` createFunctionDeclarations

createFunctionDeclarations(request): FunctionDeclaration[]
Parameters
- request: LanguageModelRequest
Returns FunctionDeclaration[]
- Defined in packages/ai-google/src/node/google-language-model.ts:337

`Protected` formatToolCallResult

formatToolCallResult(result): undefined | Record<string, unknown>
Parameters
- result: ToolCallResult
Returns undefined | Record<string, unknown>
- Defined in packages/ai-google/src/node/google-language-model.ts:330

`Protected` getSettings

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

`Protected` handleNonStreamingRequest

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

`Protected` handleStreamingRequest

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

`Protected` initializeGemini

initializeGemini(): GoogleGenAI
Returns GoogleGenAI
- Defined in packages/ai-google/src/node/google-language-model.ts:395

request

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

`Private` withRetry

withRetry<T>(fn): Promise<T>
Implements a retry mechanism for the handle(non)Streaming request functions.
Type Parameters
- T
Parameters
- fn: (() => Promise<T>)
  the wrapped function to which the retry logic should be applied.
  - - (): Promise<T>
    - Returns Promise<T>
Returns Promise<T>
the result of the wrapped function.
- Defined in packages/ai-google/src/node/google-language-model.ts:411