mirror of
https://github.com/Mintplex-Labs/langchain-python.git
synced 2026-07-19 21:33:31 -04:00
Leonid Ganeline
1c81883d42
This PR targets the `API Reference` documentation.
- Several classes and functions missed `docstrings`. These docstrings
were created.
- In several places this
```
except ImportError:
raise ValueError(
```
was replaced to
```
except ImportError:
raise ImportError(
```
64 lines
2.6 KiB
Python
64 lines
2.6 KiB
Python
"""Loader that fetches data from Modern Treasury"""
|
|
import json
|
|
import urllib.request
|
|
from base64 import b64encode
|
|
from typing import List, Optional
|
|
|
|
from langchain.docstore.document import Document
|
|
from langchain.document_loaders.base import BaseLoader
|
|
from langchain.utils import get_from_env, stringify_value
|
|
|
|
MODERN_TREASURY_ENDPOINTS = {
|
|
"payment_orders": "https://app.moderntreasury.com/api/payment_orders",
|
|
"expected_payments": "https://app.moderntreasury.com/api/expected_payments",
|
|
"returns": "https://app.moderntreasury.com/api/returns",
|
|
"incoming_payment_details": "https://app.moderntreasury.com/api/\
|
|
incoming_payment_details",
|
|
"counterparties": "https://app.moderntreasury.com/api/counterparties",
|
|
"internal_accounts": "https://app.moderntreasury.com/api/internal_accounts",
|
|
"external_accounts": "https://app.moderntreasury.com/api/external_accounts",
|
|
"transactions": "https://app.moderntreasury.com/api/transactions",
|
|
"ledgers": "https://app.moderntreasury.com/api/ledgers",
|
|
"ledger_accounts": "https://app.moderntreasury.com/api/ledger_accounts",
|
|
"ledger_transactions": "https://app.moderntreasury.com/api/ledger_transactions",
|
|
"events": "https://app.moderntreasury.com/api/events",
|
|
"invoices": "https://app.moderntreasury.com/api/invoices",
|
|
}
|
|
|
|
|
|
class ModernTreasuryLoader(BaseLoader):
|
|
"""Loader that fetches data from Modern Treasury."""
|
|
|
|
def __init__(
|
|
self,
|
|
resource: str,
|
|
organization_id: Optional[str] = None,
|
|
api_key: Optional[str] = None,
|
|
) -> None:
|
|
self.resource = resource
|
|
organization_id = organization_id or get_from_env(
|
|
"organization_id", "MODERN_TREASURY_ORGANIZATION_ID"
|
|
)
|
|
api_key = api_key or get_from_env("api_key", "MODERN_TREASURY_API_KEY")
|
|
credentials = f"{organization_id}:{api_key}".encode("utf-8")
|
|
basic_auth_token = b64encode(credentials).decode("utf-8")
|
|
self.headers = {"Authorization": f"Basic {basic_auth_token}"}
|
|
|
|
def _make_request(self, url: str) -> List[Document]:
|
|
request = urllib.request.Request(url, headers=self.headers)
|
|
|
|
with urllib.request.urlopen(request) as response:
|
|
json_data = json.loads(response.read().decode())
|
|
text = stringify_value(json_data)
|
|
metadata = {"source": url}
|
|
return [Document(page_content=text, metadata=metadata)]
|
|
|
|
def _get_resource(self) -> List[Document]:
|
|
endpoint = MODERN_TREASURY_ENDPOINTS.get(self.resource)
|
|
if endpoint is None:
|
|
return []
|
|
return self._make_request(endpoint)
|
|
|
|
def load(self) -> List[Document]:
|
|
return self._get_resource()
|