highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-09-20 21:33:56 +02:00
Code Issues Packages Projects Releases Wiki Activity

Remove unnecessary log & Add LoadConnector

Browse Source
This commit is contained in:
Skylar Kesselring
2024-10-24 11:38:29 -04:00
parent 85b56e39c9
commit 87b5975091
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
backend/danswer/connectors/freshdesk/connector.py
View File

@@ -5,14 +5,14 @@ from typing import Any, List, Optional
from bs4 import BeautifulSoup # Add this import for HTML parsing from bs4 import BeautifulSoup # Add this import for HTML parsing
from danswer.configs.app_configs import INDEX_BATCH_SIZE from danswer.configs.app_configs import INDEX_BATCH_SIZE
from danswer.configs.constants import DocumentSource from danswer.configs.constants import DocumentSource
from danswer.connectors.interfaces import GenerateDocumentsOutput, PollConnector from danswer.connectors.interfaces import GenerateDocumentsOutput, PollConnector, LoadConnector
from danswer.connectors.models import ConnectorMissingCredentialError, Document, Section from danswer.connectors.models import ConnectorMissingCredentialError, Document, Section
from danswer.utils.logger import setup_logger from danswer.utils.logger import setup_logger
logger = setup_logger() logger = setup_logger()
class FreshdeskConnector(PollConnector): class FreshdeskConnector(PollConnector, LoadConnector):
def __init__(self, batch_size: int = INDEX_BATCH_SIZE) -> None: def __init__(self, batch_size: int = INDEX_BATCH_SIZE) -> None:
self.batch_size = batch_size self.batch_size = batch_size
@@ -37,7 +37,6 @@ class FreshdeskConnector(PollConnector):
return soup.get_text() return soup.get_text()
def load_credentials(self, credentials: dict[str, Any]) -> Optional[dict[str, Any]]: def load_credentials(self, credentials: dict[str, Any]) -> Optional[dict[str, Any]]:
logger.info("Loading credentials")
self.api_key = credentials.get("freshdesk_api_key") self.api_key = credentials.get("freshdesk_api_key")
self.domain = credentials.get("freshdesk_domain") self.domain = credentials.get("freshdesk_domain")
self.password = credentials.get("freshdesk_password") self.password = credentials.get("freshdesk_password")