Home - XWiki

assistantLabel.innerHTML = `<strong>Assistant (${modelSelect.options[modelSelect.selectedIndex].text})</strong><em> - ΔT ${responseTime.toFixed(1)}s </em>`;

582

583

const usageSpan = document.createElement('span');

584

usageSpan.classList.add('usage');

585

usageSpan.innerHTML = '<em> - tokens<span class="usage-info">(📊)</span></em>';

586

assistantLabel.appendChild(usageSpan);

587

588

const usageInfo = document.createElement('div');

589

usageInfo.classList.add('usage-info-box');

590

usageInfo.innerHTML = `

591

<p>Prompt tokens: ${usageData.prompt_tokens}</p>

592

<p>Completion tokens: ${usageData.completion_tokens}</p>

593

<p>Total tokens: ${usageData.total_tokens}</p>

594

`;

595

usageSpan.appendChild(usageInfo);

}

// Handle non-streaming request

600

function handleNonStreamingRequest(request, signal, assistantMessageElement) {

601

const startTime = new Date().getTime();

602

603

XWikiAiAPI.getCompletions(request, null, signal)

604

.then(response => {

605

handleNonStreamingResponse(response, startTime, assistantMessageElement);

606

})

607

.catch(error => {

608

handleRequestError(error);

609

})

610

.finally(() => {

611

changeBtnState(true);

});

}

// Handle non-streaming response

616

function handleNonStreamingResponse(response, startTime, assistantMessageElement) {

617

const endTime = new Date().getTime();

618

const responseTime = (endTime - startTime) / 1000;

619

620

const assistantMessage = response.choices[0].message.content;

621

const llmMemory = response.choices[0].message.memory;

622

console.debug('LLM memory before response:', llmMemory);

623

624

const sourcesMatch = assistantMessage.match(/Sources:([\s\S]*?)(?=\n\n|$)/);

625

if (sourcesMatch) {

626

const sources = sourcesMatch[1].trim();

627

const content = assistantMessage.replace(sourcesMatch[0], '').trim();

628

const sourcesBubble = createSourcesBubble(sources);

629

assistantMessageElement.parentElement.insertBefore(sourcesBubble, assistantMessageElement);

630

assistantMessageElement.innerHTML = DOMPurify.sanitize(marked.parse(content), { FORBID_TAGS: ['style'], FORBID_ATTR: ['src'] });

631

} else {

632

assistantMessageElement.innerHTML = DOMPurify.sanitize(marked.parse(assistantMessage), { FORBID_TAGS: ['style'], FORBID_ATTR: ['src'] });

633

}

634

635

conversationHistory.push({ role: 'assistant', content: assistantMessage });

636

chatHistory.scrollTop = chatHistory.scrollHeight;

637

removeWaitingAnimation();

638

saveConversationHistory();

639

640

const assistantInfoElement = assistantMessageElement.previousElementSibling.querySelector('.msg-info');

641

if (assistantInfoElement) {

642

assistantInfoElement.innerHTML = `<strong>Assistant (${modelSelect.options[modelSelect.selectedIndex].text})</strong> - <em>ΔT ${responseTime.toFixed(1)}s </em>`;

643

644

if (response.usage) {

645

displayUsageInfo(assistantInfoElement, response.usage, startTime);

646

}

647

} else {

648

console.warn('Assistant info element not found');

}

}

function endTimer(startTime) {

653

const endTime = new Date().getTime();

654

const responseTime = (endTime - startTime) / 1000;

return responseTime;

}

// Handle request error

659

function handleRequestError(error, updateTimer) {

660

if (error.name === 'AbortError') {

661

console.log('Request aborted');

662

} else {

663

console.error('Failed to get chat completions:', error);

664

displayErrorMessage('An error occurred: ' + error.message);

665

}

666

removeWaitingAnimation();

667

clearInterval(updateTimer);

668

}

669

670

// Stop the current request

671

function stopRequest() {

672

if (abortController) {

673

console.log('Aborting request...');

674

abortController.abort();

675

abortController = null;

676

removeWaitingAnimation();

677

changeBtnState(true);

}

}

// Display user message in the chat history

682

function displayUserMessage(message) {

683

const messageElement = document.createElement('div');

684

messageElement.classList.add('user-message');

685

686

const userLabel = document.createElement('div');

687

userLabel.classList.add('msg-info');

688

userLabel.textContent = 'User:';

689

690

const userMessageContent = document.createElement('div');

691

userMessageContent.classList.add('message-content');

692

userMessageContent.textContent = message;

693

694

messageElement.appendChild(userLabel);

695

messageElement.appendChild(userMessageContent);

696

697

chatHistory.appendChild(messageElement);

698

chatHistory.scrollTop = chatHistory.scrollHeight;

699

}

700

701

// Display assistant message in the chat history

702

function displayAssistantMessage(message, modelName = '', responseTime = null) {

703

const messageElement = document.createElement('div');

704

messageElement.classList.add('assistant');

705

706

const assistantLabel = document.createElement('div');

707

assistantLabel.classList.add('msg-info');

708

709

if (modelName && responseTime !== null) {

710

assistantLabel.innerHTML = `<strong>Assistant (${modelName})</strong><em> - ΔT ${responseTime.toFixed(1)}s:</em>`;

711

} else {

712

assistantLabel.innerHTML = '<strong>Assistant:</strong>';

713

}

714

715

const assistantMessageContent = document.createElement('div');

716

assistantMessageContent.classList.add('message-text');

717

assistantMessageContent.innerHTML = DOMPurify.sanitize(marked.parse(message), { FORBID_TAGS: ['style'], FORBID_ATTR: ['src'] });

718

719

messageElement.appendChild(assistantLabel);

720

messageElement.appendChild(assistantMessageContent);

721

722

chatHistory.appendChild(messageElement);

723

chatHistory.scrollTop = chatHistory.scrollHeight;

724

725

return assistantMessageContent;

}

// Update the response time in the assistant label

731

function updateResponseTime(startTime, assistantInfoElement) {

732

const currentTime = new Date().getTime();

733

const elapsedTime = (currentTime - startTime) / 1000;

734

assistantInfoElement.innerHTML = `<strong>Assistant (${modelSelect.options[modelSelect.selectedIndex].text})</strong> - ΔT ${elapsedTime.toFixed(1)}s:`;

735

}

736

737

// Display error message in the chat history

738

function displayErrorMessage(message) {

739

const messageElement = document.createElement('div');

740

messageElement.classList.add('error-message');

741

messageElement.textContent = message;

742

743

chatHistory.appendChild(messageElement);

744

chatHistory.scrollTop = chatHistory.scrollHeight;

745

}

746

747

// Update user settings

748

function updateUserSettings() {

749

userSettings.model = modelSelect.value;

750

userSettings.temperature = parseFloat(temperatureInput.value);

751

userSettings.stream = streamCheckbox.checked;

saveUserSettings();

}

// Save user settings to local storage

756

function saveUserSettings() {

757

localStorage.setItem('userSettings', JSON.stringify(userSettings));

758

}

759

760

// Load user settings from local storage

761

function loadUserSettings() {

762

const storedSettings = localStorage.getItem('userSettings');

763

if (storedSettings) {

764

userSettings = JSON.parse(storedSettings);

765

modelSelect.value = userSettings.model;

766

temperatureInput.value = userSettings.temperature;

767

streamCheckbox.checked = userSettings.stream;

768

769

// Apply the collapsed state to the settings section

770

const settingsWrapper = document.querySelector('.settings-wrapper');

771

if (userSettings.settingsCollapsed) {

772

settingsWrapper.classList.add('collapsed');

773

} else {

774

settingsWrapper.classList.remove('collapsed');

}

}

}

// Save conversation history to local storage

781

function saveConversationHistory() {

782

if (conversationHistory.length === 0) {

783

localStorage.removeItem('conversationHistory');

784

} else {

785

localStorage.setItem('conversationHistory', JSON.stringify(conversationHistory));

}

}

// Load last conversation from local storage

790

function loadLastConversation() {

791

const storedConversation = localStorage.getItem('conversationHistory');

792

if (storedConversation) {

793

conversationHistory = JSON.parse(storedConversation);

794

conversationHistory.forEach(message => {

795

if (message.role === 'user') {

796

displayUserMessage(message.content);

797

} else if (message.role === 'assistant') {

798

displayAssistantMessage(message.content);

}

});

}

}

// Toggle settings visibility

805

function toggleSettings() {

806

const settingsWrapper = document.querySelector('.settings-wrapper');

807

settingsWrapper.classList.toggle('collapsed');

808

userSettings.settingsCollapsed = settingsWrapper.classList.contains('collapsed');

saveUserSettings();

}

// Call the initialization function when the DOM content is loaded

814

document.addEventListener('DOMContentLoaded', initializeChatWidget);

815

816

Wiki source code of Home

Navigation

Follow Us