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(
```
47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
"""Loader that fetches data from Stripe"""
|
|
import json
|
|
import urllib.request
|
|
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_dict
|
|
|
|
STRIPE_ENDPOINTS = {
|
|
"balance_transactions": "https://api.stripe.com/v1/balance_transactions",
|
|
"charges": "https://api.stripe.com/v1/charges",
|
|
"customers": "https://api.stripe.com/v1/customers",
|
|
"events": "https://api.stripe.com/v1/events",
|
|
"refunds": "https://api.stripe.com/v1/refunds",
|
|
"disputes": "https://api.stripe.com/v1/disputes",
|
|
}
|
|
|
|
|
|
class StripeLoader(BaseLoader):
|
|
"""Loader that fetches data from Stripe."""
|
|
|
|
def __init__(self, resource: str, access_token: Optional[str] = None) -> None:
|
|
self.resource = resource
|
|
access_token = access_token or get_from_env(
|
|
"access_token", "STRIPE_ACCESS_TOKEN"
|
|
)
|
|
self.headers = {"Authorization": f"Bearer {access_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_dict(json_data)
|
|
metadata = {"source": url}
|
|
return [Document(page_content=text, metadata=metadata)]
|
|
|
|
def _get_resource(self) -> List[Document]:
|
|
endpoint = STRIPE_ENDPOINTS.get(self.resource)
|
|
if endpoint is None:
|
|
return []
|
|
return self._make_request(endpoint)
|
|
|
|
def load(self) -> List[Document]:
|
|
return self._get_resource()
|