test: add Azure OpenAI validation tests

Author: Arivunidhi A

chromadb/test/ef/test_openai_ef.py

@@ -5,6 +5,7 @@
 from chromadb.utils.embedding_functions.openai_embedding_function import (
     OpenAIEmbeddingFunction,
 )
+from chromadb.errors import InvalidArgumentError
 
 
 def test_with_embedding_dimensions() -> None:
@@ -41,7 +42,7 @@ def test_with_incorrect_api_key() -> None:
 def test_azure_requires_deployment_id() -> None:
     """Azure OpenAI should require deployment_id parameter."""
     pytest.importorskip("openai", reason="openai not installed")
-    with pytest.raises(ValueError, match="deployment_id must be specified"):
+    with pytest.raises(InvalidArgumentError, match="deployment_id must be specified"):
         OpenAIEmbeddingFunction(
             api_key="test_key",
             api_type="azure",
@@ -54,7 +55,7 @@ def test_azure_requires_deployment_id() -> None:
 def test_azure_requires_api_version() -> None:
     """Azure OpenAI should require api_version parameter."""
     pytest.importorskip("openai", reason="openai not installed")
-    with pytest.raises(ValueError, match="api_version must be specified"):
+    with pytest.raises(InvalidArgumentError, match="api_version must be specified"):
         OpenAIEmbeddingFunction(
             api_key="test_key",
             api_type="azure",
@@ -67,7 +68,7 @@ def test_azure_requires_api_version() -> None:
 def test_azure_requires_api_base() -> None:
     """Azure OpenAI should require api_base parameter."""
     pytest.importorskip("openai", reason="openai not installed")
-    with pytest.raises(ValueError, match="api_base must be specified"):
+    with pytest.raises(InvalidArgumentError, match="api_base must be specified"):
         OpenAIEmbeddingFunction(
             api_key="test_key",
             api_type="azure",