morpheuslord / GPT_Vuln-analyzer

Uses ChatGPT API, Bard API, and Llama2, Python-Nmap, DNS Recon, PCAP and JWT recon modules and uses the GPT3 model to create vulnerability reports based on Nmap scan data, and DNS scan information. It can also perform subdomain enumeration to a great extent
MIT License
490 stars 56 forks source link

adding support for GPT-4, Anthropic, and Llama2 #53

Closed krrishdholakia closed 1 year ago

krrishdholakia commented 1 year ago

Hi @morpheuslord ,

Noticed you're calling just OpenAI (seemed like a finetuned gpt-3 model for generating JSON output). I'm working on litellm (simple library to standardize LLM API Calls - https://github.com/BerriAI/litellm) and was wondering if we could be helpful.

Added support for GPT-4, Anthropic, Llama2, Cohere and PaLM by replacing the raw openai.Completion endpoint with completion from litellm.

Would love to know if this helps.