Merge pull request #10 from code-yeongyu/feature/hello-pypi

Author: code-yeongyu

.github/workflows/check_code.yml

@@ -0,0 +1,37 @@
+name: Check Code
+on:
+  pull_request:
+
+jobs:
+  static-analysis:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v3
+        with:
+          fetch-depth: 0
+      - name: Set up Python 3.9
+        uses: actions/setup-python@v4
+        with:
+          python-version: 3.9
+
+      - uses: snok/[email protected]
+        with:
+          virtualenvs-create: true
+          virtualenvs-in-project: true
+
+      - name: Cache Dependencies
+        uses: actions/cache@v2
+        id: cache-dependencies
+        with:
+          path: .venv
+          key: venv-${{ runner.os }}-${{ hashFiles('**/poetry.lock') }}
+
+      - name: Install Dependencies if cache doesn't hit
+        if: steps.cache-dependencies.cache-hit != 'true'
+        run: poetry install
+
+      - name: Check Code Styles
+        run: poetry run invoke check-code-style
+
+      - name: Check Types
+        run: poetry run invoke check-types

.github/workflows/release.yml

@@ -0,0 +1,36 @@
+name: Release Package to PyPI
+on:
+  release:
+    types: [created]
+jobs:
+  release:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v3
+        with:
+          fetch-depth: 0
+      - name: Set up Python 3.9
+        uses: actions/setup-python@v4
+        with:
+          python-version: 3.9
+
+      - uses: snok/[email protected]
+        with:
+          virtualenvs-create: true
+          virtualenvs-in-project: true
+
+      - name: Cache Dependencies
+        uses: actions/cache@v2
+        id: cache-dependencies
+        with:
+          path: .venv
+          key: venv-${{ runner.os }}-${{ hashFiles('**/poetry.lock') }}
+
+      - name: Install Dependencies if cache doesn't hit
+        if: steps.cache-dependencies.cache-hit != 'true'
+        run: poetry install
+
+      - name: Publish to PyPI
+        run:
+          poetry config http-basic.pypi ${{ secrets.PYPI_USERNAME }} ${{ secrets.PYPI_PASSWORD }};
+          poetry run invoke release ${{ github.event.release.name }}

README.md

@@ -20,7 +20,7 @@ A simple Python code that connects to OpenAI's ChatGPT and executes the returned
 ## Installation 🔧
 
 ```sh
-curl -sL https://raw.githubusercontent.com/code-yeongyu/AiShell/master/install.sh | sh
+pip install aishell
 ```
 
 ## Prerequisites 📚

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