mirror of
https://github.com/Mintplex-Labs/langchain-python.git
synced 2026-07-19 13:26:32 -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(
```
50 lines
1.8 KiB
Python
50 lines
1.8 KiB
Python
"""Loader that fetches data from Spreedly API."""
|
|
import json
|
|
import urllib.request
|
|
from typing import List
|
|
|
|
from langchain.docstore.document import Document
|
|
from langchain.document_loaders.base import BaseLoader
|
|
from langchain.utils import stringify_dict
|
|
|
|
SPREEDLY_ENDPOINTS = {
|
|
"gateways_options": "https://core.spreedly.com/v1/gateways_options.json",
|
|
"gateways": "https://core.spreedly.com/v1/gateways.json",
|
|
"receivers_options": "https://core.spreedly.com/v1/receivers_options.json",
|
|
"receivers": "https://core.spreedly.com/v1/receivers.json",
|
|
"payment_methods": "https://core.spreedly.com/v1/payment_methods.json",
|
|
"certificates": "https://core.spreedly.com/v1/certificates.json",
|
|
"transactions": "https://core.spreedly.com/v1/transactions.json",
|
|
"environments": "https://core.spreedly.com/v1/environments.json",
|
|
}
|
|
|
|
|
|
class SpreedlyLoader(BaseLoader):
|
|
"""Loader that fetches data from Spreedly API."""
|
|
|
|
def __init__(self, access_token: str, resource: str) -> None:
|
|
self.access_token = access_token
|
|
self.resource = resource
|
|
self.headers = {
|
|
"Authorization": f"Bearer {self.access_token}",
|
|
"Accept": "application/json",
|
|
}
|
|
|
|
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_dict(json_data)
|
|
metadata = {"source": url}
|
|
return [Document(page_content=text, metadata=metadata)]
|
|
|
|
def _get_resource(self) -> List[Document]:
|
|
endpoint = SPREEDLY_ENDPOINTS.get(self.resource)
|
|
if endpoint is None:
|
|
return []
|
|
return self._make_request(endpoint)
|
|
|
|
def load(self) -> List[Document]:
|
|
return self._get_resource()
|