From f494f226ebbabcd035c5213ff86ec903818560f0 Mon Sep 17 00:00:00 2001
From: Josh Bickett <josh@bickett.net>
Date: Wed, 24 Jan 2024 09:01:23 -0800
Subject: [PATCH] Add new mode to `require_api_key`

---
 operate/config.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/operate/config.py b/operate/config.py
index 581af043..00228686 100644
--- a/operate/config.py
+++ b/operate/config.py
@@ -79,7 +79,10 @@ def validation(self, model, voice_mode):
         self.require_api_key(
             "OPENAI_API_KEY",
             "OpenAI API key",
-            model == "gpt-4" or voice_mode or model == "gpt-4-with-som",
+            model == "gpt-4"
+            or voice_mode
+            or model == "gpt-4-with-som"
+            or model == "gpt-4-with-ocr",
         )
         self.require_api_key(
             "GOOGLE_API_KEY", "Google API key", model == "gemini-pro-vision"