File size: 2,992 Bytes
67b6ab0 | 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 71 72 | from agent import AgentContext, UserMessage
from helpers.api import ApiHandler, Request, Response
from helpers import files, extension, message_queue as mq
import os
from helpers.security import safe_filename
from helpers.defer import DeferredTask
class Message(ApiHandler):
async def process(self, input: dict, request: Request) -> dict | Response:
task, context = await self.communicate(input=input, request=request)
return await self.respond(task, context)
async def respond(self, task: DeferredTask, context: AgentContext):
result = await task.result() # type: ignore
return {
"message": result,
"context": context.id,
}
async def communicate(self, input: dict, request: Request):
# Handle both JSON and multipart/form-data
if request.content_type.startswith("multipart/form-data"):
text = request.form.get("text", "")
ctxid = request.form.get("context", "")
message_id = request.form.get("message_id", None)
attachments = request.files.getlist("attachments")
attachment_paths = []
upload_folder_int = "/a0/usr/uploads"
upload_folder_ext = files.get_abs_path("usr/uploads") # for development environment
if attachments:
os.makedirs(upload_folder_ext, exist_ok=True)
for attachment in attachments:
if attachment.filename is None:
continue
filename = safe_filename(attachment.filename)
if not filename:
continue
save_path = files.get_abs_path(upload_folder_ext, filename)
attachment.save(save_path)
attachment_paths.append(os.path.join(upload_folder_int, filename))
else:
# Handle JSON request as before
input_data = request.get_json()
text = input_data.get("text", "")
ctxid = input_data.get("context", "")
message_id = input_data.get("message_id", None)
attachment_paths = []
# Now process the message
message = text
# Obtain agent context
context = self.use_context(ctxid)
# call extension point, alow it to modify data
data = { "message": message, "attachment_paths": attachment_paths }
await extension.call_extensions_async("user_message_ui", agent=context.get_agent(), data=data)
message = data.get("message", "")
attachment_paths = data.get("attachment_paths", [])
# Store attachments in agent data
# context.agent0.set_data("attachments", attachment_paths)
# Log to console and UI using helper function
mq.log_user_message(context, message, attachment_paths, message_id)
return context.communicate(UserMessage(message=message, attachments=attachment_paths, id=message_id or "")), context
|