Skip to content

Commit 325c444

Browse files
committed
Add support for OfficialChatGPT
1 parent d3c3f5e commit 325c444

File tree

1 file changed

+10
-5
lines changed

1 file changed

+10
-5
lines changed

aishell/cli.py

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -3,18 +3,23 @@
33
import typer
44
from rich.console import Console
55

6-
from aishell.query_clients import GPT3Client, QueryClient, ReverseEngineeredChatGPTClient
6+
from aishell.models.language_model import LanguageModel
7+
from aishell.query_clients import GPT3Client, OfficialChatGPTClient, QueryClient, ReverseEngineeredChatGPTClient
78

89
cli_app = typer.Typer()
910

1011

1112
@cli_app.command()
12-
def ask(question: str, use_chatgpt: bool = False):
13+
def ask(question: str, language_model: LanguageModel = LanguageModel.OFFICIAL_CHATGPT):
1314
query_client: QueryClient
14-
if use_chatgpt:
15-
query_client = ReverseEngineeredChatGPTClient()
16-
else:
15+
if language_model == LanguageModel.GPT3:
1716
query_client = GPT3Client()
17+
elif language_model == LanguageModel.OFFICIAL_CHATGPT:
18+
query_client = OfficialChatGPTClient()
19+
elif language_model == LanguageModel.REVERSE_ENGINEERED_CHATGPT:
20+
query_client = ReverseEngineeredChatGPTClient()
21+
22+
query_client.query(question)
1823

1924
console = Console()
2025
with console.status(

0 commit comments

Comments
 (0)