Skip to content
Snippets Groups Projects
Commit b87a4912 authored by “Daniel Griffiths”'s avatar “Daniel Griffiths”
Browse files

fix: schema file restored

parent ec77a1a5
No related branches found
No related tags found
No related merge requests found
from enum import Enum
from pydantic import BaseModel
from pydantic.dataclasses import dataclass
from semantic_router.encoders import (
BaseEncoder,
CohereEncoder,
OpenAIEncoder,
)
class EncoderType(Enum):
HUGGINGFACE = "huggingface"
OPENAI = "openai"
COHERE = "cohere"
class RouteChoice(BaseModel):
name: str | None = None
function_call: dict | None = None
@dataclass
class Encoder:
type: EncoderType
name: str | None
model: BaseEncoder
def __init__(self, type: str, name: str | None):
self.type = EncoderType(type)
self.name = name
if self.type == EncoderType.HUGGINGFACE:
raise NotImplementedError
elif self.type == EncoderType.OPENAI:
self.model = OpenAIEncoder(name)
elif self.type == EncoderType.COHERE:
self.model = CohereEncoder(name)
else:
raise ValueError
def __call__(self, texts: list[str]) -> list[list[float]]:
return self.model(texts)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment