fix

Author: cbornet

langchain/tests/test_langchain_integration_tests.py

@@ -1,3 +1,4 @@
+import os
 from typing import Literal
 
 import pytest
@@ -7,6 +8,11 @@
 
 
 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")
     def setup(
@@ -16,10 +22,10 @@ def setup(
         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
+        cls._environment = environment
+        cls._api_token = api_token
+        cls._org_id = org_id
+        cls._pipeline_id = pipeline_id
 
     @property
     def retriever_constructor(self) -> type[VectorizeRetriever]:
@@ -28,10 +34,10 @@ def retriever_constructor(self) -> type[VectorizeRetriever]:
     @property
     def retriever_constructor_params(self) -> dict:
         return {
-            "environment": self.environment,
-            "api_token": self.api_token,
-            "organization": self.org_id,
-            "pipeline_id": self.pipeline_id,
+            "environment": self._environment,
+            "api_token": self._api_token,
+            "organization": self._org_id,
+            "pipeline_id": self._pipeline_id,
         }
 
     @property