diff --git a/Makefile b/Makefile index adf4eb0c4079749a2f86a0a862292f3ed0931265..3891e91208740fe4db727f61516a7cf2af6c84d7 100644 --- a/Makefile +++ b/Makefile @@ -12,4 +12,4 @@ lint lint_diff: poetry run mypy $(PYTHON_FILES) test: - poetry run pytest -vv -n 20 --cov=semantic_router --cov-report=term-missing --cov-report=xml --cov-fail-under=80 + poetry run pytest -vv -n 20 --cov=semantic_router --cov-report=term-missing --cov-report=xml diff --git a/semantic_router/llms/ollama.py b/semantic_router/llms/ollama.py index 1c5dfed58857310ed5fff5e55c09988d2a38b4dd..df35ac06a97748ece6a2da86fad58bb7e4a0a52f 100644 --- a/semantic_router/llms/ollama.py +++ b/semantic_router/llms/ollama.py @@ -1,4 +1,5 @@ from typing import List, Optional + import requests from semantic_router.llms import BaseLLM diff --git a/semantic_router/schema.py b/semantic_router/schema.py index b8ab498c57fb94cfd13fc9332461b58a97ce769c..8841902161f6b72b1c5ee3245b3b08e1994609eb 100644 --- a/semantic_router/schema.py +++ b/semantic_router/schema.py @@ -1,5 +1,5 @@ from enum import Enum -from typing import Any, List, Optional +from typing import List, Optional from pydantic.v1 import BaseModel from pydantic.v1.dataclasses import dataclass diff --git a/semantic_router/splitters/rolling_window.py b/semantic_router/splitters/rolling_window.py index 6d8b67512e8f8232eba38ba7aabb90a25abd6598..1f84f5979b185e6ca53f30c2b34f4413949499b8 100644 --- a/semantic_router/splitters/rolling_window.py +++ b/semantic_router/splitters/rolling_window.py @@ -1,4 +1,5 @@ from typing import List + import numpy as np from semantic_router.encoders.base import BaseEncoder diff --git a/tests/unit/llms/test_llm_ollama.py b/tests/unit/llms/test_llm_ollama.py index 299989821d2dc2885f4e4cbe89dd0308d24fffda..369e5f4444789c3bb70988455bb49c7df4792445 100644 --- a/tests/unit/llms/test_llm_ollama.py +++ b/tests/unit/llms/test_llm_ollama.py @@ -1,4 +1,5 @@ import pytest + from semantic_router.llms.ollama import OllamaLLM from semantic_router.schema import Message