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