Vision (#230)

* fix: 🐛 fix OPENAI key setting issue and update readme

* feat: 🎸 update gpt4o

* style: format code with Black

This commit fixes the style issues introduced in 99581a8 according to the output
from Black.

Details: #229

* fix: 🐛 fix OPENAI_KEY typo

* style: format code with Black

This commit fixes the style issues introduced in 8f9091c according to the output
from Black.

Details: #230

---------

Co-authored-by: deepsource-autofix[bot] <62050782+deepsource-autofix[bot]@users.noreply.github.com>

README.md

@@ -100,7 +100,7 @@ PentestGPT is tested under `Python 3.10`. Other Python3 versions should work but
      - `pip3 install -e .`
 2. To use OpenAI API
    - **Ensure that you have link a payment method to your OpenAI account.**
-   - export your API key with `export OPENAI_KEY='<your key here>'`
+   - export your API key with `export OPENAI_API_KEY='<your key here>'`
    - export API base with `export OPENAI_BASEURL='https://api.xxxx.xxx/v1'`if you need.
    - Test the connection with `pentestgpt-connection`
 3. To verify that the connection is configured properly, you may run `pentestgpt-connection`. After a while, you should see some sample conversation with ChatGPT.

pentestgpt/_version.py

@@ -1 +1 @@
-__version__ = '"0.13.0"'
+__version__ = '"0.13.1"'

pentestgpt/utils/APIs/azure_api.py

@@ -45,7 +45,7 @@ class AzureGPTAPI(LLMAPI):
     def __init__(self, config_class, use_langfuse_logging=False):
         self.name = str(config_class.model)
         openai.api_type = "azure"
-        openai.api_key = os.getenv("OPENAI_KEY", None)
+        openai.api_key = os.getenv("OPENAI_API_KEY", None)
         openai.api_base = config_class.api_base
         self.model = config_class.model
         self.log_dir = config_class.log_dir

pentestgpt/utils/APIs/chatgpt_api.py

@@ -59,7 +59,7 @@ def __init__(self, config_class, use_langfuse_logging=False):
 
             self.langfuse = Langfuse()
 
-        openai.api_key = os.getenv("OPENAI_KEY", None)
+        openai.api_key = os.getenv("OPENAI_API_KEY", None)
         openai.api_base = config_class.api_base
         self.model = config_class.model
         self.log_dir = config_class.log_dir

pentestgpt/utils/APIs/chatgpt_vision_api.py

@@ -59,7 +59,7 @@ def __init__(self, config_class, use_langfuse_logging=False):
 
             self.langfuse = Langfuse()
 
-        openai.api_key = os.getenv("OPENAI_KEY", None)
+        openai.api_key = os.getenv("OPENAPI_API_KEY", None)
         openai.api_base = config_class.api_base
         self.model = config_class.model
         self.log_dir = config_class.log_dir

pentestgpt/utils/APIs/module_import.py

@@ -73,7 +73,9 @@ class GPT35Turbo16kConfigClass:
     # set up the openai key
     openai_key = os.getenv("OPENAI_API_KEY", None)
     if openai_key is None:
-        print("Your OPENAI_KEY is not set. Please set it in the environment variable.")
+        print(
+            "Your OPENAI_API_KEY is not set. Please set it in the environment variable."
+        )
     error_wait_time: float = 20
     is_debugging: bool = False
 

pentestgpt/utils/vectorDB.py

@@ -37,7 +37,7 @@ def __init__(self, project_name: str, vectordb_name: str):
 
         # load configurations
         pinecone_api_key = os.getenv("PINECONE_API_KEY", None)
-        os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_KEY", None)
+        os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY", None)
         # save the abs directory of the vectorDB on top of the project directory
         self.vectordb_directory = os.path.join(
             os.path.dirname(os.path.dirname(os.path.abspath(__file__))),

setup.py

@@ -7,7 +7,7 @@
 
 setup(
     name="pentestgpt",
-    version="0.13.0",
+    version="0.13.1",
     description="PentestGPT, a GPT-empowered penetration testing tool",
     long_description="""
     PentestGPT is a penetration testing tool empowered by ChatGPT.