Skip to content
Snippets Groups Projects
Unverified Commit f67b051c authored by James Briggs's avatar James Briggs
Browse files

add type definition for self.encoder

parent 2adb617a
No related branches found
No related tags found
No related merge requests found
......@@ -4,10 +4,7 @@ import os
import numpy as np
import yaml
from semantic_router.encoders import (
BaseEncoder,
OpenAIEncoder
)
from semantic_router.encoders import BaseEncoder, OpenAIEncoder
from semantic_router.linear import similarity_matrix, top_scores
from semantic_router.llms import BaseLLM, OpenAILLM
from semantic_router.route import Route
......@@ -153,6 +150,7 @@ class RouteLayer:
index: np.ndarray | None = None
categories: np.ndarray | None = None
score_threshold: float
encoder: BaseEncoder
def __init__(
self,
......
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