mirror of
https://github.com/run-llama/llama-hub.git
synced 2025-08-13 19:21:15 +00:00
33 lines
901 B
Python
33 lines
901 B
Python
"""Airtable reader."""
|
|
from typing import List
|
|
|
|
from llama_index.readers.base import BaseReader
|
|
from llama_index.readers.schema.base import Document
|
|
|
|
|
|
class AirtableReader(BaseReader):
|
|
"""Airtable reader. Reads data from a table in a base.
|
|
|
|
Args:
|
|
api_key (str): Airtable API key.
|
|
"""
|
|
|
|
def __init__(self, api_key: str) -> None:
|
|
"""Initialize Airtable reader."""
|
|
|
|
|
|
self.api_key = api_key
|
|
|
|
def load_data(self, base_id: str,table_id: str) -> List[Document]:
|
|
"""Load data from a table in a base
|
|
|
|
Args:
|
|
table_id (str): Table ID.
|
|
base_id (str): Base ID.
|
|
Returns:
|
|
List[Document]: List of documents.
|
|
"""
|
|
from pyairtable import Table
|
|
table = Table(self.api_key, base_id, table_id)
|
|
all_records=table.all()
|
|
return [Document(f"{all_records}", extra_info={})] |