[Zendesk] connector improvements

cohere-ai · Dec 22, 2023 · 3877e5f · 3877e5f
1 parent 2c74c45
commit 3877e5f
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/zendesk/provider/client.py b/zendesk/provider/client.py
@@ -8,6 +8,8 @@
 BEARER_PREFIX = "Bearer "
 DEFAULT_LIMIT = 20
 
+client = None
+
 
 class ZendeskClient:
     def __init__(self, email, domain, access_token, limit):
@@ -29,11 +31,13 @@ def search(self, query):
 
 
 def get_client():
-    assert (email := app.config.get("EMAIL")), "ZENDESK_EMAIL must be set"
-    assert (domain := app.config.get("DOMAIN")), "ZENDESK_DOMAIN must be set"
-    assert (token := app.config.get("API_TOKEN")), "ZENDESK_API_TOKEN must be set"
-    limit = app.config.get("SEARCH_LIMIT", DEFAULT_LIMIT)
-
-    client = ZendeskClient(email, domain, token, limit)
+    global client
+    if client is None:
+        assert (email := app.config.get("EMAIL")), "ZENDESK_EMAIL must be set"
+        assert (domain := app.config.get("DOMAIN")), "ZENDESK_DOMAIN must be set"
+        assert (token := app.config.get("API_TOKEN")), "ZENDESK_API_TOKEN must be set"
+        limit = app.config.get("SEARCH_LIMIT", DEFAULT_LIMIT)
+
+        client = ZendeskClient(email, domain, token, limit)
 
     return client