Compare commits
7 Commits
next
..
618a8b2b3c
| Author | SHA1 | Date | |
|---|---|---|---|
| 618a8b2b3c | |||
| f023496040 | |||
| c7c353db0a | |||
| 08642e46d4 | |||
| 987415eafd | |||
| e3a2734f03 | |||
| a038e7eaae |
@@ -1,88 +0,0 @@
|
|||||||
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 + '/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,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
@@ -1,7 +0,0 @@
|
|||||||
import { usePluginData } from '@docusaurus/useGlobalData';
|
|
||||||
|
|
||||||
import { DocuservixOptions } from '@docuservix/models/docuservix';
|
|
||||||
|
|
||||||
export function useOptions(): DocuservixOptions {
|
|
||||||
return usePluginData('docuservix') as DocuservixOptions;
|
|
||||||
}
|
|
||||||
@@ -2,11 +2,7 @@ import path from 'path';
|
|||||||
|
|
||||||
import type { LoadContext, Plugin } from '@docusaurus/types';
|
import type { LoadContext, Plugin } from '@docusaurus/types';
|
||||||
|
|
||||||
import { DocuservixOptions } from '@docuservix/models/docuservix';
|
export default function docuservix() {
|
||||||
|
|
||||||
export default function docuservix(options: Partial<DocuservixOptions> = {}) {
|
|
||||||
const { api = '/api' } = options;
|
|
||||||
|
|
||||||
return function pluginDocuservix(_context: LoadContext): Plugin {
|
return function pluginDocuservix(_context: LoadContext): Plugin {
|
||||||
return {
|
return {
|
||||||
name: 'docuservix',
|
name: 'docuservix',
|
||||||
@@ -22,11 +18,7 @@ export default function docuservix(options: Partial<DocuservixOptions> = {}) {
|
|||||||
},
|
},
|
||||||
|
|
||||||
async contentLoaded({ actions }) {
|
async contentLoaded({ actions }) {
|
||||||
const { addRoute, setGlobalData } = actions;
|
const { addRoute } = actions;
|
||||||
|
|
||||||
setGlobalData({
|
|
||||||
api,
|
|
||||||
});
|
|
||||||
|
|
||||||
addRoute({
|
addRoute({
|
||||||
path: '/chat',
|
path: '/chat',
|
||||||
|
|||||||
@@ -2,36 +2,7 @@ export interface IChat {
|
|||||||
messages: IChatMessage[];
|
messages: IChatMessage[];
|
||||||
}
|
}
|
||||||
|
|
||||||
export interface IChatSource {
|
|
||||||
file: string;
|
|
||||||
heading: string;
|
|
||||||
anchor: string;
|
|
||||||
score: number;
|
|
||||||
}
|
|
||||||
|
|
||||||
export interface IChatMessage {
|
export interface IChatMessage {
|
||||||
role: 'user' | 'assistant';
|
role: 'user' | 'assistant';
|
||||||
content: string;
|
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);
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,3 +0,0 @@
|
|||||||
export interface DocuservixOptions {
|
|
||||||
api?: string;
|
|
||||||
}
|
|
||||||
@@ -1,20 +1,30 @@
|
|||||||
import Layout from '@theme/Layout';
|
import Layout from '@theme/Layout';
|
||||||
import { ReactNode } from 'react';
|
import { ReactNode } from 'react';
|
||||||
|
|
||||||
import { useChat } from '@docuservix/hooks/useChat';
|
import { IChat } from '@docuservix/models/chat';
|
||||||
import { Chat } from '@docuservix/widgets/chat';
|
import { Chat } from '@docuservix/widgets/chat';
|
||||||
|
|
||||||
export function ChatPage(): ReactNode {
|
const dialog: IChat = {
|
||||||
const { dialog, typing, statusMessage, sendMessage } = useChat();
|
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 {
|
||||||
return (
|
return (
|
||||||
<Layout title="Чат">
|
<Layout title="Чат">
|
||||||
<main className="container margin-vert--lg">
|
<main className="container margin-vert--lg">
|
||||||
<Chat
|
<Chat
|
||||||
dialog={dialog}
|
dialog={dialog}
|
||||||
typing={typing}
|
statusMessage="Unable to connect to the server"
|
||||||
statusMessage={statusMessage}
|
typing
|
||||||
onSend={sendMessage}
|
|
||||||
/>
|
/>
|
||||||
</main>
|
</main>
|
||||||
</Layout>
|
</Layout>
|
||||||
|
|||||||
@@ -34,33 +34,28 @@
|
|||||||
border-bottom-right-radius: 0.25rem;
|
border-bottom-right-radius: 0.25rem;
|
||||||
}
|
}
|
||||||
|
|
||||||
.Message__sources {
|
/* User role overrides for light-on-dark text */
|
||||||
margin-top: 8px;
|
|
||||||
padding: 8px 1rem 0;
|
.Message_role_user .Message__content code {
|
||||||
border-top: 1px solid var(--ifm-color-emphasis-200);
|
background: rgba(255, 255, 255, 0.2);
|
||||||
}
|
}
|
||||||
|
|
||||||
.Message__sourcesLabel {
|
.Message_role_user .Message__content pre {
|
||||||
margin-bottom: 4px;
|
background: rgba(255, 255, 255, 0.15);
|
||||||
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 {
|
.Message_role_user .Message__content th,
|
||||||
display: block;
|
.Message_role_user .Message__content td {
|
||||||
overflow: hidden;
|
border-color: rgba(255, 255, 255, 0.3);
|
||||||
color: var(--ifm-color-primary);
|
|
||||||
font-size: 0.8rem;
|
|
||||||
white-space: nowrap;
|
|
||||||
text-decoration: none;
|
|
||||||
text-overflow: ellipsis;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
.Message__sourceLink:hover {
|
.Message_role_user .Message__content th {
|
||||||
text-decoration: underline;
|
background: rgba(255, 255, 255, 0.15);
|
||||||
|
}
|
||||||
|
|
||||||
|
.Message_role_user .Message__content blockquote {
|
||||||
|
border-left-color: rgba(255, 255, 255, 0.4);
|
||||||
|
color: rgba(255, 255, 255, 0.85);
|
||||||
}
|
}
|
||||||
|
|
||||||
@media (max-width: 576px) {
|
@media (max-width: 576px) {
|
||||||
|
|||||||
@@ -1,11 +1,8 @@
|
|||||||
import Link from '@docusaurus/Link';
|
|
||||||
import block from 'bem-css-modules';
|
import block from 'bem-css-modules';
|
||||||
import React, { ReactNode } from 'react';
|
import React, { ReactNode } from 'react';
|
||||||
|
|
||||||
import { MD } from '@docuservix/entities/markdown';
|
import { MD } from '@docuservix/entities/markdown';
|
||||||
|
|
||||||
import { IChatSource, sourceToPath, sourceToUrl } from '@docuservix/models/chat';
|
|
||||||
|
|
||||||
import styles from './Message.module.css';
|
import styles from './Message.module.css';
|
||||||
|
|
||||||
const b = block(styles, 'Message');
|
const b = block(styles, 'Message');
|
||||||
@@ -13,30 +10,14 @@ const b = block(styles, 'Message');
|
|||||||
interface MessageProps {
|
interface MessageProps {
|
||||||
role: 'user' | 'assistant';
|
role: 'user' | 'assistant';
|
||||||
content: string;
|
content: string;
|
||||||
sources?: IChatSource[];
|
|
||||||
}
|
}
|
||||||
|
|
||||||
export function Message({ role, content, sources }: MessageProps): ReactNode {
|
export function Message({ role, content }: MessageProps): ReactNode {
|
||||||
return (
|
return (
|
||||||
<div className={b({ role })}>
|
<div className={b({ role })}>
|
||||||
<div className={b('content')}>
|
<div className={b('content')}>
|
||||||
<MD>{content}</MD>
|
<MD>{content}</MD>
|
||||||
</div>
|
</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>
|
</div>
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -21,7 +21,6 @@ export function Messages({ messages, typing }: MessagesProps): ReactNode {
|
|||||||
key={i}
|
key={i}
|
||||||
role={msg.role}
|
role={msg.role}
|
||||||
content={msg.content}
|
content={msg.content}
|
||||||
sources={msg.sources}
|
|
||||||
/>
|
/>
|
||||||
))}
|
))}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user