diff --git a/semantic_router/schema.py b/semantic_router/schema.py index 85d428ef0494a882cd84b8c9ea8caf7c914de517..035ca8a00b77ca7b3681d833f5e18c19bff0054b 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