(feat:connectors) abstracting auth, base class

application/parser/connectors/connector_creator.py

@@ -1,30 +1,35 @@
 from application.parser.connectors.google_drive.loader import GoogleDriveLoader
+from application.parser.connectors.google_drive.auth import GoogleDriveAuth
 
 
 class ConnectorCreator:
     """
-    Factory class for creating external knowledge base connectors.
-    
+    Factory class for creating external knowledge base connectors and auth providers.
+
     These are different from remote loaders as they typically require
     authentication and connect to external document storage systems.
     """
-    
+
     connectors = {
         "google_drive": GoogleDriveLoader,
     }
 
+    auth_providers = {
+        "google_drive": GoogleDriveAuth,
+    }
+
     @classmethod
     def create_connector(cls, connector_type, *args, **kwargs):
         """
         Create a connector instance for the specified type.
-        
+
         Args:
             connector_type: Type of connector to create (e.g., 'google_drive')
             *args, **kwargs: Arguments to pass to the connector constructor
-            
+
         Returns:
             Connector instance
-            
+
         Raises:
             ValueError: If connector type is not supported
         """
@@ -33,11 +38,30 @@ class ConnectorCreator:
             raise ValueError(f"No connector class found for type {connector_type}")
         return connector_class(*args, **kwargs)
 
+    @classmethod
+    def create_auth(cls, connector_type):
+        """
+        Create an auth provider instance for the specified connector type.
+
+        Args:
+            connector_type: Type of connector auth to create (e.g., 'google_drive')
+
+        Returns:
+            Auth provider instance
+
+        Raises:
+            ValueError: If connector type is not supported for auth
+        """
+        auth_class = cls.auth_providers.get(connector_type.lower())
+        if not auth_class:
+            raise ValueError(f"No auth class found for type {connector_type}")
+        return auth_class()
+
     @classmethod
     def get_supported_connectors(cls):
         """
         Get list of supported connector types.
-        
+
         Returns:
             List of supported connector type strings
         """
@@ -47,10 +71,10 @@ class ConnectorCreator:
     def is_supported(cls, connector_type):
         """
         Check if a connector type is supported.
-        
+
         Args:
             connector_type: Type of connector to check
-            
+
         Returns:
             True if supported, False otherwise
         """