Add tests in CI

Author: cbornet

langchain/tests/test_retrievers.py

@@ -7,17 +7,11 @@
 
 import pytest
 import vectorize_client as v
+from vectorize_client import ApiClient
 
 from langchain_vectorize.retrievers import VectorizeRetriever
 
 
-@dataclass
-class TestContext:
-    api_client: v.ApiClient
-    api_token: str
-    org_id: str
-
-
 @pytest.fixture(scope="session")
 def api_token() -> str:
     token = os.getenv("VECTORIZE_TOKEN")
@@ -37,7 +31,7 @@ def org_id() -> str:
 
 
 @pytest.fixture(scope="session")
-def api_client(api_token: str) -> Iterator[TestContext]:
+def api_client(api_token: str) -> Iterator[ApiClient]:
     env = os.getenv("VECTORIZE_ENV", "prod")
     header_name = None
     header_value = None