Spaces:
Paused
Paused
File size: 2,144 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 66 67 68 69 70 |
from agent import AgentContext
from python.helpers.api import ApiHandler, Request, Response
from python.helpers.print_style import PrintStyle
from python.helpers import persist_chat
import json
class ApiResetChat(ApiHandler):
@classmethod
def requires_auth(cls) -> bool:
return False
@classmethod
def requires_csrf(cls) -> bool:
return False
@classmethod
def requires_api_key(cls) -> bool:
return True
@classmethod
def get_methods(cls) -> list[str]:
return ["POST"]
async def process(self, input: dict, request: Request) -> dict | Response:
try:
# Get context_id from input
context_id = input.get("context_id")
if not context_id:
return Response(
'{"error": "context_id is required"}',
status=400,
mimetype="application/json"
)
# Check if context exists
context = AgentContext.use(context_id)
if not context:
return Response(
'{"error": "Chat context not found"}',
status=404,
mimetype="application/json"
)
# Reset the chat context (clears history but keeps context alive)
context.reset()
# Save the reset context to persist the changes
persist_chat.save_tmp_chat(context)
# Log the reset
PrintStyle(
background_color="#3498DB", font_color="white", bold=True, padding=True
).print(f"API Chat reset: {context_id}")
# Return success response
return {
"success": True,
"message": "Chat reset successfully",
"context_id": context_id
}
except Exception as e:
PrintStyle.error(f"API reset chat error: {str(e)}")
return Response(
json.dumps({"error": f"Internal server error: {str(e)}"}),
status=500,
mimetype="application/json"
)
|