From 21d8fc39daab945d2a99b145d3d7513a20f8859d Mon Sep 17 00:00:00 2001
From: Kenneth Reitz <me@kennethreitz.org>
Date: Tue, 30 Jan 2024 10:17:06 -0500
Subject: [PATCH] Add additional imports to gen-model command

---
 Makefile | 1 +
 1 file changed, 1 insertion(+)

diff --git a/Makefile b/Makefile
index f5c783b..c3ce9f3 100644
--- a/Makefile
+++ b/Makefile
@@ -14,6 +14,7 @@ gen-model: fetch-v2-schema
 	--input v2.json \
 	--collapse-root-models \
 	--output neon_client/schema.py \
+	--additional-imports datetime.datetime,pydantic.dataclasses.dataclass \
 	--use-standard-collections \
 	--output-model-type dataclasses.dataclass \
 	# --input-file-type openapi \