mirror of
https://github.com/temporal-community/temporal-ai-agent.git
synced 2026-03-15 22:18:09 +01:00
23 lines
1.1 KiB
Python
23 lines
1.1 KiB
Python
from pathlib import Path
|
|
import json
|
|
|
|
# this is made to demonstrate functionality but it could just as durably be an API call
|
|
# this assumes it's a valid account - use check_account_valid() to verify that first
|
|
def get_account_balance(args: dict) -> dict:
|
|
|
|
account_key = args.get("accountkey")
|
|
|
|
file_path = Path(__file__).resolve().parent.parent / "data" / "customer_account_data.json"
|
|
if not file_path.exists():
|
|
return {"error": "Data file not found."}
|
|
|
|
with open(file_path, "r") as file:
|
|
data = json.load(file)
|
|
account_list = data["accounts"]
|
|
|
|
for account in account_list:
|
|
if account["email"] == account_key or account["account_id"] == account_key:
|
|
return{ "name": account["name"], "email": account["email"], "account_id": account["account_id"], "checking_balance": account["checking_balance"], "savings_balance": account["savings_balance"], "bitcoin_balance": account["bitcoin_balance"], "account_creation_date": account["account_creation_date"] }
|
|
|
|
return_msg = "Account not found with for " + account_key
|
|
return {"error": return_msg} |