Skip to content

Commit 5302797

Browse files
committed
Took dependency on new type
1 parent b07bdac commit 5302797

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

packages/vertexai/src/types/language-model.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,12 +38,12 @@ enum Availability {
3838
'downloading',
3939
'available'
4040
}
41-
interface LanguageModelCreateCoreOptions {
41+
export interface LanguageModelCreateCoreOptions {
4242
topK?: number;
4343
temperature?: number;
4444
expectedInputs?: LanguageModelExpectedInput[];
4545
}
46-
interface LanguageModelCreateOptions extends LanguageModelCreateCoreOptions {
46+
export interface LanguageModelCreateOptions extends LanguageModelCreateCoreOptions {
4747
signal?: AbortSignal;
4848
systemPrompt?: string;
4949
initialPrompts?: LanguageModelInitialPrompts;

packages/vertexai/src/types/requests.ts

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ import {
2424
HarmCategory
2525
} from './enums';
2626
import { ObjectSchemaInterface, SchemaRequest } from './schema';
27+
import {LanguageModelCreateOptions} from '../types/language-model';
2728

2829
/**
2930
* Base parameters for a number of methods.
@@ -225,7 +226,7 @@ export interface HybridParams {
225226
/**
226227
* Optional. Specifies advanced params for on-device inference.
227228
*/
228-
onDeviceParams?: AILanguageModelCreateOptionsWithSystemPrompt;
229+
onDeviceParams?: LanguageModelCreateOptions;
229230
/**
230231
* Optional. Specifies advanced params for in-cloud inference.
231232
*/

0 commit comments

Comments
 (0)