include type checker

sud0su · Jan 9, 2024 · 7073879 · 7073879
1 parent 1aeb8c4
commit 7073879
Show file tree

Hide file tree

Showing 5 changed files with 62 additions and 2 deletions.
diff --git 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
@@ -4,6 +4,7 @@
 
 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 @@
 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
diff --git 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
@@ -0,0 +1,3 @@
+{
+  "exclude": ["image_generation.py"]
+}