search

GreyDGL / PentestGPT

A GPT-empowered penetration testing tool
MIT License
7.29k stars 886 forks source link

Added functionality for api_base, which allows you to change openai's baseurl when using API functionality. #123

Closed jiayuqi7813 closed 1 year ago

jiayuqi7813 commented 1 year ago

I added the option to use --baseUrl to replace the api_base when using api functions, and to support the use of apis such as auzreai or api2d. The rest of the methods have not changed.

Below is a screenshot of the test image

image

jiayuqi7813 commented 1 year ago

I don't know why git recognizes main.py as deleted and then creates a new one, but it doesn't affect the program

jiayuqi7813 commented 1 year ago

@GreyDGL
Hey,can you see this?

GreyDGL commented 1 year ago

Hey @jiayuqi7813 thanks for the PR! I briefly went through the changes and they look good to me. I'll apply your updates to a local testing to ensure that it works as expected. Will merge it if everything works alright.

GreyDGL commented 1 year ago

I'll merge it into the main branch. Meanwhile, for future development, I recommend you use black for python formatting to keep consistency with the project. Thanks for the contribution!