12 | New
16 | {title}
17 |
21 |
--------------------------------------------------------------------------------
/src/lib/components/CodeBlock.svelte:
--------------------------------------------------------------------------------
1 |
12 |
13 |
18 | {@render children?.()}
19 |
20 |
14 |
23 |
--------------------------------------------------------------------------------
/src/lib/components/ContinueBtn.svelte:
--------------------------------------------------------------------------------
1 |
11 |
12 |
19 |
--------------------------------------------------------------------------------
/src/lib/components/ExpandNavigation.svelte:
--------------------------------------------------------------------------------
1 |
10 |
11 |
22 |
--------------------------------------------------------------------------------
/src/lib/components/HoverTooltip.svelte:
--------------------------------------------------------------------------------
1 |
18 |
19 | {@html DOMPurify.sanitize(code)}
20 | {@render children?.()}
21 |
22 |
45 |
--------------------------------------------------------------------------------
/src/lib/components/InfiniteScroll.svelte:
--------------------------------------------------------------------------------
1 |
45 |
46 |
42 | {label}
43 |
44 |
47 |
48 |
49 |
50 |
51 |
--------------------------------------------------------------------------------
/src/lib/components/PaginationArrow.svelte:
--------------------------------------------------------------------------------
1 |
13 |
14 |
20 | {#if direction === "previous"}
21 |
23 | {@render children?.()}
24 |
25 |
--------------------------------------------------------------------------------
/src/lib/components/RetryBtn.svelte:
--------------------------------------------------------------------------------
1 |
11 |
12 |
19 |
--------------------------------------------------------------------------------
/src/lib/components/ScrollToBottomBtn.svelte:
--------------------------------------------------------------------------------
1 |
39 |
40 | {#if visible}
41 |
47 | {/if}
48 |
--------------------------------------------------------------------------------
/src/lib/components/StopGeneratingBtn.svelte:
--------------------------------------------------------------------------------
1 |
11 |
12 |
19 |
--------------------------------------------------------------------------------
/src/lib/components/Switch.svelte:
--------------------------------------------------------------------------------
1 |
9 |
10 |
11 |
19 |
20 |
21 |
--------------------------------------------------------------------------------
/src/lib/components/SystemPromptModal.svelte:
--------------------------------------------------------------------------------
1 |
14 |
15 |
23 |
24 | {#if isOpen}
25 |
27 |
39 |
28 |
33 |
38 | System Prompt
29 | 32 |
18 |
25 |
21 |
24 | {message}
23 |
34 |
45 |
--------------------------------------------------------------------------------
/src/lib/components/ToolBadge.svelte:
--------------------------------------------------------------------------------
1 |
12 |
13 | 37 | {nTokens}{truncate ? `/${truncate}` : ""} 38 |
39 |
42 | Tokens usage
43 |
44 |
16 | {#if browser}
17 | {#await fetch(`${base}/api/tools/${toolId}`).then((res) => res.json()) then value}
18 | {#key value.color + value.icon}
19 |
42 |
--------------------------------------------------------------------------------
/src/lib/components/Tooltip.svelte:
--------------------------------------------------------------------------------
1 |
14 |
15 |
22 | {value.displayName}
28 | {#if value.createdByName}
29 |
39 | {/await}
40 | {/if}
41 | 30 | Created by 31 | {value.createdByName} 34 |
35 | {:else} 36 |Official HuggingChat tool
37 | {/if} 38 |
22 |
29 | {label}
30 |
31 |
--------------------------------------------------------------------------------
/src/lib/components/UploadBtn.svelte:
--------------------------------------------------------------------------------
1 |
22 |
23 |
35 |
--------------------------------------------------------------------------------
/src/lib/components/WebSearchToggle.svelte:
--------------------------------------------------------------------------------
1 |
8 |
9 |
18 |
34 |
--------------------------------------------------------------------------------
/src/lib/components/chat/Vote.svelte:
--------------------------------------------------------------------------------
1 |
18 |
19 |
30 |
41 |
--------------------------------------------------------------------------------
/src/lib/components/icons/IconChevron.svelte:
--------------------------------------------------------------------------------
1 |
8 |
9 |
25 |
--------------------------------------------------------------------------------
/src/lib/components/icons/IconCopy.svelte:
--------------------------------------------------------------------------------
1 |
8 |
9 |
31 |
--------------------------------------------------------------------------------
/src/lib/components/icons/IconInternet.svelte:
--------------------------------------------------------------------------------
1 |
8 |
9 |
28 |
--------------------------------------------------------------------------------
/src/lib/components/icons/IconLoading.svelte:
--------------------------------------------------------------------------------
1 |
8 |
9 |
23 |
33 |
27 |
32 | 28 | When enabled, the model will try to complement its answer with information queried from the 29 | web. 30 |
31 |
10 |
14 |
18 |
22 |
23 |
--------------------------------------------------------------------------------
/src/lib/components/icons/IconNew.svelte:
--------------------------------------------------------------------------------
1 |
8 |
9 |
23 |
--------------------------------------------------------------------------------
/src/lib/components/icons/IconPaperclip.svelte:
--------------------------------------------------------------------------------
1 |
8 |
9 |
25 |
--------------------------------------------------------------------------------
/src/lib/components/icons/IconScreenshot.svelte:
--------------------------------------------------------------------------------
1 |
8 |
9 |
25 |
--------------------------------------------------------------------------------
/src/lib/components/icons/Logo.svelte:
--------------------------------------------------------------------------------
1 |
13 |
14 | 
44 | {/if}
45 |
--------------------------------------------------------------------------------
/src/lib/constants/pagination.ts:
--------------------------------------------------------------------------------
1 | export const CONV_NUM_PER_PAGE = 30;
2 |
--------------------------------------------------------------------------------
/src/lib/constants/publicSepToken.ts:
--------------------------------------------------------------------------------
1 | export const PUBLIC_SEP_TOKEN = "";
2 |
--------------------------------------------------------------------------------
/src/lib/migrations/lock.ts:
--------------------------------------------------------------------------------
1 | import { collections } from "$lib/server/database";
2 | import { ObjectId } from "mongodb";
3 | import type { Semaphores } from "$lib/types/Semaphore";
4 |
5 | /**
6 | * Returns the lock id if the lock was acquired, false otherwise
7 | */
8 | export async function acquireLock(key: Semaphores): Promise
8 |
21 |
--------------------------------------------------------------------------------
/src/routes/admin/stats/compute/+server.ts:
--------------------------------------------------------------------------------
1 | import { json } from "@sveltejs/kit";
2 | import { logger } from "$lib/server/logger";
3 | import { computeAllStats } from "$lib/jobs/refresh-conversation-stats";
4 |
5 | // Triger like this:
6 | // curl -X POST "http://localhost:5173/chat/admin/stats/compute" -H "Authorization: Bearer
11 |
20 | {page.status}
12 | 13 |{page.error?.message}
14 | {#if page.error?.errorId} 15 | 16 |{page.error
17 | .errorId}
18 | {/if}
19 |
14 |
44 |
--------------------------------------------------------------------------------
/src/routes/conversation/[id]/message/[messageId]/vote/+server.ts:
--------------------------------------------------------------------------------
1 | import { authCondition } from "$lib/server/auth";
2 | import { collections } from "$lib/server/database";
3 | import { MetricsServer } from "$lib/server/metrics.js";
4 | import { error } from "@sveltejs/kit";
5 | import { ObjectId } from "mongodb";
6 | import { z } from "zod";
7 |
8 | export async function POST({ params, request, locals }) {
9 | const { score } = z
10 | .object({
11 | score: z.number().int().min(-1).max(1),
12 | })
13 | .parse(await request.json());
14 | const conversationId = new ObjectId(params.id);
15 | const messageId = params.messageId;
16 |
17 | // aggregate votes per model in order to detect model performance degradation
18 | const model = await collections.conversations
19 | .findOne(
20 | {
21 | _id: conversationId,
22 | ...authCondition(locals),
23 | },
24 | { projection: { model: 1 } }
25 | )
26 | .then((c) => c?.model);
27 |
28 | if (model) {
29 | if (score === 1) {
30 | MetricsServer.getMetrics().model.votesPositive.inc({ model });
31 | } else {
32 | MetricsServer.getMetrics().model.votesNegative.inc({ model });
33 | }
34 | }
35 |
36 | const document = await collections.conversations.updateOne(
37 | {
38 | _id: conversationId,
39 | ...authCondition(locals),
40 | "messages.id": messageId,
41 | },
42 | {
43 | ...(score !== 0
44 | ? {
45 | $set: {
46 | "messages.$.score": score,
47 | },
48 | }
49 | : { $unset: { "messages.$.score": "" } }),
50 | }
51 | );
52 |
53 | if (!document.matchedCount) {
54 | error(404, "Message not found");
55 | }
56 |
57 | return new Response();
58 | }
59 |
--------------------------------------------------------------------------------
/src/routes/conversation/[id]/stop-generating/+server.ts:
--------------------------------------------------------------------------------
1 | import { authCondition } from "$lib/server/auth";
2 | import { collections } from "$lib/server/database";
3 | import { error } from "@sveltejs/kit";
4 | import { ObjectId } from "mongodb";
5 |
6 | /**
7 | * Ideally, we'd be able to detect the client-side abort, see https://github.com/huggingface/chat-ui/pull/88#issuecomment-1523173850
8 | */
9 | export async function POST({ params, locals }) {
10 | const conversationId = new ObjectId(params.id);
11 |
12 | const conversation = await collections.conversations.findOne({
13 | _id: conversationId,
14 | ...authCondition(locals),
15 | });
16 |
17 | if (!conversation) {
18 | error(404, "Conversation not found");
19 | }
20 |
21 | await collections.abortedGenerations.updateOne(
22 | { conversationId },
23 | { $set: { updatedAt: new Date() }, $setOnInsert: { createdAt: new Date() } },
24 | { upsert: true }
25 | );
26 |
27 | return new Response();
28 | }
29 |
--------------------------------------------------------------------------------
/src/routes/healthcheck/+server.ts:
--------------------------------------------------------------------------------
1 | export async function GET() {
2 | return new Response("OK", { status: 200 });
3 | }
4 |
--------------------------------------------------------------------------------
/src/routes/login/+page.server.ts:
--------------------------------------------------------------------------------
1 | import { redirect } from "@sveltejs/kit";
2 | import { getOIDCAuthorizationUrl } from "$lib/server/auth";
3 | import { base } from "$app/paths";
4 | import { config } from "$lib/server/config";
5 |
6 | export const actions = {
7 | async default({ url, locals, request }) {
8 | const referer = request.headers.get("referer");
9 | let redirectURI = `${(referer ? new URL(referer) : url).origin}${base}/login/callback`;
10 |
11 | // TODO: Handle errors if provider is not responding
12 |
13 | if (url.searchParams.has("callback")) {
14 | const callback = url.searchParams.get("callback") || redirectURI;
15 | if (config.ALTERNATIVE_REDIRECT_URLS.includes(callback)) {
16 | redirectURI = callback;
17 | }
18 | }
19 |
20 | const authorizationUrl = await getOIDCAuthorizationUrl(
21 | { redirectURI },
22 | { sessionId: locals.sessionId }
23 | );
24 |
25 | redirect(303, authorizationUrl);
26 | },
27 | };
28 |
--------------------------------------------------------------------------------
/src/routes/logout/+page.server.ts:
--------------------------------------------------------------------------------
1 | import { dev } from "$app/environment";
2 | import { base } from "$app/paths";
3 | import { config } from "$lib/server/config";
4 | import { collections } from "$lib/server/database";
5 | import { redirect } from "@sveltejs/kit";
6 |
7 | export const actions = {
8 | async default({ cookies, locals }) {
9 | await collections.sessions.deleteOne({ sessionId: locals.sessionId });
10 |
11 | cookies.delete(config.COOKIE_NAME, {
12 | path: "/",
13 | // So that it works inside the space's iframe
14 | sameSite: dev || config.ALLOW_INSECURE_COOKIES === "true" ? "lax" : "none",
15 | secure: !dev && !(config.ALLOW_INSECURE_COOKIES === "true"),
16 | httpOnly: true,
17 | });
18 | redirect(303, `${base}/`);
19 | },
20 | };
21 |
--------------------------------------------------------------------------------
/src/routes/models/[...model]/+page.server.ts:
--------------------------------------------------------------------------------
1 | import { base } from "$app/paths";
2 | import { authCondition } from "$lib/server/auth.js";
3 | import { collections } from "$lib/server/database";
4 | import { models } from "$lib/server/models";
5 | import { redirect } from "@sveltejs/kit";
6 |
7 | export async function load({ params, locals, parent }) {
8 | const model = models.find(({ id }) => id === params.model);
9 | const data = await parent();
10 |
11 | if (!model || model.unlisted) {
12 | redirect(302, `${base}/`);
13 | }
14 |
15 | if (locals.user?._id ?? locals.sessionId) {
16 | await collections.settings.updateOne(
17 | authCondition(locals),
18 | {
19 | $set: {
20 | activeModel: model.id,
21 | updatedAt: new Date(),
22 | },
23 | $setOnInsert: {
24 | createdAt: new Date(),
25 | },
26 | },
27 | {
28 | upsert: true,
29 | }
30 | );
31 | }
32 |
33 | return {
34 | settings: {
35 | ...data.settings,
36 | activeModel: model.id,
37 | },
38 | };
39 | }
40 |
--------------------------------------------------------------------------------
/src/routes/models/[...model]/thumbnail.png/+server.ts:
--------------------------------------------------------------------------------
1 | import ModelThumbnail from "./ModelThumbnail.svelte";
2 | import { redirect, type RequestHandler } from "@sveltejs/kit";
3 |
4 | import { Resvg } from "@resvg/resvg-js";
5 | import satori from "satori";
6 | import { html } from "satori-html";
7 |
8 | import InterRegular from "$lib/server/fonts/Inter-Regular.ttf";
9 | import InterBold from "$lib/server/fonts/Inter-Bold.ttf";
10 | import { base } from "$app/paths";
11 | import { models } from "$lib/server/models";
12 | import { render } from "svelte/server";
13 |
14 | export const GET: RequestHandler = (async ({ params }) => {
15 | const model = models.find(({ id }) => id === params.model);
16 |
17 | if (!model || model.unlisted) {
18 | redirect(302, `${base}/`);
19 | }
20 | const renderedComponent = render(ModelThumbnail, {
21 | props: {
22 | name: model.name,
23 | logoUrl: model.logoUrl,
24 | },
25 | });
26 |
27 | const reactLike = html("" + renderedComponent.body);
28 |
29 | const svg = await satori(reactLike, {
30 | width: 1200,
31 | height: 648,
32 | fonts: [
33 | {
34 | name: "Inter",
35 | data: InterRegular as unknown as ArrayBuffer,
36 | weight: 500,
37 | },
38 | {
39 | name: "Inter",
40 | data: InterBold as unknown as ArrayBuffer,
41 | weight: 700,
42 | },
43 | ],
44 | });
45 |
46 | const png = new Resvg(svg, {
47 | fitTo: { mode: "original" },
48 | })
49 | .render()
50 | .asPng();
51 |
52 | return new Response(png, {
53 | headers: {
54 | "Content-Type": "image/png",
55 | },
56 | });
57 | }) satisfies RequestHandler;
58 |
--------------------------------------------------------------------------------
/src/routes/models/[...model]/thumbnail.png/ModelThumbnail.svelte:
--------------------------------------------------------------------------------
1 |
12 |
13 |
15 | {#if avatar}
16 | 
17 | {/if}
18 |
38 | {#if createdByName}
39 |
19 |
37 | 20 | 21 | 22 | {@html logo} 23 | 24 | AI assistant 25 |
26 |27 | {name} 28 |
29 |30 | {description.slice(0, 160)} 31 | {#if description.length > 160}...{/if} 32 |
33 |
34 | Start chatting
35 |
36 | 40 | An AI assistant created by {createdByName} 41 |
42 | {/if} 43 |
14 |
43 |
--------------------------------------------------------------------------------
/src/routes/privacy/+page.svelte:
--------------------------------------------------------------------------------
1 |
5 |
6 |
15 | {#if logoUrl}
16 | 
17 | {/if}
18 |
22 |
23 | 19 | {name} 20 |
21 |
27 | Try it now
28 | {#if publicConfig.isHuggingChat}
29 | on
30 | {/if}
31 |
32 | {#if publicConfig.isHuggingChat}
33 |
42 |
34 |
40 | {/if}
41 |
35 |
36 | {@html logo}
37 |
38 | HuggingChat
39 |
7 |
12 |
--------------------------------------------------------------------------------
/src/routes/r/[id]/+page.ts:
--------------------------------------------------------------------------------
1 | import { redirect, type LoadEvent } from "@sveltejs/kit";
2 |
3 | export const load = async ({ params, url }: LoadEvent) => {
4 | const leafId = url.searchParams.get("leafId");
5 |
6 | redirect(302, "../conversation/" + params.id + `?leafId=${leafId}`);
7 | };
8 |
--------------------------------------------------------------------------------
/src/routes/settings/(nav)/+layout.ts:
--------------------------------------------------------------------------------
1 | export const ssr = false;
2 |
--------------------------------------------------------------------------------
/src/routes/settings/(nav)/+page.svelte:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/huggingface/chat-ui/21dd9683d3e2670779ff974adf6e109880d414a6/src/routes/settings/(nav)/+page.svelte
--------------------------------------------------------------------------------
/src/routes/settings/(nav)/[...model]/+page.ts:
--------------------------------------------------------------------------------
1 | import { base } from "$app/paths";
2 | import { redirect } from "@sveltejs/kit";
3 |
4 | export async function load({ parent, params }) {
5 | const data = await parent();
6 |
7 | const model = data.models.find((m: { id: string }) => m.id === params.model);
8 |
9 | if (!model || model.unlisted) {
10 | redirect(302, `${base}/settings`);
11 | }
12 |
13 | return data;
14 | }
15 |
--------------------------------------------------------------------------------
/src/routes/settings/(nav)/assistants/[assistantId]/+page.ts:
--------------------------------------------------------------------------------
1 | import { base } from "$app/paths";
2 | import { redirect } from "@sveltejs/kit";
3 |
4 | export async function load({ parent, params }) {
5 | const data = await parent();
6 |
7 | const assistant = data.settings.assistants.find((id) => id === params.assistantId);
8 |
9 | if (!assistant) {
10 | redirect(302, `${base}/assistant/${params.assistantId}`);
11 | }
12 |
13 | return data;
14 | }
15 |
--------------------------------------------------------------------------------
/src/routes/settings/(nav)/assistants/[assistantId]/avatar.jpg/+server.ts:
--------------------------------------------------------------------------------
1 | import { collections } from "$lib/server/database";
2 | import { error, type RequestHandler } from "@sveltejs/kit";
3 | import { ObjectId } from "mongodb";
4 |
5 | export const GET: RequestHandler = async ({ params }) => {
6 | const assistant = await collections.assistants.findOne({
7 | _id: new ObjectId(params.assistantId),
8 | });
9 |
10 | if (!assistant) {
11 | error(404, "No assistant found");
12 | }
13 |
14 | if (!assistant.avatar) {
15 | error(404, "No avatar found");
16 | }
17 |
18 | const fileId = collections.bucket.find({ filename: assistant._id.toString() });
19 |
20 | const content = await fileId.next().then(async (file) => {
21 | if (!file?._id) {
22 | error(404, "Avatar not found");
23 | }
24 |
25 | const fileStream = collections.bucket.openDownloadStream(file?._id);
26 |
27 | const fileBuffer = await new Promise
8 |
9 | {@html marked(privacy, { gfm: true })}
10 |
11 |
35 |
38 | {/if}
39 |