From f3145412f3a79faa15361bf934097582a63b24e7 Mon Sep 17 00:00:00 2001
From: Stephen Witkowski <stephen.witkowski@66degrees.com>
Date: Mon, 1 Apr 2024 09:27:09 -0400
Subject: [PATCH] Add GoogleEncoder to EncoderType enum and Encoder class
---
semantic_router/schema.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/semantic_router/schema.py b/semantic_router/schema.py
index 85d428ef..035ca8a0 100644
--- a/semantic_router/schema.py
+++ b/semantic_router/schema.py
@@ -10,6 +10,7 @@ from semantic_router.encoders import (
FastEmbedEncoder,
MistralEncoder,
OpenAIEncoder,
+ GoogleEncoder,
)
@@ -19,6 +20,7 @@ class EncoderType(Enum):
OPENAI = "openai"
COHERE = "cohere"
MISTRAL = "mistral"
+ GOOGLE = "google"
class RouteChoice(BaseModel):
@@ -46,6 +48,8 @@ class Encoder:
self.model = CohereEncoder(name=name)
elif self.type == EncoderType.MISTRAL:
self.model = MistralEncoder(name=name)
+ elif self.type == EncoderType.GOOGLE:
+ self.model = GoogleEncoder(name=name)
else:
raise ValueError
--
GitLab