From 7073879e6e4957df31049e8c5cb30b9b5f18710f Mon Sep 17 00:00:00 2001
From: Abi Raja <abimanyuraja@gmail.com>
Date: Mon, 8 Jan 2024 17:04:59 -0800
Subject: [PATCH] include type checker

---
 backend/README.md          |  4 +++
 backend/eval.py            |  3 ++-
 backend/poetry.lock        | 53 +++++++++++++++++++++++++++++++++++++-
 backend/pyproject.toml     |  1 +
 backend/pyrightconfig.json |  3 +++
 5 files changed, 62 insertions(+), 2 deletions(-)
 create mode 100644 backend/pyrightconfig.json

diff --git a/backend/README.md b/backend/README.md
index ee55816..155bf46 100644
--- a/backend/README.md
+++ b/backend/README.md
@@ -1,3 +1,7 @@
+# Run the type checker
+
+poetry run pyright
+
 # Run tests
 
 poetry run pytest
diff --git a/backend/eval.py b/backend/eval.py
index 60ef409..75641a3 100644
--- a/backend/eval.py
+++ b/backend/eval.py
@@ -4,6 +4,7 @@ from dotenv import load_dotenv
 
 from eval_config import EVALS_DIR
 from eval_utils import image_to_data_url
+from prompts.types import Stack
 
 load_dotenv()
 
@@ -15,7 +16,7 @@ import asyncio
 from utils import pprint_prompt
 
 
-async def generate_code_core(image_url: str, stack: str) -> str:
+async def generate_code_core(image_url: str, stack: Stack) -> str:
     prompt_messages = assemble_prompt(image_url, stack)
     openai_api_key = os.environ.get("OPENAI_API_KEY")
     openai_base_url = None
diff --git a/backend/poetry.lock b/backend/poetry.lock
index fbd0691..42a44ee 100644
--- a/backend/poetry.lock
+++ b/backend/poetry.lock
@@ -212,6 +212,21 @@ files = [
     {file = "iniconfig-2.0.0.tar.gz", hash = "sha256:2d91e135bf72d31a410b17c16da610a82cb55f6b0477d1a902134b24a455b8b3"},
 ]
 
+[[package]]
+name = "nodeenv"
+version = "1.8.0"
+description = "Node.js virtual environment builder"
+category = "dev"
+optional = false
+python-versions = ">=2.7,!=3.0.*,!=3.1.*,!=3.2.*,!=3.3.*,!=3.4.*,!=3.5.*,!=3.6.*"
+files = [
+    {file = "nodeenv-1.8.0-py2.py3-none-any.whl", hash = "sha256:df865724bb3c3adc86b3876fa209771517b0cfe596beff01a92700e0e8be4cec"},
+    {file = "nodeenv-1.8.0.tar.gz", hash = "sha256:d51e0c37e64fbf47d017feac3145cdbb58836d7eee8c6f6d3b6880c5456227d2"},
+]
+
+[package.dependencies]
+setuptools = "*"
+
 [[package]]
 name = "openai"
 version = "1.3.7"
@@ -317,6 +332,25 @@ typing-extensions = ">=4.2.0"
 dotenv = ["python-dotenv (>=0.10.4)"]
 email = ["email-validator (>=1.0.3)"]
 
+[[package]]
+name = "pyright"
+version = "1.1.345"
+description = "Command line wrapper for pyright"
+category = "dev"
+optional = false
+python-versions = ">=3.7"
+files = [
+    {file = "pyright-1.1.345-py3-none-any.whl", hash = "sha256:00891361baf58698aa660d9374823d65782823ceb4a65515ff5dd159b0d4d2b1"},
+    {file = "pyright-1.1.345.tar.gz", hash = "sha256:bb8c80671cdaeb913142b49642a741959f3fcd728c99814631c2bde3a7864938"},
+]
+
+[package.dependencies]
+nodeenv = ">=1.6.0"
+
+[package.extras]
+all = ["twine (>=3.4.1)"]
+dev = ["twine (>=3.4.1)"]
+
 [[package]]
 name = "pytest"
 version = "7.4.3"
@@ -355,6 +389,23 @@ files = [
 [package.extras]
 cli = ["click (>=5.0)"]
 
+[[package]]
+name = "setuptools"
+version = "69.0.3"
+description = "Easily download, build, install, upgrade, and uninstall Python packages"
+category = "dev"
+optional = false
+python-versions = ">=3.8"
+files = [
+    {file = "setuptools-69.0.3-py3-none-any.whl", hash = "sha256:385eb4edd9c9d5c17540511303e39a147ce2fc04bc55289c322b9e5904fe2c05"},
+    {file = "setuptools-69.0.3.tar.gz", hash = "sha256:be1af57fc409f93647f2e8e4573a142ed38724b8cdd389706a867bb4efcf1e78"},
+]
+
+[package.extras]
+docs = ["furo", "jaraco.packaging (>=9.3)", "jaraco.tidelift (>=1.4)", "pygments-github-lexers (==0.0.5)", "rst.linker (>=1.9)", "sphinx (<7.2.5)", "sphinx (>=3.5)", "sphinx-favicon", "sphinx-inline-tabs", "sphinx-lint", "sphinx-notfound-page (>=1,<2)", "sphinx-reredirects", "sphinxcontrib-towncrier"]
+testing = ["build[virtualenv]", "filelock (>=3.4.0)", "flake8-2020", "ini2toml[lite] (>=0.9)", "jaraco.develop (>=7.21)", "jaraco.envs (>=2.2)", "jaraco.path (>=3.2.0)", "pip (>=19.1)", "pytest (>=6)", "pytest-black (>=0.3.7)", "pytest-checkdocs (>=2.4)", "pytest-cov", "pytest-enabler (>=2.2)", "pytest-mypy (>=0.9.1)", "pytest-perf", "pytest-ruff", "pytest-timeout", "pytest-xdist", "tomli-w (>=1.0.0)", "virtualenv (>=13.0.0)", "wheel"]
+testing-integration = ["build[virtualenv] (>=1.0.3)", "filelock (>=3.4.0)", "jaraco.envs (>=2.2)", "jaraco.path (>=3.2.0)", "packaging (>=23.1)", "pytest", "pytest-enabler", "pytest-xdist", "tomli", "virtualenv (>=13.0.0)", "wheel"]
+
 [[package]]
 name = "sniffio"
 version = "1.3.0"
@@ -547,4 +598,4 @@ files = [
 [metadata]
 lock-version = "2.0"
 python-versions = "^3.10"
-content-hash = "92aaa379ec66ff6146122266c895f277835b23254e59c5ae21f0d0cae87d3a11"
+content-hash = "00950515c8d7f2061811ce196e4aa47204eb98ea078bbeb875548072dbfa38b1"
diff --git a/backend/pyproject.toml b/backend/pyproject.toml
index 216a020..bdaaed1 100644
--- a/backend/pyproject.toml
+++ b/backend/pyproject.toml
@@ -17,6 +17,7 @@ httpx = "^0.25.1"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^7.4.3"
+pyright = "^1.1.345"
 
 [build-system]
 requires = ["poetry-core"]
diff --git a/backend/pyrightconfig.json b/backend/pyrightconfig.json
new file mode 100644
index 0000000..6e475af
--- /dev/null
+++ b/backend/pyrightconfig.json
@@ -0,0 +1,3 @@
+{
+  "exclude": ["image_generation.py"]
+}