Adding LanguageModel types.

Author: gsiddh

packages/vertexai/src/types/index.ts

@@ -17,6 +17,7 @@
 
 export * from './content';
 export * from './enums';
+export * from './language-model';
 export * from './requests';
 export * from './responses';
 export * from './error';

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

@@ -16,7 +16,7 @@ export interface LanguageModelParams {
 export interface LanguageModelCreateCoreOptions {
     topK?: number;
     temperature?: number;
-    expectedInputs?: Array<LanguageModelExpectedInput>;
+    expectedInputs?: LanguageModelExpectedInput[];
 }
 export interface LanguageModelCreateOptions extends LanguageModelCreateCoreOptions {
     signal?: AbortSignal;
@@ -25,18 +25,17 @@ export interface LanguageModelCreateOptions extends LanguageModelCreateCoreOptio
     initialPrompts?: LanguageModelInitialPrompts;
 }
 export interface LanguageModelPromptOptions {
-    responseJSONSchema?: any;
     signal?: AbortSignal;
 }
 export interface LanguageModelExpectedInput {
     type: LanguageModelMessageType;
-    languages?: Array<string>;
+    languages?: string[];
 }
-export type LanguageModelPrompt = Array<LanguageModelMessage> | Array<LanguageModelMessageShorthand> | string;
-export type LanguageModelInitialPrompts = Array<LanguageModelMessage> | Array<LanguageModelMessageShorthand>;
+export type LanguageModelPrompt = LanguageModelMessage[] | LanguageModelMessageShorthand[] | string;
+export type LanguageModelInitialPrompts = LanguageModelMessage[] | LanguageModelMessageShorthand[];
 export interface LanguageModelMessage {
     role: LanguageModelMessageRole;
-    content: Array<LanguageModelMessageContent>;
+    content: LanguageModelMessageContent[];
 }
 export interface LanguageModelMessageShorthand {
     role: LanguageModelMessageRole;