From aefbdaa8939a1b671d77befb1c0622f217850781 Mon Sep 17 00:00:00 2001
From: Shuchang Zheng <wintonzheng0325@gmail.com>
Date: Thu, 14 Sep 2023 15:03:33 -0700
Subject: [PATCH] cache output for modelbit component (#64)

---
 wyvern/components/models/modelbit_component.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/wyvern/components/models/modelbit_component.py b/wyvern/components/models/modelbit_component.py
index 9b1ada1..39c3e5e 100644
--- a/wyvern/components/models/modelbit_component.py
+++ b/wyvern/components/models/modelbit_component.py
@@ -41,6 +41,7 @@ def __init__(
         name: Optional[str] = None,
         auth_token: Optional[str] = None,
         url: Optional[str] = None,
+        cache_output: bool = False,
     ) -> None:
         """
         Args:
@@ -52,7 +53,7 @@ def __init__(
         Raises:
             WyvernModelbitTokenMissingError: If the auth token is not provided.
         """
-        super().__init__(*upstreams, name=name)
+        super().__init__(*upstreams, name=name, cache_output=cache_output)
         self._auth_token = auth_token or self.AUTH_TOKEN
         self._modelbit_url = url or self.URL
         self.headers = {