From 20133d3615dd872e79ff9ee0b876b69b01e7e7a9 Mon Sep 17 00:00:00 2001
From: Siraj R Aizlewood <siraj@aurelio.ai>
Date: Tue, 6 Feb 2024 23:43:23 +0400
Subject: [PATCH] Linting.
---
semantic_router/indices/base.py | 3 ++-
semantic_router/indices/local_index.py | 12 ++++++------
semantic_router/layer.py | 2 +-
semantic_router/schema.py | 2 +-
4 files changed, 10 insertions(+), 9 deletions(-)
diff --git a/semantic_router/indices/base.py b/semantic_router/indices/base.py
index 7b48f65d..baa5a204 100644
--- a/semantic_router/indices/base.py
+++ b/semantic_router/indices/base.py
@@ -1,5 +1,6 @@
from pydantic.v1 import BaseModel
+
class BaseIndex(BaseModel):
# Currently just a placedholder until more indexing methods are added and common attributes/methods are identified.
- pass
\ No newline at end of file
+ pass
diff --git a/semantic_router/indices/local_index.py b/semantic_router/indices/local_index.py
index eb5d63ef..3c0d32dc 100644
--- a/semantic_router/indices/local_index.py
+++ b/semantic_router/indices/local_index.py
@@ -1,14 +1,13 @@
import numpy as np
-from typing import List, Any
+from typing import List, Any, Tuple, Optional
from semantic_router.linear import similarity_matrix, top_scores
from semantic_router.indices.base import BaseIndex
-import numpy as np
-from typing import List, Any, Tuple, Optional
+
class LocalIndex(BaseIndex):
index: Optional[np.ndarray] = None
- class Config: # Stop pydantic from complaining about Optional[np.ndarray] type hints.
+ class Config: # Stop pydantic from complaining about Optional[np.ndarray] type hints.
arbitrary_types_allowed = True
def add(self, embeds: List[Any]):
@@ -27,7 +26,9 @@ class LocalIndex(BaseIndex):
def is_index_populated(self):
return self.index is not None and len(self.index) > 0
- def search(self, query_vector: Any, top_k: int = 5) -> Tuple[np.ndarray, np.ndarray]:
+ def search(
+ self, query_vector: Any, top_k: int = 5
+ ) -> Tuple[np.ndarray, np.ndarray]:
"""
Search the index for the query and return top_k results.
"""
@@ -35,4 +36,3 @@ class LocalIndex(BaseIndex):
raise ValueError("Index is not populated.")
sim = similarity_matrix(query_vector, self.index)
return top_scores(sim, top_k)
-
diff --git a/semantic_router/layer.py b/semantic_router/layer.py
index 8b7c485d..3061f29e 100644
--- a/semantic_router/layer.py
+++ b/semantic_router/layer.py
@@ -8,7 +8,6 @@ import yaml
from tqdm.auto import tqdm
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
from semantic_router.schema import Encoder, EncoderType, RouteChoice, Index
@@ -17,6 +16,7 @@ from semantic_router.indices.local_index import LocalIndex
IndexType = Union[LocalIndex, None]
+
def is_valid(layer_config: str) -> bool:
"""Make sure the given string is json format and contains the 3 keys: ["encoder_name", "encoder_type", "routes"]"""
try:
diff --git a/semantic_router/schema.py b/semantic_router/schema.py
index 917055cb..95e29bfe 100644
--- a/semantic_router/schema.py
+++ b/semantic_router/schema.py
@@ -84,4 +84,4 @@ class Index:
return LocalIndex()
# TODO: Later we'll add more index options.
else:
- raise ValueError(f"Invalid index name: {index_name}")
\ No newline at end of file
+ raise ValueError(f"Invalid index name: {index_name}")
--
GitLab