mirror of
https://github.com/agent0ai/agent-zero.git
synced 2026-05-06 08:01:31 +00:00
99a2a94813
Use call_extensions() helper Let extensions append banners directly Rename HUGGINGFACE_LOCAL_FOR_EMBEDDING to LOCAL_EMBEDDING
23 lines
850 B
Python
23 lines
850 B
Python
from python.helpers.api import ApiHandler, Request, Response
|
|
from python.helpers.extension import call_extensions
|
|
|
|
|
|
class GetBanners(ApiHandler):
|
|
"""
|
|
API endpoint for Welcome Screen banners.
|
|
Add checks as extension scripts in python/extensions/banners/ or usr/extensions/banners/
|
|
"""
|
|
|
|
async def process(self, input: dict, request: Request) -> dict | Response:
|
|
frontend_banners = input.get("banners", [])
|
|
frontend_context = input.get("context", {})
|
|
|
|
# Banners array passed by reference - extensions append directly to it
|
|
banners = []
|
|
await call_extensions("banners", agent=None, banners=banners, context=frontend_context, frontend_banners=frontend_banners)
|
|
|
|
return {"banners": banners}
|
|
|
|
@classmethod
|
|
def get_methods(cls) -> list[str]:
|
|
return ["POST"]
|