Skip to content

Commit 6449013

Browse files
committed
anthropic vertex ai support
Signed-off-by: omar <[email protected]>
1 parent e837e48 commit 6449013

File tree

2 files changed

+12
-7
lines changed

2 files changed

+12
-7
lines changed

api/v1alpha1/ai_backend.go

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -254,10 +254,13 @@ type GeminiConfig struct {
254254
ApiVersion string `json:"apiVersion"`
255255
}
256256

257-
// Publisher configures the type of publisher model to use for VertexAI. Currently, only Google is supported.
257+
// Publisher configures the type of publisher model to use for VertexAI. Google and Anthropic are supported.
258258
type Publisher string
259259

260-
const GOOGLE Publisher = "GOOGLE"
260+
const (
261+
GOOGLE Publisher = "GOOGLE"
262+
ANTHROPIC Publisher = "ANTHROPIC"
263+
)
261264

262265
// VertexAIConfig settings for the [Vertex AI](https://cloud.google.com/vertex-ai/docs) LLM provider.
263266
// To find the values for the project ID, project location, and publisher, you can check the fields of an API request, such as
@@ -293,8 +296,8 @@ type VertexAIConfig struct {
293296
// Optional: The model path to route to. Defaults to the Gemini model path, `generateContent`.
294297
ModelPath *string `json:"modelPath,omitempty"`
295298

296-
// The type of publisher model to use. Currently, only Google is supported.
297-
// +kubebuilder:validation:Enum=GOOGLE
299+
// The type of publisher model to use. Google and Anthropic are supported.
300+
// +kubebuilder:validation:Enum=GOOGLE;ANTHROPIC
298301
Publisher Publisher `json:"publisher"`
299302
}
300303

install/helm/kgateway-crds/templates/gateway.kgateway.dev_backends.yaml

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -542,10 +542,11 @@ spec:
542542
minLength: 1
543543
type: string
544544
publisher:
545-
description: The type of publisher model to use. Currently,
546-
only Google is supported.
545+
description: The type of publisher model to use. Google
546+
and Anthropic are supported.
547547
enum:
548548
- GOOGLE
549+
- ANTHROPIC
549550
type: string
550551
required:
551552
- apiVersion
@@ -1088,9 +1089,10 @@ spec:
10881089
type: string
10891090
publisher:
10901091
description: The type of publisher model to use.
1091-
Currently, only Google is supported.
1092+
Google and Anthropic are supported.
10921093
enum:
10931094
- GOOGLE
1095+
- ANTHROPIC
10941096
type: string
10951097
required:
10961098
- apiVersion

0 commit comments

Comments
 (0)