File tree Expand file tree Collapse file tree 2 files changed +4
-3
lines changed
packages/vertexai/src/types Expand file tree Collapse file tree 2 files changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -38,12 +38,12 @@ enum Availability {
38
38
'downloading' ,
39
39
'available'
40
40
}
41
- interface LanguageModelCreateCoreOptions {
41
+ export interface LanguageModelCreateCoreOptions {
42
42
topK ?: number ;
43
43
temperature ?: number ;
44
44
expectedInputs ?: LanguageModelExpectedInput [ ] ;
45
45
}
46
- interface LanguageModelCreateOptions extends LanguageModelCreateCoreOptions {
46
+ export interface LanguageModelCreateOptions extends LanguageModelCreateCoreOptions {
47
47
signal ?: AbortSignal ;
48
48
systemPrompt ?: string ;
49
49
initialPrompts ?: LanguageModelInitialPrompts ;
Original file line number Diff line number Diff line change @@ -24,6 +24,7 @@ import {
24
24
HarmCategory
25
25
} from './enums' ;
26
26
import { ObjectSchemaInterface , SchemaRequest } from './schema' ;
27
+ import { LanguageModelCreateOptions } from '../types/language-model' ;
27
28
28
29
/**
29
30
* Base parameters for a number of methods.
@@ -225,7 +226,7 @@ export interface HybridParams {
225
226
/**
226
227
* Optional. Specifies advanced params for on-device inference.
227
228
*/
228
- onDeviceParams ?: AILanguageModelCreateOptionsWithSystemPrompt ;
229
+ onDeviceParams ?: LanguageModelCreateOptions ;
229
230
/**
230
231
* Optional. Specifies advanced params for in-cloud inference.
231
232
*/
You can’t perform that action at this time.
0 commit comments