Carlos Florêncio
64 lines
2.1 KiB
TypeScript
64 lines
2.1 KiB
TypeScript
import { useCallback } from 'react';
|
|
import { getWebSocketClient } from '@/lib/ws/connection';
|
|
import { useQueue } from './domains/session/use-queue';
|
|
import type { MessageAttachment } from '@/components/task/chat/chat-input-container';
|
|
|
|
export function useMessageHandler(
|
|
resolvedSessionId: string | null,
|
|
taskId: string | null,
|
|
sessionModel: string | null,
|
|
activeModel: string | null,
|
|
planMode: boolean = false,
|
|
isAgentBusy: boolean = false
|
|
) {
|
|
const { queue } = useQueue(resolvedSessionId);
|
|
|
|
const handleSendMessage = useCallback(
|
|
async (message: string, attachments?: MessageAttachment[]) => {
|
|
if (!taskId || !resolvedSessionId) {
|
|
console.error('No active task session. Start an agent before sending a message.');
|
|
return;
|
|
}
|
|
const client = getWebSocketClient();
|
|
if (!client) return;
|
|
|
|
// Include active model in the request if it differs from the session model
|
|
const modelToSend = activeModel && activeModel !== sessionModel ? activeModel : undefined;
|
|
|
|
// Use longer timeout when sending attachments (images can be large)
|
|
const hasAttachments = attachments && attachments.length > 0;
|
|
const timeoutMs = hasAttachments ? 30000 : 10000;
|
|
|
|
// If agent is busy, queue the message instead of sending immediately
|
|
if (isAgentBusy) {
|
|
// Convert MessageAttachment[] to queue API format
|
|
const queueAttachments = attachments?.map(att => ({
|
|
type: att.type,
|
|
data: att.data,
|
|
mime_type: att.mime_type,
|
|
}));
|
|
|
|
await queue(taskId, message, modelToSend, planMode, queueAttachments);
|
|
return;
|
|
}
|
|
|
|
// Agent not busy - send message normally
|
|
await client.request(
|
|
'message.add',
|
|
{
|
|
task_id: taskId,
|
|
session_id: resolvedSessionId,
|
|
content: message,
|
|
...(modelToSend && { model: modelToSend }),
|
|
...(planMode && { plan_mode: true }),
|
|
...(hasAttachments && { attachments }),
|
|
},
|
|
timeoutMs
|
|
);
|
|
},
|
|
[resolvedSessionId, taskId, activeModel, sessionModel, planMode, isAgentBusy, queue]
|
|
);
|
|
|
|
return { handleSendMessage };
|
|
}
|