github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-08 22:42:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update to the latest Psychic python library version (#6804)

Browse Source 
Update the Psychic document loader to use the latest `psychicapi` python
library version: `0.8.0`
This commit is contained in:
Ayan Bandyopadhyay
2023-06-27 22:26:38 -07:00
committed by GitHub
parent f3d178f600
commit f92ccf70fd
4 changed files with 53 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/unit_tests/document_loaders/test_psychic.py
View File

@@ -23,7 +23,7 @@ def mock_connector_id(): # type: ignore
class TestPsychicLoader:
MOCK_API_KEY = "api_key"
MOCK_CONNECTOR_ID = "notion"
MOCK_CONNECTION_ID = "connection_id"
MOCK_ACCOUNT_ID = "account_id"
def test_psychic_loader_initialization(
self, mock_psychic: MagicMock, mock_connector_id: MagicMock
@@ -31,17 +31,21 @@ class TestPsychicLoader:
PsychicLoader(
api_key=self.MOCK_API_KEY,
connector_id=self.MOCK_CONNECTOR_ID,
connection_id=self.MOCK_CONNECTION_ID,
account_id=self.MOCK_ACCOUNT_ID,
)
mock_psychic.assert_called_once_with(secret_key=self.MOCK_API_KEY)
mock_connector_id.assert_called_once_with(self.MOCK_CONNECTOR_ID)
def test_psychic_loader_load_data(self, mock_psychic: MagicMock) -> None:
mock_psychic.get_documents.return_value = [
mock_get_documents_response = MagicMock()
mock_get_documents_response.documents = [
self._get_mock_document("123"),
self._get_mock_document("456"),
]
mock_get_documents_response.next_page_cursor = None
mock_psychic.get_documents.return_value = mock_get_documents_response
psychic_loader = self._get_mock_psychic_loader(mock_psychic)
@@ -57,7 +61,7 @@ class TestPsychicLoader:
psychic_loader = PsychicLoader(
api_key=self.MOCK_API_KEY,
connector_id=self.MOCK_CONNECTOR_ID,
connection_id=self.MOCK_CONNECTION_ID,
account_id=self.MOCK_ACCOUNT_ID,
)
psychic_loader.psychic = mock_psychic
return psychic_loader