mirror of
https://github.com/langgenius/dify.git
synced 2026-11-11 05:41:09 +08:00
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
45 lines
1.0 KiB
Python
45 lines
1.0 KiB
Python
"""Typed resource references for app ownership chains."""
|
|
|
|
from typing import NamedTuple
|
|
|
|
from models.model import App
|
|
|
|
|
|
class AppRef(NamedTuple):
|
|
"""App identifiers used to scope downstream resource lookups."""
|
|
|
|
tenant_id: str
|
|
app_id: str
|
|
|
|
|
|
class MessageRef(NamedTuple):
|
|
"""Message identifiers used to scope downstream resource lookups."""
|
|
|
|
app: AppRef
|
|
message_id: str
|
|
end_user_id: str | None = None
|
|
account_id: str | None = None
|
|
|
|
|
|
class AppRefService:
|
|
"""Factory helpers for app and child resource refs."""
|
|
|
|
@staticmethod
|
|
def create_app_ref(app: App) -> AppRef:
|
|
return AppRef(tenant_id=app.tenant_id, app_id=app.id)
|
|
|
|
@staticmethod
|
|
def create_message_ref(
|
|
app_ref: AppRef,
|
|
message_id: str,
|
|
*,
|
|
end_user_id: str | None = None,
|
|
account_id: str | None = None,
|
|
) -> MessageRef:
|
|
return MessageRef(
|
|
app=app_ref,
|
|
message_id=message_id,
|
|
end_user_id=end_user_id,
|
|
account_id=account_id,
|
|
)
|