mirror of
https://github.com/tiennm99/DocsGPT.git
synced 2026-10-04 08:13:02 +00:00
Siddhant Rai
749eed3d0b
- Enhanced settings.py to include Confluence client ID and secret - Created ConfluenceAuth class for handling authentication with Confluence - Implemented ConfluenceLoader class for loading data from Confluence - Updated connector_creator.py to register Confluence as a connector - Added confluence.svg asset for UI representation - Modified ConnectorAuth component to support Confluence connection - Updated FilePicker component to include Confluence as a file source - Added localization support for Confluence in multiple languages (de, en, es, jp, ru, zh-TW, zh) - Enhanced Upload component to handle Confluence file selection - Updated ingestor types to include Confluence and its configuration
90 lines
2.8 KiB
Python
90 lines
2.8 KiB
Python
from application.parser.connectors.confluence.auth import ConfluenceAuth
|
|
from application.parser.connectors.confluence.loader import ConfluenceLoader
|
|
from application.parser.connectors.google_drive.auth import GoogleDriveAuth
|
|
from application.parser.connectors.google_drive.loader import GoogleDriveLoader
|
|
from application.parser.connectors.share_point.auth import SharePointAuth
|
|
from application.parser.connectors.share_point.loader import SharePointLoader
|
|
|
|
|
|
class ConnectorCreator:
|
|
"""
|
|
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 = {
|
|
"confluence": ConfluenceLoader,
|
|
"google_drive": GoogleDriveLoader,
|
|
"share_point": SharePointLoader,
|
|
}
|
|
|
|
auth_providers = {
|
|
"confluence": ConfluenceAuth,
|
|
"google_drive": GoogleDriveAuth,
|
|
"share_point": SharePointAuth,
|
|
}
|
|
|
|
@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
|
|
"""
|
|
connector_class = cls.connectors.get(connector_type.lower())
|
|
if not connector_class:
|
|
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
|
|
"""
|
|
return list(cls.connectors.keys())
|
|
|
|
@classmethod
|
|
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
|
|
"""
|
|
return connector_type.lower() in cls.connectors
|