Skip to content
Snippets Groups Projects
Commit f3145412 authored by Stephen Witkowski's avatar Stephen Witkowski
Browse files

Add GoogleEncoder to EncoderType enum and Encoder class

parent 83b05bb5
Branches
Tags
No related merge requests found
...@@ -10,6 +10,7 @@ from semantic_router.encoders import ( ...@@ -10,6 +10,7 @@ from semantic_router.encoders import (
FastEmbedEncoder, FastEmbedEncoder,
MistralEncoder, MistralEncoder,
OpenAIEncoder, OpenAIEncoder,
GoogleEncoder,
) )
...@@ -19,6 +20,7 @@ class EncoderType(Enum): ...@@ -19,6 +20,7 @@ class EncoderType(Enum):
OPENAI = "openai" OPENAI = "openai"
COHERE = "cohere" COHERE = "cohere"
MISTRAL = "mistral" MISTRAL = "mistral"
GOOGLE = "google"
class RouteChoice(BaseModel): class RouteChoice(BaseModel):
...@@ -46,6 +48,8 @@ class Encoder: ...@@ -46,6 +48,8 @@ class Encoder:
self.model = CohereEncoder(name=name) self.model = CohereEncoder(name=name)
elif self.type == EncoderType.MISTRAL: elif self.type == EncoderType.MISTRAL:
self.model = MistralEncoder(name=name) self.model = MistralEncoder(name=name)
elif self.type == EncoderType.GOOGLE:
self.model = GoogleEncoder(name=name)
else: else:
raise ValueError raise ValueError
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment