Skip to content

Commit

Permalink
Merge remote-tracking branch 'origin/chore-update-registry1-weekly-bu…
Browse files Browse the repository at this point in the history
…ndle-0.13.0' into 835-upgrade-vllm-for-gptq-bfloat16-inferencing
  • Loading branch information
justinthelaw committed Oct 1, 2024
2 parents a4c5ace + 5ee07cf commit df60811
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions tests/integration/api/test_vector_stores.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
"""Test the API endpoints for assistants."""

import json
import os
import time

import pytest
Expand All @@ -19,6 +18,7 @@
)
from leapfrogai_api.routers.openai.vector_stores import router as vector_store_router
from leapfrogai_api.routers.openai.files import router as files_router
from tests.utils.client import create_test_user
from tests.utils.data_path import data_path, TXT_FILE

INSTRUCTOR_XL_EMBEDDING_SIZE: int = 768
Expand All @@ -37,11 +37,11 @@ class MissingEnvironmentVariable(Exception):
headers: dict[str, str] = {}

try:
headers = {"Authorization": f"Bearer {os.environ['SUPABASE_USER_JWT']}"}
headers = {"Authorization": f"Bearer {create_test_user()}"}
except KeyError as exc:
raise MissingEnvironmentVariable(
"SUPABASE_USER_JWT must be defined for the test to pass. "
"Please check the api README for instructions on obtaining this token."
"Please check the packages/api and src/leapfrogai_api READMEs for instructions on obtaining this token."
) from exc

vector_store_client = TestClient(vector_store_router, headers=headers)
Expand Down

0 comments on commit df60811

Please sign in to comment.