diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index 7e3ad27128f3cdd09c40ce555b0882dd32d78618..4e13b9382771a2da57007ebe4170d098e99a72cb 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -46,6 +46,8 @@ jobs: run: | python -m nltk.downloader punkt stopwords wordnet - name: Pytest + env: + PINECONE_API_KEY: ${{ secrets.PINECONE_API_KEY }} run: | make test - name: Upload coverage to Codecov diff --git a/semantic_router/index/qdrant.py b/semantic_router/index/qdrant.py index ea90e46e4529f54320dfcfe71dc1d739200fcde2..a2292da0eae2a9245a0782a4a2b40c34e2696b18 100644 --- a/semantic_router/index/qdrant.py +++ b/semantic_router/index/qdrant.py @@ -167,7 +167,7 @@ class QdrantIndex(BaseIndex): List[Tuple]: A list of (route_name, utterance) objects. """ - import grpc + from qdrant_client import grpc results = [] next_offset = None diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py index 8af6e1cc1135c2c1256913044550de8f1a24fe18..59830da1ac3e06060f944b5ad4e071503ebac90a 100644 --- a/tests/unit/test_layer.py +++ b/tests/unit/test_layer.py @@ -258,7 +258,8 @@ class TestRouteLayer: assert query_result in ["Route 1"] def test_query_filter_pinecone(self, openai_encoder, routes, index_cls): - pineconeindex = PineconeIndex() + pinecone_api_key = os.environ["PINECONE_API_KEY"] + pineconeindex = PineconeIndex(api_key=pinecone_api_key) route_layer = RouteLayer( encoder=openai_encoder, routes=routes, index=pineconeindex )