kexsh/agent-chat-ui
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

lint and format

Browse Source
This commit is contained in:
starmorph
2025-05-15 16:42:04 -07:00
parent aa32e58602
commit 9ac2228039
3 changed files with 62 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
next.config.mjs
View File

@@ -1,10 +1,10 @@
/** @type {import('next').NextConfig} */
const nextConfig = {
experimental: {
serverActions: {
bodySizeLimit: '10mb',
},
},
experimental: {
serverActions: {
bodySizeLimit: "10mb",
},
},
};
export default nextConfig;

66
src/components/thread/index.tsx
View File

@@ -40,6 +40,11 @@ import {
import type { Base64ContentBlock } from "@/lib/pdf";
type MessageContentType = Message["content"];
interface UploadedBlock {
id: string;
name: string;
block: Base64ContentBlock;
}
function StickyToBottomContent(props: {
content: ReactNode;
@@ -106,12 +111,6 @@ function OpenGitHubRepo() {
);
}
interface UploadedBlock {
id: string;
name: string;
block: Base64ContentBlock;
}
export function Thread() {
const [threadId, setThreadId] = useQueryState("threadId");
const [chatHistoryOpen, setChatHistoryOpen] = useQueryState(
@@ -192,7 +191,6 @@ export function Thread() {
...pdfUrlList.map((item) => item.block),
] as MessageContentType,
};
const toolMessages = ensureToolCallsHaveResponses(stream.messages);
stream.submit(
@@ -240,14 +238,13 @@ export function Thread() {
};
reader.readAsDataURL(file);
});
})
}),
);
setImageUrlList([...imageUrlList, ...imageFiles]);
}
e.target.value = "";
};
const handlePDFUpload = async (e: ChangeEvent<HTMLInputElement>) => {
const files = e.target.files;
if (files) {
@@ -274,7 +271,7 @@ export function Thread() {
};
reader.readAsDataURL(file);
});
})
}),
);
console.log(pdfFiles[0]);
setPdfUrlList([...pdfUrlList, ...pdfFiles]);
@@ -282,9 +279,6 @@ export function Thread() {
e.target.value = "";
};
const handleRegenerate = (
parentCheckpoint: Checkpoint | null | undefined,
) => {
@@ -317,12 +311,17 @@ export function Thread() {
if (!e.dataTransfer) return;
const files = Array.from(e.dataTransfer.files);
const imageFiles = files.filter((file) =>
file.type.startsWith("image/"),
);
const imageFiles = files.filter((file) => file.type.startsWith("image/"));
if (files.some(file => !file.type.startsWith("image/") || file.type !== "application/pdf")) {
toast.error("You have uploaded invalid file type. Please upload an image or a PDF.");
if (
files.some(
(file) =>
!file.type.startsWith("image/") || file.type !== "application/pdf",
)
) {
toast.error(
"You have uploaded invalid file type. Please upload an image or a PDF.",
);
}
/**
@@ -354,7 +353,7 @@ export function Thread() {
};
reader.readAsDataURL(file);
});
})
}),
);
setImageUrlList([...imageUrlList, ...imageFilesData]);
}
@@ -363,8 +362,10 @@ export function Thread() {
* If there are any PDF files in the dropped files, this block previews the file name of each uploaded PDF
* by rendering a list of file names above the input area, with a remove button for each.
*/
if (files.some(file => file.type === "application/pdf")) {
const pdfFiles = files.filter(file => file.type === "application/pdf");
if (files.some((file) => file.type === "application/pdf")) {
const pdfFiles = files.filter(
(file) => file.type === "application/pdf",
);
const pdfFilesData: UploadedBlock[] = await Promise.all(
pdfFiles.map((file) => {
return new Promise<UploadedBlock>((resolve) => {
@@ -373,7 +374,8 @@ export function Thread() {
const result = reader.result as string;
const base64 = result.split(",")[1];
const match = result.match(/^data:(.*);base64/);
const mimeType = match && match[1] ? match[1] : "application/pdf";
const mimeType =
match && match[1] ? match[1] : "application/pdf";
resolve({
id: uuidv4(),
name: file.name,
@@ -388,7 +390,7 @@ export function Thread() {
};
reader.readAsDataURL(file);
});
})
}),
);
setPdfUrlList([...pdfUrlList, ...pdfFilesData]);
}
@@ -418,7 +420,6 @@ export function Thread() {
};
});
return (
<div className="flex h-screen w-full overflow-hidden">
<div className="relative hidden lg:flex">
@@ -640,13 +641,22 @@ export function Thread() {
</div>
)}
{pdfUrlList.length > 0 && (
<div className="flex flex-wrap gap-2 p-3.5 pb-0 ">
<div className="flex flex-wrap gap-2 p-3.5 pb-0">
{pdfUrlList.map((pdf) => (
<div className="relative flex items-center gap-2 bg-gray-100 rounded px-2 py-1 border-1 border-teal-700 bg-teal-900 text-white rounded-md px-2 py-2" key={pdf.id}>
<span className=" truncate max-w-xs text-sm">{pdf.name}</span>
<div
className="relative flex items-center gap-2 rounded rounded-md border-1 border-teal-700 bg-gray-100 bg-teal-900 px-2 py-1 py-2 text-white"
key={pdf.id}
>
<span className="max-w-xs truncate text-sm">
{pdf.name}
</span>
<CircleX
className="size-4 cursor-pointer text-teal-600 hover:text-teal-500"
onClick={() => setPdfUrlList(pdfUrlList.filter((p) => p.id !== pdf.id))}
onClick={() =>
setPdfUrlList(
pdfUrlList.filter((p) => p.id !== pdf.id),
)
}
/>
</div>
))}

40
src/lib/pdf.ts
View File

@@ -1,30 +1,28 @@
"use server"
import { MessageContentText } from "@langchain/core/messages";
"use server";
import { MessageContentText } from "@langchain/core/messages";
import { WebPDFLoader } from "@langchain/community/document_loaders/web/pdf";
// import { Base64ContentBlock } from "@langchain/core/messages";
// switch local import with above import
export interface Base64ContentBlock {
data: string;
metadata?: Record<string, unknown>;
mime_type?: string;
source_type: "base64";
type: "image" | "audio" | "file";
data: string;
metadata?: Record<string, unknown>;
mime_type?: string;
source_type: "base64";
type: "image" | "audio" | "file";
}
export const extractPdfText = async (file: File): Promise<MessageContentText> => {
const loader = new WebPDFLoader(file, { splitPages: false });
const docs = await loader.load();
return {
type: "text",
text: docs[0].pageContent,
};
export const extractPdfText = async (
file: File,
): Promise<MessageContentText> => {
const loader = new WebPDFLoader(file, { splitPages: false });
const docs = await loader.load();
return {
type: "text",
text: docs[0].pageContent,
};
};
const cleanBase64 = (base64String: string): string => {
return base64String.replace(/^data:.*?;base64,/, "");
};
const cleanBase64 = (base64String: string): string => {
return base64String.replace(/^data:.*?;base64,/, "");
};