capitalize the env variables

Author: z7ye

ads/llm/serialize.py

@@ -62,11 +62,11 @@ def load(config: dict, **kwargs):
         return OpenSearchVectorSearch(
             **config["kwargs"],
             http_auth=(
-                os.environ.get("oci_opensearch_username", None),
-                os.environ.get("oci_opensearch_password", None),
+                os.environ.get("OCI_OPENSEARCH_USERNAME", None),
+                os.environ.get("OCI_OPENSEARCH_PASSWORD", None),
             ),
-            verify_certs=True if os.environ.get("oci_opensearch_verify_certs", None).lower() == "true" else False,
-            ca_certs=os.environ.get("oci_opensearch_ca_certs", None),
+            verify_certs=True if os.environ.get("OCI_OPENSEARCH_VERIFY_CERTS", None).lower() == "true" else False,
+            ca_certs=os.environ.get("OCI_OPENSEARCH_CA_CERTS", None),
         )
 
     @staticmethod

tests/unitary/with_extras/langchain/test_serializers.py

@@ -40,10 +40,10 @@ def embed_query(self, text: str) -> List[float]:
 class TestOpensearchSearchVectorSerializers(unittest.TestCase):
     @classmethod
     def setUpClass(cls):
-        cls.env_patcher = mock.patch.dict(os.environ, {"oci_opensearch_username": "username",
-                                                    "oci_opensearch_password": "password",
-                                                    "oci_opensearch_verify_certs": "True",
-                                                    "oci_opensearch_ca_certs": "/path/to/cert.pem"})
+        cls.env_patcher = mock.patch.dict(os.environ, {"OCI_OPENSEARCH_USERNAME": "username",
+                                                    "OCI_OPENSEARCH_PASSWORD": "password",
+                                                    "OCI_OPENSEARCH_VERIFY_CERTS": "True",
+                                                    "OCI_OPENSEARCH_CA_CERTS": "/path/to/cert.pem"})
         cls.env_patcher.start()
         cls.index_name = "test_index"
         cls.embeddings = FakeEmbeddings()
@@ -52,9 +52,9 @@ def setUpClass(cls):
             embedding_function=cls.embeddings,
             index_name=cls.index_name,
             engine="lucene",
-            http_auth=(os.environ["oci_opensearch_username"], os.environ["oci_opensearch_password"]),
-            verify_certs=os.environ["oci_opensearch_verify_certs"],
-            ca_certs=os.environ["oci_opensearch_ca_certs"],
+            http_auth=(os.environ["OCI_OPENSEARCH_USERNAME"], os.environ["OCI_OPENSEARCH_PASSWORD"]),
+            verify_certs=os.environ["OCI_OPENSEARCH_VERIFY_CERTS"],
+            ca_certs=os.environ["OCI_OPENSEARCH_CA_CERTS"],
         )
         cls.serializer = OpenSearchVectorDBSerializer()
         super().setUpClass()