misc: fixing pylint issues

Author: muhammadyaseen

graphgen/models/llm/api/openai_client.py

@@ -59,7 +59,9 @@ def __init__(
         self.rpm = rpm or RPM()
         self.tpm = tpm or TPM()
 
-        assert backend in ["openai_api", "azure_openai_api"], f"Unsupported backend {backend}. Use 'openai_api' or 'azure_openai_api'."
+        assert (
+            backend in ("openai_api", "azure_openai_api")
+        ), f"Unsupported backend '{backend}'. Use 'openai_api' or 'azure_openai_api'."
         self.backend = backend
 
         self.__post_init__()
@@ -68,7 +70,6 @@ def __post_init__(self):
 
         api_name = self.backend.replace("_", " ") 
         assert self.api_key is not None, f"Please provide api key to access {api_name}."
-
         if self.backend == "openai_api":    
             self.client = AsyncOpenAI(
                 api_key=self.api_key or "dummy", base_url=self.base_url

graphgen/operators/init/init_llm.py

@@ -27,10 +27,10 @@ def create_llm_wrapper(backend: str, config: Dict[str, Any]) -> BaseLLMWrapper:
             from graphgen.models.llm.api.http_client import HTTPClient
 
             return HTTPClient(**config)
-        if backend == "openai_api" or backend == "azure_openai_api":
+        if backend in ("openai_api", "azure_openai_api"):
             from graphgen.models.llm.api.openai_client import OpenAIClient
-            
-            # pass in concrete backend to the OpenAIClient so that internally we can distinguish between OpenAI and Azure OpenAI
+            # pass in concrete backend to the OpenAIClient so that internally we can distinguish 
+            # between OpenAI and Azure OpenAI
             return OpenAIClient(**config, backend=backend)
         if backend == "ollama_api":
             from graphgen.models.llm.api.ollama_client import OllamaClient