Spaces:
Paused
Paused
File size: 2,274 Bytes
8d1819a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 |
from agent import AgentContext
from python.helpers.api import ApiHandler, Request, Response
class ApiLogGet(ApiHandler):
@classmethod
def get_methods(cls) -> list[str]:
return ["GET", "POST"]
@classmethod
def requires_auth(cls) -> bool:
return False # No web auth required
@classmethod
def requires_csrf(cls) -> bool:
return False # No CSRF required
@classmethod
def requires_api_key(cls) -> bool:
return True # Require API key
async def process(self, input: dict, request: Request) -> dict | Response:
# Extract parameters (support both query params for GET and body for POST)
if request.method == "GET":
context_id = request.args.get("context_id", "")
length = int(request.args.get("length", 100))
else:
context_id = input.get("context_id", "")
length = input.get("length", 100)
if not context_id:
return Response('{"error": "context_id is required"}', status=400, mimetype="application/json")
# Get context
context = AgentContext.use(context_id)
if not context:
return Response('{"error": "Context not found"}', status=404, mimetype="application/json")
try:
# Get total number of log items
total_items = len(context.log.logs)
# Calculate start position (from newest, so we work backwards)
start_pos = max(0, total_items - length)
# Get log items from the calculated start position
log_items = context.log.output(start=start_pos)
# Return log data with metadata
return {
"context_id": context_id,
"log": {
"guid": context.log.guid,
"total_items": total_items,
"returned_items": len(log_items),
"start_position": start_pos,
"progress": context.log.progress,
"progress_active": context.log.progress_active,
"items": log_items
}
}
except Exception as e:
return Response(f'{{"error": "{str(e)}"}}', status=500, mimetype="application/json")
|