mirror of
https://github.com/run-llama/template-workflow-data-extraction.git
synced 2026-07-01 21:34:19 -04:00
Adrian Lyjak
29 lines
760 B
Python
29 lines
760 B
Python
import functools
|
|
import os
|
|
|
|
from extracted_data_client import AuthenticatedClient
|
|
from llama_cloud_services import LlamaExtract
|
|
|
|
import dotenv
|
|
|
|
dotenv.load_dotenv()
|
|
|
|
# Add getters for clients and environment variables here.
|
|
|
|
|
|
@functools.lru_cache(maxsize=None)
|
|
def get_extract_api() -> LlamaExtract:
|
|
return LlamaExtract(
|
|
api_key=os.environ["LLAMA_CLOUD_API_KEY"],
|
|
project_id=os.environ.get("LLAMA_CLOUD_PROJECT_ID", None),
|
|
organization_id=os.environ.get("LLAMA_CLOUD_ORG_ID", None),
|
|
)
|
|
|
|
|
|
@functools.lru_cache(maxsize=None)
|
|
def get_extracted_data_client():
|
|
return AuthenticatedClient(
|
|
base_url=os.environ.get("EXTRACTED_DATA_BASE_URL", "http://localhost:9182"),
|
|
token=os.environ["EXTRACTED_DATA_TOKEN"],
|
|
)
|