Yingjie He
86 lines
3.0 KiB
TypeScript
86 lines
3.0 KiB
TypeScript
import React, { useState } from 'react';
|
|
import { ConfirmToolRequest } from '../utils/toolConfirm';
|
|
import { snakeToTitleCase } from '../utils';
|
|
|
|
export default function ToolConfirmation({
|
|
isCancelledMessage,
|
|
isClicked,
|
|
toolConfirmationId,
|
|
toolName,
|
|
}) {
|
|
const [clicked, setClicked] = useState(isClicked);
|
|
const [status, setStatus] = useState('unknown');
|
|
|
|
const handleButtonClick = (confirmed) => {
|
|
setClicked(true);
|
|
setStatus(confirmed ? 'approved' : 'denied');
|
|
ConfirmToolRequest(toolConfirmationId, confirmed);
|
|
};
|
|
|
|
return isCancelledMessage ? (
|
|
<div className="goose-message-content bg-bgSubtle rounded-2xl px-4 py-2 text-textStandard">
|
|
Tool call confirmation is cancelled.
|
|
</div>
|
|
) : (
|
|
<>
|
|
<div className="goose-message-content bg-bgSubtle rounded-2xl px-4 py-2 rounded-b-none text-textStandard">
|
|
Goose would like to call the above tool. Allow?
|
|
</div>
|
|
{clicked ? (
|
|
<div className="goose-message-tool bg-bgApp border border-borderSubtle dark:border-gray-700 rounded-b-2xl px-4 pt-4 pb-2 flex gap-4 mt-1">
|
|
<div className="flex items-center">
|
|
{status === 'approved' && (
|
|
<svg
|
|
className="w-5 h-5 text-gray-500"
|
|
xmlns="http://www.w3.org/2000/svg"
|
|
fill="none"
|
|
viewBox="0 0 24 24"
|
|
stroke="currentColor"
|
|
strokeWidth={2}
|
|
>
|
|
<path strokeLinecap="round" strokeLinejoin="round" d="M5 13l4 4L19 7" />
|
|
</svg>
|
|
)}
|
|
{status === 'denied' && (
|
|
<svg
|
|
className="w-5 h-5 text-gray-500"
|
|
xmlns="http://www.w3.org/2000/svg"
|
|
fill="none"
|
|
viewBox="0 0 24 24"
|
|
stroke="currentColor"
|
|
strokeWidth={2}
|
|
>
|
|
<path strokeLinecap="round" strokeLinejoin="round" d="M6 18L18 6M6 6l12 12" />
|
|
</svg>
|
|
)}
|
|
<span className="ml-2 text-textStandard">
|
|
{isClicked
|
|
? 'Tool confirmation is not available'
|
|
: `${snakeToTitleCase(toolName.substring(toolName.lastIndexOf('__') + 2))} is ${status}`}
|
|
</span>
|
|
</div>
|
|
</div>
|
|
) : (
|
|
<div className="goose-message-tool bg-bgApp border border-borderSubtle dark:border-gray-700 rounded-b-2xl px-4 pt-4 pb-2 flex gap-4 mt-1">
|
|
<button
|
|
className={
|
|
'bg-black text-white dark:bg-white dark:text-black rounded-full px-6 py-2 transition'
|
|
}
|
|
onClick={() => handleButtonClick(true)}
|
|
>
|
|
Allow tool
|
|
</button>
|
|
<button
|
|
className={
|
|
'bg-white text-black dark:bg-black dark:text-white border border-gray-300 dark:border-gray-700 rounded-full px-6 py-2 transition'
|
|
}
|
|
onClick={() => handleButtonClick(false)}
|
|
>
|
|
Deny
|
|
</button>
|
|
</div>
|
|
)}
|
|
</>
|
|
);
|
|
}
|