Merge pull request #190 from intuit/stationeros-patch-15

Update constants.py to externalise TTL
intuit · Nov 8, 2023 · d7e371f · d7e371f
2 parents 2c88af2 + 0f4acec
commit d7e371f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/labs/TrapheusAI/constants.py b/labs/TrapheusAI/constants.py
@@ -5,4 +5,5 @@
 GRAPH_DEFAULT_COLOR="cyan"
 GRAPH_SELECTED_COLOR="red"
 GRAPH_HEIGHT=600
-GRAPH_WIDTH="100%"
+GRAPH_WIDTH="100%"
+TTL=3600
diff --git a/labs/TrapheusAI/providers/github_provider.py b/labs/TrapheusAI/providers/github_provider.py
@@ -22,7 +22,7 @@ def connect(self, input: str):
         return request_headers, request_params
 
 
-    @streamlit.cache_data(ttl=3600)
+    @streamlit.cache_data(ttl=constants.TTL)
     def query(_self, input: str):
         request_headers, request_params = _self.connect(input)
         response = requests.get(constants.GITHUB_END_POINT + constants.GITHUB_SEARCH_PATH, params=request_params,
@@ -31,7 +31,7 @@ def query(_self, input: str):
         response_body = response.json()
         return response_body
 
-    @streamlit.cache_data(ttl=3600)
+    @streamlit.cache_data(ttl=constants.TTL)
     def extract_data(_self, input: str):
         formatted_results = []
         data = _self.query(input).get("items", [])