Skip to content
Open
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions src/collections/config/types/generative.ts
Original file line number Diff line number Diff line change
Expand Up @@ -88,6 +88,7 @@ export type GenerativePaLMConfig = GenerativeGoogleConfig;

export type GenerativeGoogleConfig = {
apiEndpoint?: string;
location?: string;
maxOutputTokens?: number;
model?: string;
/** @deprecated Use `model` instead. */
Expand Down
2 changes: 2 additions & 0 deletions src/collections/configure/unit.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -2226,6 +2226,7 @@ describe('Unit testing of the generative factory class', () => {
it('should create the correct GenerativeGoogleConfig type with all values', () => {
const config = configure.generative.google({
apiEndpoint: 'api-endpoint',
location: 'europe-west4',
maxOutputTokens: 100,
model: 'model-id',
projectId: 'project-id',
Expand All @@ -2237,6 +2238,7 @@ describe('Unit testing of the generative factory class', () => {
name: 'generative-google',
config: {
apiEndpoint: 'api-endpoint',
location: 'europe-west4',
maxOutputTokens: 100,
model: 'model-id',
modelId: 'model-id',
Expand Down
1 change: 1 addition & 0 deletions src/collections/types/generate.ts
Original file line number Diff line number Diff line change
Expand Up @@ -300,6 +300,7 @@ export type GenerativeGoogleConfigRuntime = {
projectId?: string | undefined;
endpointId?: string | undefined;
region?: string | undefined;
location?: string | undefined;
};

export type GenerativeMistralConfigRuntime = {
Expand Down