Wrap AiShell as module

Author: code-yeongyu

aishell/__init__.py

@@ -0,0 +1,5 @@
+from .cli import cli_app
+
+
+def main():
+    cli_app()

aishell/__main__.py

@@ -0,0 +1,4 @@
+import aishell
+
+if __name__ == '__main__':
+    aishell.main()

aishell/cli.py

@@ -0,0 +1,24 @@
+import os
+
+import typer
+from rich.console import Console
+
+from aishell.query_clients import ChatGPTClient, GPT3Client, QueryClient
+
+cli_app = typer.Typer()
+
+
+@cli_app.command()
+def ask(question: str, use_chatgpt: bool = False):
+    query_client: QueryClient
+    if use_chatgpt:
+        query_client = ChatGPTClient()
+    else:
+        query_client = GPT3Client()
+
+    console = Console()
+    with console.status(f'[green] Asking `{question}` ...'):
+        response = query_client.query(question)
+    console.print(f'[italic]ai$hell: {response}\n')
+
+    os.system(response)

aishell/main.py

@@ -36,8 +36,8 @@ def query(self, prompt: str) -> str:
 
         response_text = ''
         for data in chatbot.ask(prompt):
-            responsed_text = data['message']
+            response_text = data['message']
 
-        responsed_text = make_executable_command(responsed_text)
+        response_text = make_executable_command(response_text)
 
         return response_text

aishell/query_clients/chatgpt_client.py

@@ -2,8 +2,9 @@
 from typing import cast
 
 import openai
-from models import OpenAIResponseModel
-from utils import make_executable_command
+
+from aishell.models import OpenAIResponseModel
+from aishell.utils import make_executable_command
 
 from .query_client import QueryClient