fix

Author: cbornet

langchain/tests/conftest.py

@@ -14,7 +14,7 @@
 
 @pytest.fixture(scope="session")
 def api_token() -> str:
-    token = os.getenv("VECTORIZE_TOKEN")
+    token = os.getenv("VECTORIZE_TOKEN", "wdcwd")
     if not token:
         msg = "Please set the VECTORIZE_TOKEN environment variable"
         raise ValueError(msg)
@@ -23,7 +23,7 @@ def api_token() -> str:
 
 @pytest.fixture(scope="session")
 def org_id() -> str:
-    org = os.getenv("VECTORIZE_ORG")
+    org = os.getenv("VECTORIZE_ORG", "wdcd")
     if not org:
         msg = "Please set the VECTORIZE_ORG environment variable"
         raise ValueError(msg)

langchain/tests/test_langchain_integration_tests.py

@@ -7,24 +7,19 @@
 
 
 class TestVectorizeRetrieverIntegration(RetrieversIntegrationTests):
-    _environment: Literal["prod", "dev", "local", "staging"]
-    _api_token: str
-    _org_id: str
-    _pipeline_id: str
 
-    @classmethod
-    @pytest.fixture(autouse=True, scope="class")
+    @pytest.fixture(autouse=True)
     def setup(
-        cls,
+        self,
         environment: Literal["prod", "dev", "local", "staging"],
         api_token: str,
         org_id: str,
         pipeline_id: str,
     ) -> None:
-        cls._environment = environment
-        cls._api_token = api_token
-        cls._org_id = org_id
-        cls._pipeline_id = pipeline_id
+        self._environment = environment
+        self._api_token = api_token
+        self._org_id = org_id
+        self._pipeline_id = pipeline_id
 
     @property
     def retriever_constructor(self) -> type[VectorizeRetriever]: