fix: Modality detection improvement for text-based PDF attachments
This commit is contained in:
parent
171a0926a1
commit
dd30810d0a
|
|
@ -8,6 +8,7 @@ import { AttachmentType } from '$lib/enums';
|
||||||
import type {
|
import type {
|
||||||
DatabaseConversation,
|
DatabaseConversation,
|
||||||
DatabaseMessage,
|
DatabaseMessage,
|
||||||
|
DatabaseMessageExtraPdfFile,
|
||||||
ExportedConversations
|
ExportedConversations
|
||||||
} from '$lib/types/database';
|
} from '$lib/types/database';
|
||||||
import type { ModelModalities } from '$lib/types/models';
|
import type { ModelModalities } from '$lib/types/models';
|
||||||
|
|
@ -95,9 +96,19 @@ class ConversationsStore {
|
||||||
if (!message.extra) continue;
|
if (!message.extra) continue;
|
||||||
|
|
||||||
for (const extra of message.extra) {
|
for (const extra of message.extra) {
|
||||||
if (extra.type === AttachmentType.IMAGE || extra.type === AttachmentType.PDF) {
|
if (extra.type === AttachmentType.IMAGE) {
|
||||||
modalities.vision = true;
|
modalities.vision = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// PDF only requires vision if processed as images
|
||||||
|
if (extra.type === AttachmentType.PDF) {
|
||||||
|
const pdfExtra = extra as DatabaseMessageExtraPdfFile;
|
||||||
|
|
||||||
|
if (pdfExtra.processedAsImages) {
|
||||||
|
modalities.vision = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
if (extra.type === AttachmentType.AUDIO) {
|
if (extra.type === AttachmentType.AUDIO) {
|
||||||
modalities.audio = true;
|
modalities.audio = true;
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue