Welcome to this website!
\nThe current time is: .
\n{% endblock %}" + }, + { + "role": "user", + "content": "[FILE-STRUCTURE]: A very simple CLI which returns the user's IP address." + }, + { + "role": "assistant", + "content": "cli.py\nREADME.md\ntests.py\n" + }, + { + "role": "user", + "content": "[FILE-CONTENT]: cli.py" + }, + { + "role": "assistant", + "content": "\"\"\"Get the IP address of the client.\"\"\"\n\nimport typer\nimport requests\n\napp = typer.Typer()\n\n@app.command()\ndef main():\n \"\"\"Shows the IP address of the client.\"\"\"\n print(requests.get('https://checkip.amazonaws.com', timeout=5).text.strip())\n\nif __name__ == '__main__':\n typer.run(main)\n" + }, + { + "role": "user", + "content": "[FILE-CONTENT]: README.md" + }, + { + "role": "assistant", + "content": "# `GetIP`\nA very simple CLI app that displays the user's IP address.\n\n## Get started\nIt's very simple to run this app. Just run the following command in the terminal:\n\n```bash\npython cli.py\n```\n\nExample output:\n```\n123.123.34.34\n```\n" + }, + { + "role": "user", + "content": "[FILE-CONTENT]: tests.py" + }, + { + "role": "assistant", + "content": "\"\"\"LUNA Tests.\"\"\"\n\nimport os\nimport requests\nimport subprocess\n\nclass Tester:\n \"\"\"Tests if the program is working as expected.\"\"\"\n\n def __init__(self):\n \"\"\"Runs the tests.\"\"\"\n\n # check the print output of cli.app.main()\n os.chdir(os.path.dirname(__file__))\n output = subprocess.check_output(['python', 'cli.py']).decode('utf8')\n\n ip_addr = requests.get('https://checkip.amazonaws.com', timeout=5).text.strip()\n assert ip_addr in output\n\n def stop(self):\n \"\"\"Runs after the tests are done, whether they have been successful or not.\"\"\"\n # no need to stop anything\n" + } +] \ No newline at end of file diff --git a/luna/prompts.py b/luna/prompts.py new file mode 100644 index 0000000..36eb851 --- /dev/null +++ b/luna/prompts.py @@ -0,0 +1,76 @@ +"""The prompt module contains the demo "conversation" messages used to provide an example for the +AI of how to respond to the user.""" + +import os +import json + +messages = [ + {'role': 'system', 'content': """You program Python projects by providing every file's +contents, a documentation in the form of a Markdown file as well as a list of all files. + +It is really important that also tests are provided, which check the functionality of the +project! The tests should check the functionality of the project, and should be created at the end. +The first file content to provide is the README.md file, which should contain a description of the +project, as well as a simple documentation on how to run the project, as well as how it works. + +For example, for a web app, you should not only provide the commands to run the project, but also +the URL paths and what they do. + +Keep in mind to also use a nice CSS design if the project is a web app! +Completely avoid using APIs which require an API key, as this would make it impossible to run the project! +Whenever possible, use APIs which are free to use and don't require an API key instead! +"""} +] + +def generate_messages(): + """Generates the messages for the AI "conversation". + """ + + for example in os.listdir('luna/training'): + example_path = f'luna/training/{example}' + + with open(f'{example_path}/_prompt.luna.txt', encoding='utf8') as prompt_file: + prompt = prompt_file.read() + + messages.append({ + 'role': 'user', + 'content': f'[FILE-STRUCTURE]: {prompt}' + }) + + contents = {} + + for root, _, files in os.walk(example_path): + for file_ in files: + file_path = os.path.join(root, file_)\ +.replace('\\', '/').replace(example_path + '/', '') + + if '__pycache__' in file_path or '.luna.' in file_path: + continue + + contents[file_path] = open(os.path.join(root, file_), encoding='utf8').read() + + structure = '' + + for file_path in contents: + structure += f'{file_path}\n' + + messages.append({ + 'role': 'assistant', + 'content': structure + }) + + for file_path, content in contents.items(): + messages.append({ + 'role': 'user', + 'content': f'[FILE-CONTENT]: {file_path}' + }) + + messages.append({ + 'role': 'assistant', + 'content': content + }) + + return messages + +if __name__ == '__main__': + json.dump(generate_messages(), open('luna/prompts.json', 'w', encoding='utf8'), indent=4) diff --git a/luna/testing.py b/luna/testing.py new file mode 100644 index 0000000..144f37d --- /dev/null +++ b/luna/testing.py @@ -0,0 +1,32 @@ +"""Provides a function to run tests in a module.""" + +import sys +import traceback + +def run_tests(tests_module): + """Runs the tests in the given module.""" + + success = False + + tester = None + + try: + tester = tests_module.Tester() + + # pylint: disable=broad-except + except Exception: + print('-----BEGIN LUNA ERROR-----') + traceback.print_exc() + print('-----END LUNA ERROR-----') + + else: + success = True + print('-----LUNA SUCCESS-----') + + finally: + try: + tester.stop() + except AttributeError: + pass + + sys.exit(int(not success)) diff --git a/luna/tokens.py b/luna/tokens.py new file mode 100644 index 0000000..94d20d4 --- /dev/null +++ b/luna/tokens.py @@ -0,0 +1,31 @@ +import tiktoken + +def num_tokens_from_messages(messages, model="gpt-3.5-turbo-0301"): + """Returns the number of tokens used by a list of messages.""" + try: + encoding = tiktoken.encoding_for_model(model) + except KeyError: + encoding = tiktoken.get_encoding("cl100k_base") + if model == "gpt-3.5-turbo": + return num_tokens_from_messages(messages, model="gpt-3.5-turbo-0301") + elif model == "gpt-4": + return num_tokens_from_messages(messages, model="gpt-4-0314") + elif model == "gpt-3.5-turbo-0301": + tokens_per_message = 4 # every message follows <|start|>{role/name}\n{content}<|end|>\n + tokens_per_name = -1 # if there's a name, the role is omitted + elif model == "gpt-4-0314": + tokens_per_message = 3 + tokens_per_name = 1 + else: + raise NotImplementedError(f"""num_tokens_from_messages() is not implemented for model {model}. See https://github.com/openai/openai-python/blob/main/chatml.md for information on how messages are converted to tokens.""") + num_tokens = 0 + for message in messages: + num_tokens += tokens_per_message + for key, value in message.items(): + num_tokens += len(encoding.encode(value)) + if key == "name": + num_tokens += tokens_per_name + num_tokens += 3 # every reply is primed with <|start|>assistant<|message|> + return num_tokens + +print(num_tokens_from_messages(messages)) diff --git a/luna/training/clock/README.md b/luna/training/clock/README.md new file mode 100644 index 0000000..f0ee2b3 --- /dev/null +++ b/luna/training/clock/README.md @@ -0,0 +1,12 @@ +# SimpleClock +A simple web app which displays the current time. + +## Get started +To get started, follow these steps: + +1. Run the app using +```bash +python main.py +``` +2. Open the app in your browser at http://localhost:5000 +3. Enjoy! \ No newline at end of file diff --git a/luna/training/clock/_prompt.luna.txt b/luna/training/clock/_prompt.luna.txt new file mode 100644 index 0000000..c28ce3f --- /dev/null +++ b/luna/training/clock/_prompt.luna.txt @@ -0,0 +1 @@ +A simple website called SimpleClock. It should display the current time on /. \ No newline at end of file diff --git a/luna/training/clock/main.py b/luna/training/clock/main.py new file mode 100644 index 0000000..f74eab1 --- /dev/null +++ b/luna/training/clock/main.py @@ -0,0 +1,12 @@ +"""Web server start module.""" + +import sys + +from clock import create_app + +def run(port=sys.argv[1] if len(sys.argv) > 1 else 5000): + """Starts the web server.""" + create_app().run(port=port) + +if __name__ == '__main__': + run() diff --git a/luna/training/clock/simpleclock/__init__.py b/luna/training/clock/simpleclock/__init__.py new file mode 100644 index 0000000..a95c280 --- /dev/null +++ b/luna/training/clock/simpleclock/__init__.py @@ -0,0 +1,13 @@ +import flask + +def create_app(): + """Returns the Flask app.""" + + app = flask.Flask(__name__) + + @app.route('/') + def index(): + """Display the current time.""" + return flask.render_template('index.html') + + return app diff --git a/luna/training/clock/simpleclock/static/index.js b/luna/training/clock/simpleclock/static/index.js new file mode 100644 index 0000000..43edc5c --- /dev/null +++ b/luna/training/clock/simpleclock/static/index.js @@ -0,0 +1,9 @@ +// use momentjs to display the time, nicely formatted. +// updates every second + +function update() { + time.innerHTML = moment().format('h:mm:ss a'); + } + +update(); +setInterval(update, 1000); diff --git a/luna/training/clock/simpleclock/static/style.css b/luna/training/clock/simpleclock/static/style.css new file mode 100644 index 0000000..0407081 --- /dev/null +++ b/luna/training/clock/simpleclock/static/style.css @@ -0,0 +1,27 @@ +body { + font-size: 1.2rem; + scroll-behavior: smooth; +} + +* { + box-sizing: border-box; + font-family: sans-serif; +} + +main { + max-width: 800px; + margin: 0 auto; + padding: 1rem; +} + +h1 { + font-size: 2rem; +} + +p { + font-size: 1.2rem; +} + +#time { + font-weight: bold; +} diff --git a/luna/training/clock/simpleclock/templates/base.html b/luna/training/clock/simpleclock/templates/base.html new file mode 100644 index 0000000..7a55c95 --- /dev/null +++ b/luna/training/clock/simpleclock/templates/base.html @@ -0,0 +1,15 @@ + + + +Welcome to this website!
+The current time is: .
+{% endblock %} \ No newline at end of file diff --git a/luna/training/clock/tests.py b/luna/training/clock/tests.py new file mode 100644 index 0000000..b68f78e --- /dev/null +++ b/luna/training/clock/tests.py @@ -0,0 +1,26 @@ +"""LUNA Tests.""" + +import time +import multiprocessing +import requests + +class Tester: + """Tests if the program is working as expected.""" + + def __init__(self): + """Runs the tests.""" + # pylint: disable=import-outside-toplevel + import main + + self.server = multiprocessing.Process(target=main.run, args=(5000,)) + self.server.start() + + time.sleep(1) + print('Server started!') + + resp = requests.get('http://localhost:5000/', timeout=5) + assert 'The current time is:' in resp.text + + def stop(self): + """Runs after the tests are done, whether they have been successful or not.""" + self.server.terminate() diff --git a/luna/training/getip/README.md b/luna/training/getip/README.md new file mode 100644 index 0000000..db89df2 --- /dev/null +++ b/luna/training/getip/README.md @@ -0,0 +1,14 @@ +# `GetIP` +A very simple CLI app that displays the user's IP address. + +## Get started +It's very simple to run this app. Just run the following command in the terminal: + +```bash +python cli.py +``` + +Example output: +``` +123.123.34.34 +``` diff --git a/luna/training/getip/_prompt.luna.txt b/luna/training/getip/_prompt.luna.txt new file mode 100644 index 0000000..0f39659 --- /dev/null +++ b/luna/training/getip/_prompt.luna.txt @@ -0,0 +1 @@ +A very simple CLI which returns the user's IP address. \ No newline at end of file diff --git a/luna/training/getip/cli.py b/luna/training/getip/cli.py new file mode 100644 index 0000000..135cc36 --- /dev/null +++ b/luna/training/getip/cli.py @@ -0,0 +1,14 @@ +"""Get the IP address of the client.""" + +import typer +import requests + +app = typer.Typer() + +@app.command() +def main(): + """Shows the IP address of the client.""" + print(requests.get('https://checkip.amazonaws.com', timeout=5).text.strip()) + +if __name__ == '__main__': + typer.run(main) diff --git a/luna/training/getip/tests.py b/luna/training/getip/tests.py new file mode 100644 index 0000000..fa7682e --- /dev/null +++ b/luna/training/getip/tests.py @@ -0,0 +1,22 @@ +"""LUNA Tests.""" + +import os +import requests +import subprocess + +class Tester: + """Tests if the program is working as expected.""" + + def __init__(self): + """Runs the tests.""" + + # check the print output of cli.app.main() + os.chdir(os.path.dirname(__file__)) + output = subprocess.check_output(['python', 'cli.py']).decode('utf8') + + ip_addr = requests.get('https://checkip.amazonaws.com', timeout=5).text.strip() + assert ip_addr in output + + def stop(self): + """Runs after the tests are done, whether they have been successful or not.""" + # no need to stop anything