Extended CLI to support Bing's Chatbot and updated README
This commit is contained in:
parent
dcb724f902
commit
379374d097
2 changed files with 29 additions and 8 deletions
12
README.md
12
README.md
|
@ -14,3 +14,15 @@ You need an API key for the official ChatGPT API.
|
|||
### Unofficial API for Bing Chatbot
|
||||
|
||||
This requires your Cookies to be exported to a json file. See [EdgeGPT's README](https://github.com/acheong08/EdgeGPT) for more infos.
|
||||
|
||||
## CLI
|
||||
|
||||
Using the official API via ChatGPT:
|
||||
```
|
||||
OPENAI_API_KEY="..." ./cli.py Hello
|
||||
```
|
||||
|
||||
Using Bing's Chatbot:
|
||||
```
|
||||
BING_COOKIES_FILE="~/cookies.json" ./cli.py bing Hello
|
||||
```
|
||||
|
|
25
cli.py
25
cli.py
|
@ -5,13 +5,22 @@
|
|||
from os import environ
|
||||
from sys import argv
|
||||
|
||||
from chatgpt_pyapi.openai import ChatGPT, Message, Models
|
||||
|
||||
# Read the API key from a environment variable
|
||||
API_KEY = environ["OPENAI_API_KEY"]
|
||||
|
||||
if __name__ == "__main__":
|
||||
# Create ChatGPT API instance
|
||||
cgpt = ChatGPT(API_KEY, model=Models.GPT_35_TURBO_0301)
|
||||
user_input = " ".join(argv[1:])
|
||||
bing = False
|
||||
if len(argv) > 2:
|
||||
bing = argv[1] == "bing"
|
||||
if bing:
|
||||
from chatgpt_pyapi.bing import ChatGPT, Message, ConversationStyle
|
||||
# Read the path to the cookies file from a environment variable
|
||||
BING_COOKIES_FILE = environ["BING_COOKIES_FILE"]
|
||||
# Create ChatGPT API instance
|
||||
cgpt = ChatGPT(BING_COOKIES_FILE, ConversationStyle.balanced)
|
||||
user_input = " ".join(argv[2:])
|
||||
else:
|
||||
from chatgpt_pyapi.openai import ChatGPT, Message, Models
|
||||
# Read the API key from a environment variable
|
||||
API_KEY = environ["OPENAI_API_KEY"]
|
||||
# Create ChatGPT API instance
|
||||
cgpt = ChatGPT(API_KEY, model=Models.GPT_35_TURBO_0301)
|
||||
user_input = " ".join(argv[1:])
|
||||
print(cgpt.chat(Message(user_input)).text)
|
||||
|
|
Reference in a new issue