ChaoticByte/ChatGPT-PyAPI
Archived
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Extended CLI to support Bing's Chatbot and updated README

Browse source
This commit is contained in:
Julian Müller (ChaoticByte) 2023-03-13 11:00:27 +01:00
parent dcb724f902
commit 379374d097
2 changed files with 29 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

25
cli.py
View file

@ -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)