llms-py 2.0.9__py3-none-any.whl → 3.0.10__py3-none-any.whl

llms_py-2.0.9.data/data/ui/ChatPrompt.mjs

@@ -1,389 +0,0 @@
-import { ref, nextTick, inject } from 'vue'
-import { useRouter } from 'vue-router'
-import { lastRightPart } from '@servicestack/client'
-import { deepClone, fileToDataUri, fileToBase64, addCopyButtons } from './utils.mjs'
-
-const imageExts = 'png,webp,jpg,jpeg,gif,bmp,svg,tiff,ico'.split(',')
-const audioExts = 'mp3,wav,ogg,flac,m4a,opus,webm'.split(',')
-
-export function useChatPrompt() {
-    const messageText = ref('')
-    const attachedFiles = ref([])
-    const isGenerating = ref(false)
-    const errorStatus = ref(null)
-    const errorMessage = ref(null)
-    const hasImage = () => attachedFiles.value.some(f => imageExts.includes(lastRightPart(f.name, '.')))
-    const hasAudio = () => attachedFiles.value.some(f => audioExts.includes(lastRightPart(f.name, '.')))
-    const hasFile = () => attachedFiles.value.length > 0
-    // const hasText = () => !hasImage() && !hasAudio() && !hasFile()
-
-    function reset() {
-        // Ensure initial state is ready to accept input
-        isGenerating.value = false
-        attachedFiles.value = []
-        messageText.value = ''
-    }
-
-    return {
-        messageText,
-        attachedFiles,
-        errorStatus,
-        errorMessage,
-        isGenerating,
-        get generating() {
-            return isGenerating.value
-        },
-        hasImage,
-        hasAudio,
-        hasFile,
-        // hasText,
-        reset,
-    }
-}
-
-export default {
-    template:`
-    <div class="mx-auto max-w-3xl">
-        <div class="flex space-x-3">
-            <!-- Attach (+) button -->
-            <div>
-                <button type="button"
-                        @click="triggerFilePicker"
-                        :disabled="isGenerating || !model"
-                        class="mt-2 h-10 w-10 flex items-center justify-center rounded-md border border-gray-300 text-gray-600 hover:bg-gray-50 disabled:text-gray-400 disabled:cursor-not-allowed"
-                        title="Attach image or audio">
-                    <svg class="w-5 h-5" xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round">
-                        <line x1="12" y1="5" x2="12" y2="19"></line>
-                        <line x1="5" y1="12" x2="19" y2="12"></line>
-                    </svg>
-                </button>
-                <!-- Hidden file input -->
-                <input ref="fileInput" type="file" multiple @change="onFilesSelected" 
-                    class="hidden" accept="image/*,audio/*,.pdf,.doc,.docx,.xml,application/msword,application/vnd.openxmlformats-officedocument.wordprocessingml.document"
-                    />
-            </div>
-
-            <div class="flex-1">
-                <textarea
-                    ref="messageInput"
-                    v-model="messageText"
-                    @keydown.enter.exact.prevent="sendMessage"
-                    @keydown.enter.shift.exact="addNewLine"
-                    placeholder="Type your message... (Enter to send, Shift+Enter for new line)"
-                    rows="2"
-                    class="block w-full rounded-md border border-gray-300 px-3 py-2 text-sm placeholder-gray-500 focus:border-blue-500 focus:outline-none focus:ring-1 focus:ring-blue-500"
-                    :disabled="isGenerating || !model"
-                ></textarea>
-
-                <!-- Attached files preview -->
-                <div v-if="attachedFiles.length" class="mt-2 flex flex-wrap gap-2">
-                    <div v-for="(f,i) in attachedFiles" :key="i" class="flex items-center gap-2 px-2 py-1 rounded-md border border-gray-300 text-xs text-gray-700 bg-gray-50">
-                        <span class="truncate max-w-48" :title="f.name">{{ f.name }}</span>
-                        <button type="button" class="text-gray-500 hover:text-gray-700" @click="removeAttachment(i)" title="Remove Attachment">
-                            <svg class="w-4 h-4" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><line x1="18" y1="6" x2="6" y2="18"></line><line x1="6" y1="6" x2="18" y2="18"></line></svg>
-                        </button>
-                    </div>
-                </div>
-
-                <div v-if="!model" class="mt-2 text-sm text-red-600">
-                    Please select a model
-                </div>
-            </div>
-
-            <div>
-                <button title="Send (Enter)" type="button"
-                    @click="sendMessage"
-                    :disabled="!messageText.trim() || isGenerating || !model"
-                    class="mt-2 p-2 flex items-center justify-center rounded-full bg-gray-700 text-white transition-colors hover:opacity-70 focus-visible:outline-none focus-visible:outline-black disabled:bg-[#D7D7D7] disabled:text-[#f4f4f4] disabled:hover:opacity-100 dark:bg-white dark:text-black dark:focus-visible:outline-white disabled:dark:bg-token-text-quaternary dark:disabled:text-token-main-surface-secondary">
-                    <svg v-if="isGenerating" class="size-6 animate-spin" fill="none" viewBox="0 0 24 24">
-                        <circle class="opacity-25" cx="12" cy="12" r="10" stroke="currentColor" stroke-width="4"></circle>
-                        <path class="opacity-75" fill="currentColor" d="M4 12a8 8 0 018-8V0C5.373 0 0 5.373 0 12h4zm2 5.291A7.962 7.962 0 014 12H0c0 3.042 1.135 5.824 3 7.938l3-2.647z"></path>
-                    </svg>
-                    <svg v-else class="size-6" xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24"><path fill="currentColor" d="m3.165 19.503l7.362-16.51c.59-1.324 2.355-1.324 2.946 0l7.362 16.51c.667 1.495-.814 3.047-2.202 2.306l-5.904-3.152c-.459-.245-1-.245-1.458 0l-5.904 3.152c-1.388.74-2.87-.81-2.202-2.306"></path></svg>
-                </button>
-            </div>
-        </div>
-    </div>    
-    `,
-    props: {
-        model: {
-            type: String,
-            default: ''
-        },
-        systemPrompt: {
-            type: String,
-            default: ''
-        }
-    },
-    setup(props) {
-        const router = useRouter()
-        const config = inject('config')
-        const chatPrompt = inject('chatPrompt')
-        const { 
-            messageText, 
-            attachedFiles, 
-            isGenerating, 
-            errorStatus,
-            errorMessage,
-            hasImage, 
-            hasAudio, 
-            hasFile 
-        } = chatPrompt
-        const threads = inject('threads')
-        const {
-            currentThread,
-        } = threads
-
-        const fileInput = ref(null)
-
-        // File attachments (+) handlers
-        const triggerFilePicker = () => {
-            if (fileInput.value) fileInput.value.click()
-        }
-        const onFilesSelected = (e) => {
-            const files = Array.from(e.target?.files || [])
-            if (files.length) attachedFiles.value.push(...files)
-            // allow re-selecting the same file
-            if (fileInput.value) fileInput.value.value = ''
-
-            if (!messageText.value.trim()) {
-                if (hasImage()) {
-                    messageText.value = getTextContent(config.defaults.image)
-                } else if (hasAudio()) {
-                    messageText.value = getTextContent(config.defaults.audio)
-                } else {
-                    messageText.value = getTextContent(config.defaults.file)
-                }
-            }
-        }
-        const removeAttachment = (i) => {
-            attachedFiles.value.splice(i, 1)
-        }
-
-        function createChatRequest() {
-            if (hasImage()) {
-                return deepClone(config.defaults.image)
-            }
-            if (hasAudio()) {
-                return deepClone(config.defaults.audio)
-            }
-            if (attachedFiles.value.length) {
-                return deepClone(config.defaults.file)
-            }
-            const text = deepClone(config.defaults.text)
-            return text
-        }
-
-        function getTextContent(chat) {
-            const textMessage = chat.messages.find(m =>
-                m.role === 'user' && Array.isArray(m.content) && m.content.some(c => c.type === 'text'))
-            return textMessage?.content.find(c => c.type === 'text')?.text || ''
-        }
-
-        // Send message
-        const sendMessage = async () => {
-            if (!messageText.value.trim() || isGenerating.value || !props.model) return
-
-            // Clear any existing error message
-            errorStatus.value = errorMessage.value = null
-
-            let message = messageText.value.trim()
-            if (attachedFiles.value.length) {
-                const names = attachedFiles.value.map(f => f.name).join(', ')
-                const mediaType = imageExts.some(ext => names.includes(ext))
-                    ? '🖼️'
-                    : audioExts.some(ext => names.includes(ext))
-                        ? '🔉'
-                        : '📎'
-                message += `\n\n[${mediaType} ${names}]`
-            }
-            messageText.value = ''
-
-            try {
-                let threadId
-
-                // Create thread if none exists
-                if (!currentThread.value) {
-                    const newThread = await threads.createThread('New Chat', props.model, props.systemPrompt)
-                    threadId = newThread.id
-                    // Navigate to the new thread URL
-                    router.push(`/c/${newThread.id}`)
-                } else {
-                    threadId = currentThread.value.id
-                    // Update the existing thread's model and systemPrompt to match current selection
-                    await threads.updateThread(threadId, {
-                        model: props.model,
-                        systemPrompt: props.systemPrompt
-                    })
-                }
-
-                // Add user message
-                await threads.addMessageToThread(threadId, {
-                    role: 'user',
-                    content: message
-                })
-
-                isGenerating.value = true
-
-                // Get the updated thread to prepare chat request
-                const thread = await threads.getThread(threadId)
-                const messages = [...thread.messages]
-
-                // Add system prompt if present
-                if (props.systemPrompt?.trim()) {
-                    messages.unshift({
-                        role: 'system',
-                        content: props.systemPrompt.trim()
-                    })
-                }
-
-                const chatRequest = createChatRequest()
-                chatRequest.model = props.model
-
-                console.log('chatRequest', chatRequest, hasImage(), hasAudio(), attachedFiles.value.length, attachedFiles.value)
-
-                function setContentText(chatRequest, text) {
-                    // Replace text message
-                    const textImage = chatRequest.messages.find(m =>
-                        m.role === 'user' && Array.isArray(m.content) && m.content.some(c => c.type === 'text'))
-                    for (const c of textImage.content) {
-                        if (c.type === 'text') {
-                            c.text = text
-                        }
-                    }
-                }
-
-                if (hasImage()) {
-                    const imageMessage = chatRequest.messages.find(m =>
-                        m.role === 'user' && Array.isArray(m.content) && m.content.some(c => c.type === 'image_url'))
-                    console.log('hasImage', chatRequest, imageMessage)
-                    if (imageMessage) {
-                        const imgs = []
-                        let imagePart = deepClone(imageMessage.content.find(c => c.type === 'image_url'))
-                        for (const f of attachedFiles.value) {
-                            if (imageExts.includes(lastRightPart(f.name, '.'))) {
-                                imagePart.image_url.url = await fileToDataUri(f)
-                            }
-                            imgs.push(imagePart)
-                        }
-                        imageMessage.content = imageMessage.content.filter(c => c.type !== 'image_url')
-                        imageMessage.content = [...imgs, ...imageMessage.content]
-                        setContentText(chatRequest, message)
-                    }
-
-                } else if (hasAudio()) {
-                    console.log('hasAudio', chatRequest)
-                    const audioMessage = chatRequest.messages.find(m =>
-                        m.role === 'user' && Array.isArray(m.content) && m.content.some(c => c.type === 'input_audio'))
-                    if (audioMessage) {
-                        const audios = []
-                        let audioPart = deepClone(audioMessage.content.find(c => c.type === 'input_audio'))
-                        for (const f of attachedFiles.value) {
-                            if (audioExts.includes(lastRightPart(f.name, '.'))) {
-                                audioPart.input_audio.data = await fileToBase64(f)
-                            }
-                            audios.push(audioPart)
-                        }
-                        audioMessage.content = audioMessage.content.filter(c => c.type !== 'input_audio')
-                        audioMessage.content = [...audios, ...audioMessage.content]
-                        setContentText(chatRequest, message)
-                    }
-                } else if (attachedFiles.value.length) {
-                    console.log('hasFile', chatRequest)
-                    const fileMessage = chatRequest.messages.find(m =>
-                        m.role === 'user' && Array.isArray(m.content) && m.content.some(c => c.type === 'file'))
-                    if (fileMessage) {
-                        const files = []
-                        let filePart = deepClone(fileMessage.content.find(c => c.type === 'file'))
-                        for (const f of attachedFiles.value) {
-                            filePart.file.file_data = await fileToDataUri(f)
-                            filePart.file.filename = f.name
-                            files.push(filePart)
-                        }
-                        fileMessage.content = fileMessage.content.filter(c => c.type !== 'file')
-                        fileMessage.content = [...files, ...fileMessage.content]
-                        setContentText(chatRequest, message)
-                    }
-
-                } else {
-                    console.log('hasText', chatRequest)
-                    // Chat template message needs to be empty
-                    chatRequest.messages = []
-                    messages.forEach(m => chatRequest.messages.push({
-                        role: m.role,
-                        content: m.content
-                    }))
-                }
-
-                // Send to API
-                const response = await fetch('/v1/chat/completions', {
-                    method: 'POST',
-                    headers: {
-                        'Content-Type': 'application/json'
-                    },
-                    body: JSON.stringify(chatRequest)
-                })
-
-                if (!response.ok) {
-                    errorStatus.value = `HTTP ${response.status} ${response.statusText}`
-                    let errorBody = ''
-                    try {
-                        errorBody = await response.text()
-                        if (errorBody) {
-                            // Try to parse as JSON for better formatting
-                            try {
-                                const errorJson = JSON.parse(errorBody)
-                                errorBody = JSON.stringify(errorJson, null, 2)
-                            } catch (e) {
-                            }
-                        }
-                    } catch (e) {
-                        // If we can't read the response body, just use the status
-                    }
-                    throw new Error(errorBody || '')
-                }
-
-                const result = await response.json()
-
-                if (result.error) {
-                    throw new Error(result.error)
-                }
-
-                // Add assistant response (save entire message including reasoning)
-                const assistantMessage = result.choices?.[0]?.message
-                await threads.addMessageToThread(threadId, assistantMessage)
-
-                nextTick(addCopyButtons)
-
-                attachedFiles.value = []
-
-            } catch (error) {
-                console.error('Error sending message:', error)
-                errorMessage.value = error.message
-
-                // Error will be cleared when user sends next message (no auto-timeout)
-            } finally {
-                isGenerating.value = false
-            }
-        }
-
-        const addNewLine = () => {
-            // Enter key already adds new line
-            //messageText.value += '\n'
-        }
-
-        return {
-            isGenerating,
-            attachedFiles,
-            errorStatus,
-            errorMessage,
-            messageText,
-            fileInput,
-            triggerFilePicker,
-            onFilesSelected,
-            removeAttachment,
-            sendMessage,
-            addNewLine,
-        }
-    }
-}