mirror of
https://github.com/QuivrHQ/quivr.git
synced 2024-12-20 12:51:33 +03:00
0c0545b7ae
# Description Please include a summary of the changes and the related issue. Please also include relevant motivation and context. ## Checklist before requesting a review Please delete options that are not relevant. - [ ] My code follows the style guidelines of this project - [ ] I have performed a self-review of my code - [ ] I have commented hard-to-understand areas - [ ] I have ideally added tests that prove my fix is effective or that my feature works - [ ] New and existing unit tests pass locally with my changes - [ ] Any dependent changes have been merged ## Screenshots (if appropriate):
42 lines
1.5 KiB
Python
42 lines
1.5 KiB
Python
from logger import get_logger
|
|
from modules.brain.entity.integration_brain import IntegrationEntity
|
|
from modules.brain.repository.integration_brains import IntegrationBrain
|
|
from modules.knowledge.repository.knowledge_interface import KnowledgeInterface
|
|
from modules.knowledge.service.knowledge_service import KnowledgeService
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
|
|
class SQLConnector(IntegrationBrain):
|
|
"""A class to interact with an SQL database"""
|
|
|
|
credentials: dict[str, str] = None
|
|
integration_details: IntegrationEntity = None
|
|
brain_id: str = None
|
|
user_id: str = None
|
|
knowledge_service: KnowledgeInterface
|
|
|
|
def __init__(self, brain_id: str, user_id: str):
|
|
super().__init__()
|
|
self.brain_id = brain_id
|
|
self.user_id = user_id
|
|
self._load_credentials()
|
|
self.knowledge_service = KnowledgeService()
|
|
|
|
def _load_credentials(self) -> dict[str, str]:
|
|
"""Load the Notion credentials"""
|
|
self.integration_details = self.get_integration_brain(
|
|
self.brain_id
|
|
)
|
|
if self.credentials is None:
|
|
logger.info("Loading Notion credentials")
|
|
self.integration_details.credentials = {
|
|
"uri": self.integration_details.settings.get("uri", "")
|
|
}
|
|
self.update_integration_brain(
|
|
self.brain_id, self.user_id, self.integration_details
|
|
)
|
|
self.credentials = self.integration_details.credentials
|
|
else: # pragma: no cover
|
|
self.credentials = self.integration_details.credentials
|