Add openai key path

pandasai/llm/openai.py

@@ -49,6 +49,7 @@ class OpenAI(BaseOpenAI):
     def __init__(
         self,
         api_token: Optional[str] = None,
+        api_key_path: Optional[str] = None,
         **kwargs,
     ):
         """
@@ -59,11 +60,16 @@ def __init__(
             **kwargs: Extended Parameters inferred from BaseOpenAI class
 
         """
-
         self.api_token = api_token or os.getenv("OPENAI_API_KEY") or None
-        if self.api_token is None:
-            raise APIKeyNotFoundError("OpenAI API key is required")
-        openai.api_key = self.api_token
+        self.api_key_path = api_key_path
+
+        if (not self.api_token) and (not self.api_key_path):
+            raise APIKeyNotFoundError("Either OpenAI API key or key path is required")
+
+        if self.api_token:
+            openai.api_key = self.api_token
+        else:
+            openai.api_key_path = self.api_key_path
 
         self.openai_proxy = kwargs.get("openai_proxy") or os.getenv("OPENAI_PROXY")
         if self.openai_proxy:

tests/llms/test_openai.py

@@ -25,6 +25,9 @@ def test_type_without_token(self):
     def test_type_with_token(self):
         assert OpenAI(api_token="test").type == "openai"
 
+    def test_type_with_key_path(self):
+        assert OpenAI(api_key_path=".key").type == "openai"
+
     def test_proxy(self):
         proxy = "http://proxy.mycompany.com:8080"
         client = OpenAI(api_token="test", openai_proxy=proxy)