mirror of
https://github.com/stripe/ai.git
synced 2026-09-14 18:39:59 +08:00
Steve Kaliski
64 lines
1.5 KiB
Python
64 lines
1.5 KiB
Python
from typing import Literal, Optional
|
|
from typing_extensions import TypedDict
|
|
|
|
# Define Object type
|
|
Object = Literal[
|
|
"customers",
|
|
"invoices",
|
|
"invoiceItems",
|
|
"paymentLinks",
|
|
"products",
|
|
"prices",
|
|
"balance",
|
|
"refunds",
|
|
]
|
|
|
|
|
|
# Define Permission type
|
|
class Permission(TypedDict, total=False):
|
|
create: Optional[bool]
|
|
update: Optional[bool]
|
|
read: Optional[bool]
|
|
|
|
|
|
# Define BalancePermission type
|
|
class BalancePermission(TypedDict, total=False):
|
|
read: Optional[bool]
|
|
|
|
|
|
# Define Actions type
|
|
class Actions(TypedDict, total=False):
|
|
customers: Optional[Permission]
|
|
invoices: Optional[Permission]
|
|
invoice_items: Optional[Permission]
|
|
payment_links: Optional[Permission]
|
|
products: Optional[Permission]
|
|
prices: Optional[Permission]
|
|
balance: Optional[BalancePermission]
|
|
refunds: Optional[Permission]
|
|
|
|
|
|
# Define Context type
|
|
class Context(TypedDict, total=False):
|
|
account: Optional[str]
|
|
|
|
|
|
# Define Configuration type
|
|
class Configuration(TypedDict, total=False):
|
|
actions: Optional[Actions]
|
|
context: Optional[Context]
|
|
|
|
|
|
def is_tool_allowed(tool, configuration):
|
|
for resource, permissions in tool.get("actions").items():
|
|
if resource not in configuration.get("actions", {}):
|
|
return False
|
|
for permission in permissions:
|
|
if (
|
|
not configuration["actions"]
|
|
.get(resource, {})
|
|
.get(permission, False)
|
|
):
|
|
return False
|
|
return True
|