Confluence connector improvements

cohere-ai · Dec 4, 2023 · 086e366 · 086e366
1 parent 3e632fc
commit 086e366
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/confluence/provider/client.py b/confluence/provider/client.py
@@ -10,12 +10,12 @@
 
 
 class ConfluenceClient:
-    def __init__(self, url, user, password, space, search_limit=10):
+    def __init__(self, url, user, api_token, space, search_limit=10):
         try:
             self.confluence = Confluence(
                 url=url,
                 username=user,
-                password=password,
+                password=api_token,
             )
             self.space = space
             self.search_limit = search_limit
@@ -77,9 +77,9 @@ def get_client():
 
     assert (url := app.config.get("PRODUCT_URL")), "CONFLUENCE_PRODUCT_URL must be set"
     assert (user := app.config.get("USER")), "CONFLUENCE_USER must be set"
-    assert (password := app.config.get("API_TOKEN")), "CONFLUENCE_API_TOKEN must be set"
+    assert (api_token := app.config.get("API_TOKEN")), "CONFLUENCE_API_TOKEN must be set"
     assert (space := app.config.get("SPACE_NAME")), "CONFLUENCE_SPACE_NAME must be set"
     search_limit = app.config.get("SEARCH_LIMIT", 10)
-    client = ConfluenceClient(url, user, password, space, search_limit)
+    client = ConfluenceClient(url, user, api_token, space, search_limit)
 
     return client