From b83bfc6a8ff23c008d4cd431bb0565dc42747fc0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E2=80=9CDaniel=20Griffiths=E2=80=9D?=
 <Danielgriffiths1790@gmail.com>
Date: Wed, 3 Jan 2024 15:51:07 +0000
Subject: [PATCH] refactor: seperate schema files removed

---
 semantic_router/schemas/encoder.py        | 42 -----------------------
 semantic_router/schemas/route.py          |  7 ----
 semantic_router/schemas/semantic_space.py | 17 ---------
 3 files changed, 66 deletions(-)
 delete mode 100644 semantic_router/schemas/encoder.py
 delete mode 100644 semantic_router/schemas/route.py
 delete mode 100644 semantic_router/schemas/semantic_space.py

diff --git a/semantic_router/schemas/encoder.py b/semantic_router/schemas/encoder.py
deleted file mode 100644
index fbbfb2d7..00000000
--- a/semantic_router/schemas/encoder.py
+++ /dev/null
@@ -1,42 +0,0 @@
-from enum import Enum
-
-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)
diff --git a/semantic_router/schemas/route.py b/semantic_router/schemas/route.py
deleted file mode 100644
index b70bc60f..00000000
--- a/semantic_router/schemas/route.py
+++ /dev/null
@@ -1,7 +0,0 @@
-from pydantic import BaseModel
-
-
-class Route(BaseModel):
-    name: str
-    utterances: list[str]
-    description: str | None = None
diff --git a/semantic_router/schemas/semantic_space.py b/semantic_router/schemas/semantic_space.py
deleted file mode 100644
index 92e7adaf..00000000
--- a/semantic_router/schemas/semantic_space.py
+++ /dev/null
@@ -1,17 +0,0 @@
-from pydantic.dataclasses import dataclass
-
-from semantic_router.schemas.route import Route
-
-
-@dataclass
-class SemanticSpace:
-    id: str
-    routes: list[Route]
-    encoder: str = ""
-
-    def __init__(self, routes: list[Route] = []):
-        self.id = ""
-        self.routes = routes
-
-    def add(self, route: Route):
-        self.routes.append(route)
-- 
GitLab