From 513cac7dafbc187d03cfc02730f74c9b9aa3caa6 Mon Sep 17 00:00:00 2001
From: Dixing Xu <i@dex.moe>
Date: Fri, 22 Nov 2024 18:08:53 +0800
Subject: [PATCH] :wrench: Add Makefile

---
 Makefile | 50 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 50 insertions(+)
 create mode 100644 Makefile

diff --git a/Makefile b/Makefile
new file mode 100644
index 0000000..8d61233
--- /dev/null
+++ b/Makefile
@@ -0,0 +1,50 @@
+.PHONY: docker docker-build docker-run install clean
+
+# Docker image name
+IMAGE_NAME = aide
+
+# Python version and venv
+PYTHON = python3.10
+VENV_NAME = .venv
+
+# Default directories for logs and workspaces
+WORKSPACE_BASE ?= $(PWD)/workspaces
+LOGS_DIR ?= $(PWD)/logs
+
+# Virtual environment installation
+install:
+	@echo "Creating virtual environment..."
+	@$(PYTHON) -m venv $(VENV_NAME)
+	@echo "Installing dependencies..."
+	@. $(VENV_NAME)/bin/activate && \
+		pip install --upgrade pip && \
+		pip install -r requirements.txt && \
+		pip install -e .
+	@echo "Installation complete. Activate the virtual environment with: source $(VENV_NAME)/bin/activate"
+
+# Docker commands combined
+docker: docker-build docker-run
+
+# Build Docker image
+docker-build:
+	docker build -t $(IMAGE_NAME) .
+
+# Run Docker container
+docker-run:
+	@mkdir -p "$(LOGS_DIR)" "$(WORKSPACE_BASE)"
+	docker run -it --rm \
+		-v "$(LOGS_DIR):/app/logs" \
+		-v "$(WORKSPACE_BASE):/app/workspaces" \
+		-v "$(PWD)/aide/example_tasks:/app/data" \
+		-e OPENAI_API_KEY="$(OPENAI_API_KEY)" \
+		$(IMAGE_NAME) \
+		data_dir=/app/data/house_prices \
+		goal="Predict the sales price for each house" \
+		eval="Use the RMSE metric between the logarithm of the predicted and observed values."
+
+# Clean up
+clean:
+	@echo "Cleaning up..."
+	rm -rf $(VENV_NAME)
+	rm -rf workspaces/* logs/*
+	docker rmi $(IMAGE_NAME) || true