clai 0.1.0__tar.gz → 0.3.0__tar.gz

{clai-0.1.0 → clai-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: clai
-Version: 0.1.0
+Version: 0.3.0
 Summary: Command Line AI- this tool lets you call ChatGPT from a CLI
 License: Proprietary
 Author: apockill
@@ -11,16 +11,21 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
+Requires-Dist: PyAutoGUI (>=0.9.53,<0.10.0)
+Requires-Dist: PyWinCtl (>=0.0.43,<0.0.44)
 Requires-Dist: openai (>=0.27.0,<0.28.0)
+Requires-Dist: pytesseract (>=0.3.10,<0.4.0)
 Description-Content-Type: text/markdown
 
 # clai
 Command Line AI- this tool lets you call ChatGPT from a CLI. 
 
 I'm designing this to be used in conjunction with a fork of [shin][shin], which will allow you
-to call `clai` from any textbox in your computer. Finally, chatGPT everywhere!
+to call `clai` from any textbox in your computer. Finally, ChatGPT everywhere!
+
+The long-term vision for this project is to add support for extracting context. For example, it would
+read the current text on a window and be able to add to it, or answer questions about it.
 
-[shin]: https://github.com/apockill/shin
 _________________
 
 [![PyPI version](https://badge.fury.io/py/clai.svg)](http://badge.fury.io/py/clai)
@@ -35,17 +40,30 @@ _________________
 _________________
 
 ## Installation
+
 1. The recommended installation method is to use `pipx`, via
-    ```
+    ```bash
     pipx install clai
     ```
-
+   Optionally, install `tesseract` so that `clai` can read the screen context and send that along with requests:
+   ```bash
+   sudo apt install tesseract-ocr scrot
+   ```
 1. Then go to [OpenAI] and create an API Key. Once it's generated, add the following to 
-   your `.bashrc`:
+   your `~/.profile`:
    ```bash
    export OPENAI_API_TOKEN=<paste here>
    ```
 
+1. The best way to use this tool is in conjunction with the tool [shin][shin], which allows you
+   to run arbitrary bash commands in any textbox in a linux computer, using ibus. To use 
+   that, install 'shin' via the fork above, then configure
+   it in your `~/.profile` to call `clai` by default:
+   ```bash
+   export SHIN_DEFAULT_COMMAND="clai"
+   ```
+1. Log out then log back in for the changes to take effect!
+
 [OpenAI]: https://platform.openai.com/account/api-keys
 
 ## Usage
@@ -54,6 +72,7 @@ Invoke the assistant with the format `clai <your prompt>`. For example:
 clai Write an email saying I'll be late to work because I'm working on commandline AIs
 ```
 
+
 ## Development
 
 ### Installing python dependencies
@@ -76,5 +95,5 @@ bash .github/format.sh
 bash .github/check_lint.sh
 ```
 
-# TODO: remove references to tests?
+[shin]: https://github.com/apockill/shin
 

{clai-0.1.0 → clai-0.3.0}/README.md

@@ -2,9 +2,11 @@
 Command Line AI- this tool lets you call ChatGPT from a CLI. 
 
 I'm designing this to be used in conjunction with a fork of [shin][shin], which will allow you
-to call `clai` from any textbox in your computer. Finally, chatGPT everywhere!
+to call `clai` from any textbox in your computer. Finally, ChatGPT everywhere!
+
+The long-term vision for this project is to add support for extracting context. For example, it would
+read the current text on a window and be able to add to it, or answer questions about it.
 
-[shin]: https://github.com/apockill/shin
 _________________
 
 [![PyPI version](https://badge.fury.io/py/clai.svg)](http://badge.fury.io/py/clai)
@@ -19,17 +21,30 @@ _________________
 _________________
 
 ## Installation
+
 1. The recommended installation method is to use `pipx`, via
-    ```
+    ```bash
     pipx install clai
     ```
-
+   Optionally, install `tesseract` so that `clai` can read the screen context and send that along with requests:
+   ```bash
+   sudo apt install tesseract-ocr scrot
+   ```
 1. Then go to [OpenAI] and create an API Key. Once it's generated, add the following to 
-   your `.bashrc`:
+   your `~/.profile`:
    ```bash
    export OPENAI_API_TOKEN=<paste here>
    ```
 
+1. The best way to use this tool is in conjunction with the tool [shin][shin], which allows you
+   to run arbitrary bash commands in any textbox in a linux computer, using ibus. To use 
+   that, install 'shin' via the fork above, then configure
+   it in your `~/.profile` to call `clai` by default:
+   ```bash
+   export SHIN_DEFAULT_COMMAND="clai"
+   ```
+1. Log out then log back in for the changes to take effect!
+
 [OpenAI]: https://platform.openai.com/account/api-keys
 
 ## Usage
@@ -38,6 +53,7 @@ Invoke the assistant with the format `clai <your prompt>`. For example:
 clai Write an email saying I'll be late to work because I'm working on commandline AIs
 ```
 
+
 ## Development
 
 ### Installing python dependencies
@@ -60,4 +76,4 @@ bash .github/format.sh
 bash .github/check_lint.sh
 ```
 
-# TODO: remove references to tests?
+[shin]: https://github.com/apockill/shin

{clai-0.1.0 → clai-0.3.0}/clai/__init__.py

@@ -3,4 +3,4 @@
 Command Line AI- this tool lets you call ChatGPT from a CLI
 """
 __version__ = "0.1.0"
-from .invoke import invoke
+from .main import main

clai-0.3.0/clai/behavior_context.py

@@ -0,0 +1,161 @@
+from dataclasses import dataclass
+from typing import Literal, Union
+
+from clai.ocr_drivers import WindowContext
+
+USER_PROMPT_FORMAT = """
+User Prompt:
+```
+{user_prompt}
+```
+"""
+OCR_EXTRACTION_FORMAT = """
+Active Window Title: {active_window_title}
+
+Active Window OCR Extracted Text (RAW):
+------ OCR DATA START ------
+```
+{ocr_text}
+```
+------ OCR DATA END ------
+
+{user_prompt}
+
+Please answer "User Prompt" using the raw OCR text as context to the message.
+"""
+
+
+@dataclass
+class Prompt:
+    context: WindowContext
+    prompt: str
+
+    def __str__(self) -> str:
+        """Serialize the Prompt with differing formats, depending on whether window
+        content was available
+
+        :return: The window context and prompt in a standardized format
+        """
+        """."""
+        user_prompt = USER_PROMPT_FORMAT.format(user_prompt=self.prompt.strip())
+        if self.context.clean_screen_text and self.context.active_window_name:
+            return OCR_EXTRACTION_FORMAT.format(
+                active_window_title=self.context.active_window_name.strip(),
+                ocr_text=self.context.clean_screen_text.strip(),
+                user_prompt=user_prompt.strip(),
+            )
+        return user_prompt.strip()
+
+
+@dataclass
+class Message:
+    role: Literal["system", "user", "assistant"]
+    content: Union[Prompt, str]
+
+    def to_api(self) -> dict[str, str]:
+        """To OpenAPI format"""
+        if isinstance(self.content, str) and self.role == "user":
+            raise RuntimeError("The user message must be of type Prompt!")
+
+        return {"role": self.role, "content": str(self.content)}
+
+
+_DEFAULT_ASSISTANT_ROLE = """
+You are an assistant that is capable of being called anywhere on a desktop computer. You
+may be called within the context of an email, a URL box, commandline, a text editor, or
+even word documents!
+
+Your role is to answer the users request as shortly and succinctly as possible. You
+will follow the following rules:
+
+When asked to write long-form text content:
+1) Never ask for more information. If something is to be guessed, write it in template
+   format. For example, if asked to write an email use <Insert time here> when writing
+   the portion of an email that specifies something that was not included in the users
+   question.
+2) Only assume the content is long form if the user mentions email, or 'long message'.
+
+When asked to write a command, code, formulas, or any one-line response task:
+1) NEVER WRITE EXPLANATIONS! Only include the command/code/etc, ready to be run
+2) NEVER WRITE USAGE INSTRUCTIONS! Do not explain how to use the command/code/formulas.
+3) NEVER WRITE NOTES ABOUT THE IMPLEMENTATION!
+   Do not explain what it does or it's limitations.
+4) Remember, the text that you write will immediately be run, do not include code blocks
+5) If there is something that requires user input, such as a cell in a sheet or a
+   variable from the user, write it inside of brackets, like this: <INPUT DESCRIBER>,
+   where the insides of the bracket have an example of what is needed to be filled in.
+6) Assume a linux desktop environment in a bash shell. Use freely available unix tools.
+
+You will receive OCR context and window title names, for some prompts. They are very 
+noisy, use best-effort when reading them.
+"""
+_EXAMPLE_EMAIL = """
+Dear <Recipient's Name>,
+
+I hope this email finds you well. I am writing to request a meeting with you on <Date and Time>, and I would appreciate it if you could confirm your availability at your earliest convenience.
+
+The purpose of this meeting is to discuss <Purpose of the Meeting> with you. Specifically, I would like to <Agenda Item 1>, <Agenda Item 2>, and <Agenda Item 3>. The meeting will last approximately <Meeting Duration> and will take place at <Meeting Location>.
+
+Please let me know if this date and time work for you. If not, please suggest an alternative time that is convenient for you. Additionally, if there are any documents or information you would like me to review before the meeting, please let me know, and I will make sure to review them.
+
+I look forward to hearing from you soon.
+
+Best regards,
+
+<Your Name>
+"""  # noqa: E501
+_EXAMPLE_REGEX = '=IFERROR(REGEXEXTRACT(<INPUT CELL HERE>, "[A-z0-9._%+-]+@[A-z0-9.-]+\.[A-z]{2,4}");"")'  # noqa
+_EXAMPLE_PYTHON = """
+def fibonacci(n: int) -> Generator[int, None, None]:
+    a, b = 0, 1
+    for _ in range(n):
+        yield a
+        a, b = b, a + b
+"""
+_EXAMPLE_GOOGLE_SHEETS = '=IFERROR(REGEXEXTRACT(<INPUT CELL HERE>, "[A-z0-9._%+-]+@[A-z0-9.-]+\.[A-z]{2,4}");"")'  # noqa
+_EXAMPLE_BASH_COMMAND = "grep -rnw . -e 'bruh'"
+
+MESSAGE_CONTEXT: list[Message] = [
+    Message(role="system", content=_DEFAULT_ASSISTANT_ROLE),
+    Message(
+        role="user",
+        content=Prompt(
+            WindowContext(),
+            prompt="commandline search for files with the name 'bruh' in them",
+        ),
+    ),
+    Message(role="assistant", content=_EXAMPLE_BASH_COMMAND),
+    Message(
+        role="user",
+        content=Prompt(
+            context=WindowContext(), prompt="email set up a meeting next week"
+        ),
+    ),
+    Message(role="assistant", content=_EXAMPLE_EMAIL),
+    Message(
+        role="user",
+        content=Prompt(
+            context=WindowContext(),
+            prompt="google sheets formula extracts an email from string of text",
+        ),
+    ),
+    Message(role="assistant", content=_EXAMPLE_GOOGLE_SHEETS),
+    Message(
+        role="user",
+        content=Prompt(
+            context=WindowContext(),
+            prompt="google sheets formula extracts an email from string of text",
+        ),
+    ),
+    Message(role="assistant", content=_EXAMPLE_REGEX),
+    Message(
+        role="user",
+        content=Prompt(
+            context=WindowContext(),
+            prompt="python fibonacci function in form of a generator",
+        ),
+    ),
+    Message(role="assistant", content=_EXAMPLE_PYTHON),
+]
+
+__all__ = ["MESSAGE_CONTEXT", "Message", "Prompt"]

clai-0.3.0/clai/main.py

@@ -0,0 +1,22 @@
+from argparse import ArgumentParser
+
+from .api import initialize_api
+from .message_creation import create_message_context
+
+
+def main() -> None:
+    parser = ArgumentParser("CLAI- your own command line AI!")
+    parser.add_argument("prompt", type=str, nargs="+")
+    parser.add_argument("-m", "--model", default="gpt-3.5-turbo")
+    args = parser.parse_args()
+
+    openai = initialize_api()
+
+    prompt = " ".join(args.prompt)
+    response = openai.ChatCompletion.create(
+        model=args.model,
+        messages=create_message_context(prompt),
+    )
+
+    best_response = response["choices"][0]["message"]["content"]
+    print(best_response)

clai-0.3.0/clai/message_creation.py

@@ -0,0 +1,20 @@
+from .behavior_context import MESSAGE_CONTEXT, Message, Prompt
+from .ocr_drivers import get_driver
+
+
+def create_message_context(prompt: str) -> list[dict[str, str]]:
+    """An important part of a ChatBot is the prior context. Here, we carefully
+    construct the context for the messages we are sending.
+
+    This function returns an OpenAI-API ready message with the prompt included.
+    :param prompt: The prompt for the chatbot
+    :return: The API-ready message
+    """
+    driver = get_driver()
+    window_context = driver.extract_context()
+    new_context = MESSAGE_CONTEXT[:]
+    new_context.append(
+        Message(role="user", content=Prompt(context=window_context, prompt=prompt))
+    )
+    api_format = [m.to_api() for m in new_context]
+    return api_format

clai-0.3.0/clai/ocr_drivers/__init__.py

@@ -0,0 +1,7 @@
+from .base_driver import BaseOCRDriver, WindowContext
+from .linux_driver import LinuxOCRDriver
+
+
+def get_driver() -> BaseOCRDriver:
+    """In the future, we can return other OS compatible OCR solutions here"""
+    return LinuxOCRDriver()

clai-0.3.0/clai/ocr_drivers/base_driver.py

@@ -0,0 +1,39 @@
+from abc import ABC, abstractmethod
+from dataclasses import dataclass
+from typing import Optional
+
+_MIN_CHARACTERS_PER_LINE = 10
+
+
+@dataclass
+class WindowContext:
+    raw_screen_text: Optional[str] = None
+    """If the driver supports it, the text extracted from the active window will be
+    filled here."""
+
+    active_window_name: Optional[str] = None
+    """If the driver supports it, the current active window name will be filled here."""
+
+    @property
+    def clean_screen_text(self) -> Optional[str]:
+        if not self.raw_screen_text:
+            return None
+
+        lines: list[str] = self.raw_screen_text.split("\n")
+        clean_lines = []
+        for line in lines:
+            line = line.strip()
+            if len(line) > _MIN_CHARACTERS_PER_LINE:
+                clean_lines.append(line)
+
+        clean_text = "\n".join(clean_lines)
+        return clean_text
+
+
+class BaseOCRDriver(ABC):
+    """This base class can be used to standardize the interface for future OS's"""
+
+    @abstractmethod
+    def extract_context(self) -> WindowContext:
+        """A method to extract the current useful context from the in-focus window"""
+        pass

clai-0.3.0/clai/ocr_drivers/linux_driver.py

@@ -0,0 +1,35 @@
+from typing import cast
+
+import pyautogui
+import pytesseract
+import pywinctl
+from PIL.Image import Image
+
+from .base_driver import BaseOCRDriver, WindowContext
+
+
+class LinuxOCRDriver(BaseOCRDriver):
+    def extract_context(self) -> WindowContext:
+        screenshot = self._extract_active_window_screenshot()
+
+        # Perform OCR and clean up the text
+        raw_ocr_text = pytesseract.image_to_string(screenshot)
+
+        return WindowContext(
+            raw_screen_text=raw_ocr_text,
+            active_window_name=cast(str, pywinctl.getActiveWindowTitle()),
+        )
+
+    def _extract_active_window_screenshot(self) -> Image:
+        # Get the active window object
+        active_window = pywinctl.getActiveWindow()  # type: ignore
+        region = (
+            active_window.left,
+            active_window.top,
+            active_window.width,
+            active_window.height,
+        )
+
+        # Take a screenshot of the active window
+        screenshot = pyautogui.screenshot(region=region)
+        return screenshot

{clai-0.1.0 → clai-0.3.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "clai"
-version = "0.1.0"
+version = "0.3.0"
 description = "Command Line AI- this tool lets you call ChatGPT from a CLI"
 authors = ["apockill <apocthiel@gmail.com>"]
 license = "Proprietary"
@@ -9,6 +9,9 @@ readme = "README.md"
 [tool.poetry.dependencies]
 python = "^3.8"
 openai = "^0.27.0"
+pytesseract = "^0.3.10"
+PyAutoGUI = "^0.9.53"
+PyWinCtl = "^0.0.43"
 
 [tool.poetry.dev-dependencies]
 darglint = "^1.8.1"
@@ -25,8 +28,8 @@ portray = "^1.7"
 cruft = "^2.10"
 
 [tool.poetry.scripts]
-clai = "clai:invoke"
-ai = "clai:invoke"
+clai = "clai:main"
+ai = "clai:main"
 
 [build-system]
 requires = ["poetry>=0.12"]

clai-0.3.0/setup.py

@@ -0,0 +1,37 @@
+# -*- coding: utf-8 -*-
+from setuptools import setup
+
+packages = \
+['clai', 'clai.ocr_drivers']
+
+package_data = \
+{'': ['*']}
+
+install_requires = \
+['PyAutoGUI>=0.9.53,<0.10.0',
+ 'PyWinCtl>=0.0.43,<0.0.44',
+ 'openai>=0.27.0,<0.28.0',
+ 'pytesseract>=0.3.10,<0.4.0']
+
+entry_points = \
+{'console_scripts': ['ai = clai:main', 'clai = clai:main']}
+
+setup_kwargs = {
+    'name': 'clai',
+    'version': '0.3.0',
+    'description': 'Command Line AI- this tool lets you call ChatGPT from a CLI',
+    'long_description': '# clai\nCommand Line AI- this tool lets you call ChatGPT from a CLI. \n\nI\'m designing this to be used in conjunction with a fork of [shin][shin], which will allow you\nto call `clai` from any textbox in your computer. Finally, ChatGPT everywhere!\n\nThe long-term vision for this project is to add support for extracting context. For example, it would\nread the current text on a window and be able to add to it, or answer questions about it.\n\n_________________\n\n[![PyPI version](https://badge.fury.io/py/clai.svg)](http://badge.fury.io/py/clai)\n[![Test Status](https://github.com/apockill/clai/workflows/Test/badge.svg?branch=main)](https://github.com/apockill/clai/actions?query=workflow%3ATest)\n[![Lint Status](https://github.com/apockill/clai/workflows/Lint/badge.svg?branch=main)](https://github.com/apockill/clai/actions?query=workflow%3ALint)\n[![codecov](https://codecov.io/gh/apockill/clai/branch/main/graph/badge.svg)](https://codecov.io/gh/apockill/clai)\n[![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)\n[![Imports: isort](https://img.shields.io/badge/%20imports-isort-%231674b1?style=flat&labelColor=ef8336)](https://timothycrosley.github.io/isort/)\n_________________\n\n[Read Latest Documentation](https://apockill.github.io/clai/) - [Browse GitHub Code Repository](https://github.com/apockill/clai/)\n_________________\n\n## Installation\n\n1. The recommended installation method is to use `pipx`, via\n    ```bash\n    pipx install clai\n    ```\n   Optionally, install `tesseract` so that `clai` can read the screen context and send that along with requests:\n   ```bash\n   sudo apt install tesseract-ocr scrot\n   ```\n1. Then go to [OpenAI] and create an API Key. Once it\'s generated, add the following to \n   your `~/.profile`:\n   ```bash\n   export OPENAI_API_TOKEN=<paste here>\n   ```\n\n1. The best way to use this tool is in conjunction with the tool [shin][shin], which allows you\n   to run arbitrary bash commands in any textbox in a linux computer, using ibus. To use \n   that, install \'shin\' via the fork above, then configure\n   it in your `~/.profile` to call `clai` by default:\n   ```bash\n   export SHIN_DEFAULT_COMMAND="clai"\n   ```\n1. Log out then log back in for the changes to take effect!\n\n[OpenAI]: https://platform.openai.com/account/api-keys\n\n## Usage\nInvoke the assistant with the format `clai <your prompt>`. For example:\n```\nclai Write an email saying I\'ll be late to work because I\'m working on commandline AIs\n```\n\n\n## Development\n\n### Installing python dependencies\n```shell\npoetry install\n```\n\n### Running Tests\n```shell\npytest .\n```\n\n### Formatting Code\n```shell\nbash .github/format.sh\n```\n\n### Linting\n```shell\nbash .github/check_lint.sh\n```\n\n[shin]: https://github.com/apockill/shin\n',
+    'author': 'apockill',
+    'author_email': 'apocthiel@gmail.com',
+    'maintainer': 'None',
+    'maintainer_email': 'None',
+    'url': 'None',
+    'packages': packages,
+    'package_data': package_data,
+    'install_requires': install_requires,
+    'entry_points': entry_points,
+    'python_requires': '>=3.8,<4.0',
+}
+
+
+setup(**setup_kwargs)

clai-0.1.0/clai/invoke.py

@@ -1,103 +0,0 @@
-from argparse import ArgumentParser
-
-from .api import initialize_api
-
-DEFAULT_ASSISTANT_ROLE = """
-You are an assistant that is capable of being called anywhere on a desktop computer. You
-may be called within the context of an email, a URL box, commandline, a text editor, or
-even word documents!
-
-Your role is to answer the users request as shortly and succinctly as possible. You
-will follow the following rules:
-
-When asked to write a command:
-1) Assume a linux desktop environment in a bash shell
-2) Feel free to use commonly installed unix tools
-
-When asked to write a command, code, formulas, or any one-line response task:
-1) NEVER WRITE EXPLANATIONS. Only include the command/code/etc, ready to be run
-2) NEVER WRITE USAGE INSTRUCTIONS. 
-3) Remember, the text that you write will immediately be run, do not include code blocks
-4) If there is something that requires user input, such as a cell in a sheet or a
-   variable from the user, write it inside of brackets, like this: <INPUT DESCRIBER>,
-   where the insides of the bracket have an example of what is needed to be filled in.
-
-When asked to write long-form text content:
-1) Never ask for more information. If something is to be guessed, write it in template
-   format. For example, if asked to write an email use <Insert time here> when writing
-   the portion of an email that specifies something that was not included in the users
-   question.
-2) Only assume the content is long form if the user mentions email, or 'long message'.
-"""
-
-EXAMPLE_EMAIL = """
-Dear <Recipient's Name>,
-
-I hope this email finds you well. I am writing to request a meeting with you on <Date and Time>, and I would appreciate it if you could confirm your availability at your earliest convenience.
-
-The purpose of this meeting is to discuss <Purpose of the Meeting> with you. Specifically, I would like to <Agenda Item 1>, <Agenda Item 2>, and <Agenda Item 3>. The meeting will last approximately <Meeting Duration> and will take place at <Meeting Location>.
-
-Please let me know if this date and time work for you. If not, please suggest an alternative time that is convenient for you. Additionally, if there are any documents or information you would like me to review before the meeting, please let me know, and I will make sure to review them.
-
-I look forward to hearing from you soon.
-
-Best regards,
-
-<Your Name>
-"""  # noqa: E501
-
-EXAMPLE_REGEX = '=IFERROR(REGEXEXTRACT(<INPUT CELL HERE>, "[A-z0-9._%+-]+@[A-z0-9.-]+\.[A-z]{2,4}");"")'  # noqa
-
-EXAMPLE_PYTHON = """
-def fibonacci(n: int) -> Generator[int, None, None]:
-    a, b = 0, 1
-    for _ in range(n):
-        yield a
-        a, b = b, a + b
-"""
-CONTEXT = [
-    {"role": "system", "content": DEFAULT_ASSISTANT_ROLE},
-    {
-        "role": "user",
-        "content": "commandline search for files with the name 'bruh' in them",
-    },
-    {"role": "assistant", "content": "grep -rnw . -e 'bruh'"},
-    {
-        "role": "user",
-        "content": "email set up a meeting next week",
-    },
-    {"role": "assistant", "content": EXAMPLE_EMAIL},
-    {
-        "role": "user",
-        "content": "google sheets formula extracts an email from string of text",
-    },
-    {
-        "role": "assistant",
-        "content": EXAMPLE_REGEX,
-    },
-    {
-        "role": "user",
-        "content": "python fibonacci function in form of a generator",
-    },
-    {
-        "role": "assistant",
-        "content": EXAMPLE_PYTHON,
-    },
-]
-
-
-def invoke() -> None:
-    parser = ArgumentParser("CLAI- your own command line AI!")
-    parser.add_argument("prompt", type=str, nargs="+")
-    parser.add_argument("-e", "--engine", default="")
-    args = parser.parse_args()
-
-    openai = initialize_api()
-
-    prompt = " ".join(args.prompt)
-    response = openai.ChatCompletion.create(
-        model="gpt-3.5-turbo", messages=[*CONTEXT, {"role": "user", "content": prompt}]
-    )
-
-    best_response = response["choices"][0]["message"]["content"]
-    print(best_response.strip())

clai-0.1.0/setup.py

@@ -1,34 +0,0 @@
-# -*- coding: utf-8 -*-
-from setuptools import setup
-
-packages = \
-['clai']
-
-package_data = \
-{'': ['*']}
-
-install_requires = \
-['openai>=0.27.0,<0.28.0']
-
-entry_points = \
-{'console_scripts': ['ai = clai:invoke', 'clai = clai:invoke']}
-
-setup_kwargs = {
-    'name': 'clai',
-    'version': '0.1.0',
-    'description': 'Command Line AI- this tool lets you call ChatGPT from a CLI',
-    'long_description': "# clai\nCommand Line AI- this tool lets you call ChatGPT from a CLI. \n\nI'm designing this to be used in conjunction with a fork of [shin][shin], which will allow you\nto call `clai` from any textbox in your computer. Finally, chatGPT everywhere!\n\n[shin]: https://github.com/apockill/shin\n_________________\n\n[![PyPI version](https://badge.fury.io/py/clai.svg)](http://badge.fury.io/py/clai)\n[![Test Status](https://github.com/apockill/clai/workflows/Test/badge.svg?branch=main)](https://github.com/apockill/clai/actions?query=workflow%3ATest)\n[![Lint Status](https://github.com/apockill/clai/workflows/Lint/badge.svg?branch=main)](https://github.com/apockill/clai/actions?query=workflow%3ALint)\n[![codecov](https://codecov.io/gh/apockill/clai/branch/main/graph/badge.svg)](https://codecov.io/gh/apockill/clai)\n[![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)\n[![Imports: isort](https://img.shields.io/badge/%20imports-isort-%231674b1?style=flat&labelColor=ef8336)](https://timothycrosley.github.io/isort/)\n_________________\n\n[Read Latest Documentation](https://apockill.github.io/clai/) - [Browse GitHub Code Repository](https://github.com/apockill/clai/)\n_________________\n\n## Installation\n1. The recommended installation method is to use `pipx`, via\n    ```\n    pipx install clai\n    ```\n\n1. Then go to [OpenAI] and create an API Key. Once it's generated, add the following to \n   your `.bashrc`:\n   ```bash\n   export OPENAI_API_TOKEN=<paste here>\n   ```\n\n[OpenAI]: https://platform.openai.com/account/api-keys\n\n## Usage\nInvoke the assistant with the format `clai <your prompt>`. For example:\n```\nclai Write an email saying I'll be late to work because I'm working on commandline AIs\n```\n\n## Development\n\n### Installing python dependencies\n```shell\npoetry install\n```\n\n### Running Tests\n```shell\npytest .\n```\n\n### Formatting Code\n```shell\nbash .github/format.sh\n```\n\n### Linting\n```shell\nbash .github/check_lint.sh\n```\n\n# TODO: remove references to tests?\n",
-    'author': 'apockill',
-    'author_email': 'apocthiel@gmail.com',
-    'maintainer': 'None',
-    'maintainer_email': 'None',
-    'url': 'None',
-    'packages': packages,
-    'package_data': package_data,
-    'install_requires': install_requires,
-    'entry_points': entry_points,
-    'python_requires': '>=3.8,<4.0',
-}
-
-
-setup(**setup_kwargs)