Skip to content

Commit a3d869b

Browse files
committed
Define HybridParams
1 parent 02c4ba7 commit a3d869b

File tree

1 file changed

+21
-0
lines changed
  • packages/vertexai/src/types

1 file changed

+21
-0
lines changed

packages/vertexai/src/types/ai.ts

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,21 @@
1+
/**
2+
* Shims @types/dom-chromium-ai
3+
* TODO: replace with @types/dom-chromium-ai once we can use es2020.intl.
4+
*/
5+
interface AILanguageModelCreateOptions {
6+
topK?: number;
7+
temperature?: number;
8+
}
9+
10+
export interface AILanguageModelCreateOptionsWithSystemPrompt
11+
extends AILanguageModelCreateOptions {
12+
systemPrompt?: string;
13+
initialPrompts?: AILanguageModelPrompt[];
14+
}
15+
16+
type AILanguageModelPromptRole = 'user' | 'assistant';
17+
18+
interface AILanguageModelPrompt {
19+
role: AILanguageModelPromptRole;
20+
content: string;
21+
}

0 commit comments

Comments
 (0)