Compare commits
3
Commits
eff0bee8e0
...
next
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
18eb34f541 | ||
|
|
8ff4ba6c94 | ||
|
|
33236cf6a9 |
@@ -0,0 +1,88 @@
|
||||
import { useLocation } from '@docusaurus/router';
|
||||
import { useCallback, useEffect, useRef, useState } from 'react';
|
||||
|
||||
import { useOptions } from '@docuservix/hooks/useOptions';
|
||||
import { IChat, IChatMessage, IChatSource } from '@docuservix/models/chat';
|
||||
|
||||
interface UseChatResult {
|
||||
dialog: IChat;
|
||||
typing: boolean;
|
||||
statusMessage?: string;
|
||||
sendMessage: (text: string) => void;
|
||||
}
|
||||
|
||||
function useQuery(): string {
|
||||
const location = useLocation();
|
||||
const params = new URLSearchParams(location.search);
|
||||
|
||||
return params.get('q') ?? '';
|
||||
}
|
||||
|
||||
export function useChat(): UseChatResult {
|
||||
const chatEndpoint = useOptions().api + '/v1/chat';
|
||||
const urlQuery = useQuery();
|
||||
|
||||
const [messages, setMessages] = useState<IChatMessage[]>([]);
|
||||
const [loading, setLoading] = useState(false);
|
||||
const [error, setError] = useState<string | null>(null);
|
||||
|
||||
const initialSentRef = useRef(false);
|
||||
const messagesEndRef = useRef(messages);
|
||||
|
||||
messagesEndRef.current = messages;
|
||||
|
||||
const sendMessage = useCallback(
|
||||
async (text: string) => {
|
||||
const content = text.trim();
|
||||
|
||||
if (!content) {
|
||||
return;
|
||||
}
|
||||
|
||||
const userMessage: IChatMessage = { role: 'user', content };
|
||||
const newHistory = [...messagesEndRef.current, userMessage];
|
||||
|
||||
setMessages(newHistory);
|
||||
setLoading(true);
|
||||
setError(null);
|
||||
|
||||
try {
|
||||
const res = await fetch(chatEndpoint, {
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({ messages: newHistory }),
|
||||
});
|
||||
|
||||
if (!res.ok) {
|
||||
throw new Error(`HTTP ${res.status}`);
|
||||
}
|
||||
|
||||
const data: { answer: string; sources?: IChatSource[] } = await res.json();
|
||||
|
||||
setMessages((prev) => [
|
||||
...prev,
|
||||
{ role: 'assistant', content: data.answer, sources: data.sources },
|
||||
]);
|
||||
} catch (err) {
|
||||
setError(err instanceof Error ? err.message : 'Ошибка при обращении к серверу');
|
||||
} finally {
|
||||
setLoading(false);
|
||||
}
|
||||
},
|
||||
[chatEndpoint],
|
||||
);
|
||||
|
||||
useEffect(() => {
|
||||
if (urlQuery && !initialSentRef.current) {
|
||||
initialSentRef.current = true;
|
||||
sendMessage(urlQuery);
|
||||
}
|
||||
}, [urlQuery, sendMessage]);
|
||||
|
||||
return {
|
||||
dialog: { messages },
|
||||
typing: loading,
|
||||
statusMessage: error ?? undefined,
|
||||
sendMessage,
|
||||
};
|
||||
}
|
||||
@@ -0,0 +1,119 @@
|
||||
import { useCallback, useEffect, useRef, useState } from 'react';
|
||||
|
||||
import { useOptions } from '@docuservix/hooks/useOptions';
|
||||
import { ChatStatus } from '@docuservix/models/chat';
|
||||
|
||||
interface UseChatStatusResult {
|
||||
status: ChatStatus;
|
||||
apiUrl: string;
|
||||
recheck: () => void;
|
||||
}
|
||||
|
||||
const POLL_INTERVAL_MS = 30_000;
|
||||
const REQUEST_TIMEOUT_MS = 5_000;
|
||||
|
||||
function sameStatus(a: ChatStatus, b: ChatStatus): boolean {
|
||||
if (a.kind !== b.kind) return false;
|
||||
if (a.kind === 'server' && b.kind === 'server') return a.label === b.label;
|
||||
return true;
|
||||
}
|
||||
|
||||
export function useChatStatus(): UseChatStatusResult {
|
||||
const apiUrl = useOptions().api ?? '';
|
||||
const statusApiUrl = apiUrl + '/v1/status';
|
||||
|
||||
const [status, setStatus] = useState<ChatStatus>({ kind: 'connecting' });
|
||||
|
||||
const abortRef = useRef<AbortController | null>(null);
|
||||
const timerRef = useRef<ReturnType<typeof setTimeout> | null>(null);
|
||||
|
||||
const updateStatus = useCallback((next: ChatStatus): void => {
|
||||
setStatus((prev) => (sameStatus(prev, next) ? prev : next));
|
||||
}, []);
|
||||
|
||||
const check = useCallback(async (): Promise<void> => {
|
||||
abortRef.current?.abort();
|
||||
|
||||
const controller = new AbortController();
|
||||
|
||||
abortRef.current = controller;
|
||||
|
||||
let timedOut = false;
|
||||
const timeoutId = setTimeout(() => {
|
||||
timedOut = true;
|
||||
controller.abort();
|
||||
}, REQUEST_TIMEOUT_MS);
|
||||
|
||||
try {
|
||||
const res = await fetch(statusApiUrl, { signal: controller.signal });
|
||||
|
||||
if (!res.ok) {
|
||||
updateStatus({ kind: 'offline' });
|
||||
return;
|
||||
}
|
||||
|
||||
const parsed = (await res.json()) as { status?: unknown } | null | undefined;
|
||||
const raw = typeof parsed?.status === 'string' ? parsed.status : null;
|
||||
|
||||
if (raw === null || raw.length === 0) {
|
||||
updateStatus({ kind: 'offline' });
|
||||
return;
|
||||
}
|
||||
|
||||
updateStatus({
|
||||
kind: 'server',
|
||||
label: raw.charAt(0).toUpperCase() + raw.slice(1),
|
||||
});
|
||||
} catch {
|
||||
if (!controller.signal.aborted || timedOut) {
|
||||
updateStatus({ kind: 'offline' });
|
||||
}
|
||||
} finally {
|
||||
clearTimeout(timeoutId);
|
||||
}
|
||||
}, [statusApiUrl, updateStatus]);
|
||||
|
||||
const scheduleNext = useCallback((): void => {
|
||||
if (timerRef.current !== null) {
|
||||
clearTimeout(timerRef.current);
|
||||
}
|
||||
|
||||
timerRef.current = setTimeout(async () => {
|
||||
if (document.visibilityState === 'visible') {
|
||||
await check();
|
||||
}
|
||||
scheduleNext();
|
||||
}, POLL_INTERVAL_MS);
|
||||
}, [check]);
|
||||
|
||||
const recheck = useCallback((): void => {
|
||||
updateStatus({ kind: 'connecting' });
|
||||
void check();
|
||||
scheduleNext();
|
||||
}, [check, scheduleNext, updateStatus]);
|
||||
|
||||
useEffect(() => {
|
||||
void check();
|
||||
scheduleNext();
|
||||
|
||||
const onVisibilityChange = (): void => {
|
||||
if (document.visibilityState === 'visible') {
|
||||
void check();
|
||||
scheduleNext();
|
||||
}
|
||||
};
|
||||
|
||||
document.addEventListener('visibilitychange', onVisibilityChange);
|
||||
|
||||
return (): void => {
|
||||
document.removeEventListener('visibilitychange', onVisibilityChange);
|
||||
abortRef.current?.abort();
|
||||
|
||||
if (timerRef.current !== null) {
|
||||
clearTimeout(timerRef.current);
|
||||
}
|
||||
};
|
||||
}, [check, scheduleNext]);
|
||||
|
||||
return { status, apiUrl, recheck };
|
||||
}
|
||||
@@ -0,0 +1,7 @@
|
||||
import { usePluginData } from '@docusaurus/useGlobalData';
|
||||
|
||||
import { DocuservixOptions } from '@docuservix/models/docuservix';
|
||||
|
||||
export function useOptions(): DocuservixOptions {
|
||||
return usePluginData('docuservix') as DocuservixOptions;
|
||||
}
|
||||
@@ -2,7 +2,11 @@ import path from 'path';
|
||||
|
||||
import type { LoadContext, Plugin } from '@docusaurus/types';
|
||||
|
||||
export default function docuservix() {
|
||||
import { DocuservixOptions } from '@docuservix/models/docuservix';
|
||||
|
||||
export default function docuservix(options: Partial<DocuservixOptions> = {}) {
|
||||
const api = process.env.DOCUSERVIX_API || options.api || '/api';
|
||||
|
||||
return function pluginDocuservix(_context: LoadContext): Plugin {
|
||||
return {
|
||||
name: 'docuservix',
|
||||
@@ -18,7 +22,11 @@ export default function docuservix() {
|
||||
},
|
||||
|
||||
async contentLoaded({ actions }) {
|
||||
const { addRoute } = actions;
|
||||
const { addRoute, setGlobalData } = actions;
|
||||
|
||||
setGlobalData({
|
||||
api,
|
||||
});
|
||||
|
||||
addRoute({
|
||||
path: '/chat',
|
||||
|
||||
@@ -1,8 +1,42 @@
|
||||
export type ChatStatus =
|
||||
| { kind: 'connecting' }
|
||||
| { kind: 'offline' }
|
||||
| { kind: 'server'; label: string };
|
||||
|
||||
export interface IChat {
|
||||
messages: IChatMessage[];
|
||||
}
|
||||
|
||||
export interface IChatSource {
|
||||
file: string;
|
||||
heading: string;
|
||||
anchor: string;
|
||||
score: number;
|
||||
}
|
||||
|
||||
export interface IChatMessage {
|
||||
role: 'user' | 'assistant';
|
||||
content: string;
|
||||
sources?: IChatSource[];
|
||||
}
|
||||
|
||||
function stripNumericPrefixes(p: string): string {
|
||||
return p
|
||||
.split('/')
|
||||
.map((seg) => seg.replace(/^\d+-/, ''))
|
||||
.join('/');
|
||||
}
|
||||
|
||||
export function sourceToUrl(file: string, anchor: string): string {
|
||||
let p = file.replace(/^docs\//, '').replace(/\.md$/, '');
|
||||
|
||||
p = stripNumericPrefixes(p);
|
||||
|
||||
return `/docs/${p}${anchor ? `#${anchor}` : ''}`;
|
||||
}
|
||||
|
||||
export function sourceToPath(file: string): string {
|
||||
const p = file.replace(/^docs\//, '').replace(/\.md$/, '');
|
||||
|
||||
return stripNumericPrefixes(p);
|
||||
}
|
||||
|
||||
@@ -0,0 +1,3 @@
|
||||
export interface DocuservixOptions {
|
||||
api?: string;
|
||||
}
|
||||
@@ -1,30 +1,20 @@
|
||||
import Layout from '@theme/Layout';
|
||||
import { ReactNode } from 'react';
|
||||
|
||||
import { IChat } from '@docuservix/models/chat';
|
||||
import { useChat } from '@docuservix/hooks/useChat';
|
||||
import { Chat } from '@docuservix/widgets/chat';
|
||||
|
||||
const dialog: IChat = {
|
||||
messages: [
|
||||
{
|
||||
role: 'user',
|
||||
content: 'Can you show me some CSS animations? It can be simple tools like chatbots...',
|
||||
},
|
||||
{
|
||||
role: 'assistant',
|
||||
content: "Hello! I'm your **AI assistant**. How can I help you today?",
|
||||
},
|
||||
],
|
||||
};
|
||||
|
||||
export function ChatPage(): ReactNode {
|
||||
const { dialog, typing, statusMessage, sendMessage } = useChat();
|
||||
|
||||
return (
|
||||
<Layout title="Чат">
|
||||
<main className="container margin-vert--lg">
|
||||
<Chat
|
||||
dialog={dialog}
|
||||
statusMessage="Unable to connect to the server"
|
||||
typing
|
||||
typing={typing}
|
||||
statusMessage={statusMessage}
|
||||
onSend={sendMessage}
|
||||
/>
|
||||
</main>
|
||||
</Layout>
|
||||
|
||||
@@ -24,9 +24,3 @@
|
||||
color: var(--ifm-font-color-base);
|
||||
font-size: 1.125rem;
|
||||
}
|
||||
|
||||
.Header__info p {
|
||||
margin: 0;
|
||||
color: var(--ifm-color-emphasis-600);
|
||||
font-size: 0.85rem;
|
||||
}
|
||||
|
||||
@@ -3,6 +3,7 @@ import React, { ReactNode } from 'react';
|
||||
|
||||
import styles from './Header.module.css';
|
||||
import { RobotIcon } from './icons';
|
||||
import { Status } from './Status';
|
||||
|
||||
const b = block(styles, 'Header');
|
||||
|
||||
@@ -14,7 +15,7 @@ export function Header(): ReactNode {
|
||||
</div>
|
||||
<div className={b('info')}>
|
||||
<h3>AI Assistant</h3>
|
||||
<p>Ready to help</p>
|
||||
<Status />
|
||||
</div>
|
||||
</div>
|
||||
);
|
||||
|
||||
@@ -34,6 +34,35 @@
|
||||
border-bottom-right-radius: 0.25rem;
|
||||
}
|
||||
|
||||
.Message__sources {
|
||||
margin-top: 8px;
|
||||
padding: 8px 1rem 0;
|
||||
border-top: 1px solid var(--ifm-color-emphasis-200);
|
||||
}
|
||||
|
||||
.Message__sourcesLabel {
|
||||
margin-bottom: 4px;
|
||||
color: var(--ifm-color-emphasis-600);
|
||||
font-weight: var(--ifm-font-weight-semibold);
|
||||
font-size: 0.75rem;
|
||||
letter-spacing: 0.04em;
|
||||
text-transform: uppercase;
|
||||
}
|
||||
|
||||
.Message__sourceLink {
|
||||
display: block;
|
||||
overflow: hidden;
|
||||
color: var(--ifm-color-primary);
|
||||
font-size: 0.8rem;
|
||||
white-space: nowrap;
|
||||
text-decoration: none;
|
||||
text-overflow: ellipsis;
|
||||
}
|
||||
|
||||
.Message__sourceLink:hover {
|
||||
text-decoration: underline;
|
||||
}
|
||||
|
||||
@media (max-width: 576px) {
|
||||
.Message {
|
||||
max-width: 100%;
|
||||
|
||||
@@ -1,8 +1,11 @@
|
||||
import Link from '@docusaurus/Link';
|
||||
import block from 'bem-css-modules';
|
||||
import React, { ReactNode } from 'react';
|
||||
|
||||
import { MD } from '@docuservix/entities/markdown';
|
||||
|
||||
import { IChatSource, sourceToPath, sourceToUrl } from '@docuservix/models/chat';
|
||||
|
||||
import styles from './Message.module.css';
|
||||
|
||||
const b = block(styles, 'Message');
|
||||
@@ -10,14 +13,30 @@ const b = block(styles, 'Message');
|
||||
interface MessageProps {
|
||||
role: 'user' | 'assistant';
|
||||
content: string;
|
||||
sources?: IChatSource[];
|
||||
}
|
||||
|
||||
export function Message({ role, content }: MessageProps): ReactNode {
|
||||
export function Message({ role, content, sources }: MessageProps): ReactNode {
|
||||
return (
|
||||
<div className={b({ role })}>
|
||||
<div className={b('content')}>
|
||||
<MD>{content}</MD>
|
||||
</div>
|
||||
|
||||
{sources && sources.length > 0 && (
|
||||
<div className={b('sources')}>
|
||||
<div className={b('sourcesLabel')}>Источники:</div>
|
||||
{sources.map((src, j) => (
|
||||
<Link
|
||||
key={j}
|
||||
to={sourceToUrl(src.file, src.anchor)}
|
||||
className={b('sourceLink')}
|
||||
>
|
||||
{src.heading || sourceToPath(src.file)}
|
||||
</Link>
|
||||
))}
|
||||
</div>
|
||||
)}
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
@@ -21,6 +21,7 @@ export function Messages({ messages, typing }: MessagesProps): ReactNode {
|
||||
key={i}
|
||||
role={msg.role}
|
||||
content={msg.content}
|
||||
sources={msg.sources}
|
||||
/>
|
||||
))}
|
||||
|
||||
|
||||
@@ -0,0 +1,53 @@
|
||||
.Status {
|
||||
display: inline-flex;
|
||||
align-items: center;
|
||||
gap: 0.4rem;
|
||||
padding: 0;
|
||||
background: none;
|
||||
border: none;
|
||||
cursor: pointer;
|
||||
color: var(--ifm-color-emphasis-600);
|
||||
font-size: 0.85rem;
|
||||
line-height: 1;
|
||||
}
|
||||
|
||||
.Status:hover .Status__label {
|
||||
color: var(--ifm-color-emphasis-800);
|
||||
}
|
||||
|
||||
.Status:focus-visible {
|
||||
outline: 2px solid var(--ifm-color-primary);
|
||||
outline-offset: 2px;
|
||||
border-radius: 4px;
|
||||
}
|
||||
|
||||
.Status__dot {
|
||||
width: 8px;
|
||||
height: 8px;
|
||||
border-radius: 50%;
|
||||
background: var(--ifm-color-emphasis-400);
|
||||
flex-shrink: 0;
|
||||
}
|
||||
|
||||
.Status_kind_server .Status__dot {
|
||||
background: #22c55e;
|
||||
}
|
||||
|
||||
.Status_kind_offline .Status__dot {
|
||||
background: #ef4444;
|
||||
}
|
||||
|
||||
.Status_kind_connecting .Status__dot {
|
||||
background: #eab308;
|
||||
animation: Status__pulse 1s ease-in-out infinite;
|
||||
}
|
||||
|
||||
@keyframes Status__pulse {
|
||||
0%,
|
||||
100% {
|
||||
opacity: 1;
|
||||
}
|
||||
50% {
|
||||
opacity: 0.4;
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,38 @@
|
||||
import block from 'bem-css-modules';
|
||||
import React, { ReactNode } from 'react';
|
||||
|
||||
import { useChatStatus } from '@docuservix/hooks/useChatStatus';
|
||||
import { ChatStatus } from '@docuservix/models/chat';
|
||||
|
||||
import styles from './Status.module.css';
|
||||
|
||||
const b = block(styles, 'Status');
|
||||
|
||||
function getLabel(status: ChatStatus): string {
|
||||
switch (status.kind) {
|
||||
case 'connecting':
|
||||
return 'Подключение…';
|
||||
case 'offline':
|
||||
return 'Offline';
|
||||
case 'server':
|
||||
return status.label;
|
||||
}
|
||||
}
|
||||
|
||||
export function Status(): ReactNode {
|
||||
const { status, apiUrl, recheck } = useChatStatus();
|
||||
const label = getLabel(status);
|
||||
|
||||
return (
|
||||
<button
|
||||
type="button"
|
||||
className={b({ kind: status.kind })}
|
||||
onClick={recheck}
|
||||
title={apiUrl}
|
||||
aria-label={`Статус: ${label}. Нажмите для проверки`}
|
||||
>
|
||||
<span className={b('dot')} />
|
||||
<span className={b('label')}>{label}</span>
|
||||
</button>
|
||||
);
|
||||
}
|
||||
Reference in New Issue
Block a user