I wanted to test PentestGPT using Gemini-1.0 and Gemini-1.5. I used the pentestgpt/utils/APIs/chatgpt_api.py file as a template to create the pentestgpt/utilis/APIs/gemini_api.py file. Due to implementation differences between ChatGPT's chat_completion and Gemini's start_chat API functions the GeminiAPI class overrides all of the functions in the LLMAPI. Other than I modified the module_import.py file to contain the Gemini configs and added google-generativeai as a dependency to both the pyproject.toml and requirements.txt
I wanted to test PentestGPT using Gemini-1.0 and Gemini-1.5. I used the pentestgpt/utils/APIs/chatgpt_api.py file as a template to create the pentestgpt/utilis/APIs/gemini_api.py file. Due to implementation differences between ChatGPT's chat_completion and Gemini's start_chat API functions the GeminiAPI class overrides all of the functions in the LLMAPI. Other than I modified the module_import.py file to contain the Gemini configs and added google-generativeai as a dependency to both the pyproject.toml and requirements.txt