Update code model defaults for code-gecko in GoogleVertexAI (langchain-ai#2153)

Yongtae723 · web-flow · commit 091c7f05344e · 2023-08-03T01:20:42.000-07:00
diff --git a/langchain/src/llms/googlevertexai.ts b/langchain/src/llms/googlevertexai.ts
@@ -67,7 +67,7 @@ export class GoogleVertexAI extends BaseLLM implements GoogleVertexAITextInput {
     this.model = fields?.model ?? this.model;
 
     // Change the defaults for code models
-    if (this.model === "code-gecko") {
+    if (this.model.startsWith("code-gecko")) {
       this.maxOutputTokens = 64;
     }
     if (this.model.startsWith("code-")) {

Original file line number	Diff line number	Diff line change
`@@ -67,7 +67,7 @@ export class GoogleVertexAI extends BaseLLM implements GoogleVertexAITextInput {`
`67`	`67`	`this.model = fields?.model ?? this.model;`
`68`	`68`
`69`	`69`	`// Change the defaults for code models`
`70`		`- if (this.model === "code-gecko") {`
	`70`	`+ if (this.model.startsWith("code-gecko")) {`
`71`	`71`	`this.maxOutputTokens = 64;`
`72`	`72`	`}`
`73`	`73`	`if (this.model.startsWith("code-")) {`