@gendive/chatllm 0.4.0 → 0.5.0

package/dist/react/index.d.mts

@@ -57,6 +57,14 @@ interface AlternativeResponse {
     model: string;
     content: string;
     timestamp: number;
+    sources?: SourceItem[];
+}
+interface SourceItem {
+    id: string;
+    title: string;
+    url: string;
+    snippet?: string;
+    favicon?: string;
 }
 interface ChatMessage {
     id: string;
@@ -65,6 +73,7 @@ interface ChatMessage {
     model?: string;
     timestamp: number;
     alternatives?: AlternativeResponse[];
+    sources?: SourceItem[];
 }
 interface ChatSession {
     id: string;
@@ -139,7 +148,7 @@ interface ChatUIProps {
     /** CSS 클래스 */
     className?: string;
     /** 메시지 전송 핸들러 (커스텀 API 사용 시) */
-    onSendMessage?: (params: SendMessageParams) => Promise<ReadableStream<Uint8Array> | string>;
+    onSendMessage?: (params: SendMessageParams) => Promise<SendMessageResponse>;
     /** 세션 변경 핸들러 */
     onSessionChange?: (session: ChatSession | null) => void;
     /** 에러 핸들러 */
@@ -155,6 +164,11 @@ interface SendMessageParams {
     apiKey?: string;
     systemPrompt?: string;
 }
+/** onSendMessage 반환 타입 */
+type SendMessageResponse = ReadableStream<Uint8Array> | string | {
+    content: string;
+    sources?: SourceItem[];
+};
 interface ChatUIComponents {
     /** 사이드바 컴포넌트 */
     Sidebar?: React.ComponentType<SidebarProps>;
@@ -322,7 +336,7 @@ interface UseChatUIOptions {
     /** 압축 후 유지할 메시지 수 */
     keepRecentMessages?: number;
     /** 커스텀 메시지 전송 핸들러 */
-    onSendMessage?: (params: SendMessageParams) => Promise<ReadableStream<Uint8Array> | string>;
+    onSendMessage?: (params: SendMessageParams) => Promise<SendMessageResponse>;
     /** 세션 변경 핸들러 */
     onSessionChange?: (session: ChatSession | null) => void;
     /** 에러 핸들러 */
@@ -493,4 +507,4 @@ interface SettingsModalProps {
 }
 declare const SettingsModal: React$1.FC<SettingsModalProps>;
 
-export { type ActionItem, type ActionMenuProps, type AlternativeResponse, ChatHeader, ChatInput, type ChatMessage, type ChatSession, ChatSidebar, ChatUI, type ChatUIComponents, type ChatUIProps, EmptyState, type EmptyStateProps, type HeaderProps, Icon, type IconName, type IconProps, IconSvg, type InputProps, LinkChip, type LinkChipProps, MarkdownRenderer, type MarkdownRendererProps, type MemoryItem, MemoryPanel, type MemoryPanelProps, MessageBubble, type MessageBubbleProps, MessageList, type MessageListProps, type ModelConfig, type ModelSelectorProps, type PersonalizationConfig, type PromptTemplate, type ProviderType, type ResponseStyle, type SendMessageParams, SettingsModal, type SettingsModalProps, type SettingsTab, type SidebarProps, type ThemeConfig, type ThemeMode, type UseChatUIOptions, type UseChatUIReturn, type UserProfile, useChatUI };
+export { type ActionItem, type ActionMenuProps, type AlternativeResponse, ChatHeader, ChatInput, type ChatMessage, type ChatSession, ChatSidebar, ChatUI, type ChatUIComponents, type ChatUIProps, EmptyState, type EmptyStateProps, type HeaderProps, Icon, type IconName, type IconProps, IconSvg, type InputProps, LinkChip, type LinkChipProps, MarkdownRenderer, type MarkdownRendererProps, type MemoryItem, MemoryPanel, type MemoryPanelProps, MessageBubble, type MessageBubbleProps, MessageList, type MessageListProps, type ModelConfig, type ModelSelectorProps, type PersonalizationConfig, type PromptTemplate, type ProviderType, type ResponseStyle, type SendMessageParams, type SendMessageResponse, SettingsModal, type SettingsModalProps, type SettingsTab, type SidebarProps, type SourceItem, type ThemeConfig, type ThemeMode, type UseChatUIOptions, type UseChatUIReturn, type UserProfile, useChatUI };

package/dist/react/index.d.ts

@@ -57,6 +57,14 @@ interface AlternativeResponse {
     model: string;
     content: string;
     timestamp: number;
+    sources?: SourceItem[];
+}
+interface SourceItem {
+    id: string;
+    title: string;
+    url: string;
+    snippet?: string;
+    favicon?: string;
 }
 interface ChatMessage {
     id: string;
@@ -65,6 +73,7 @@ interface ChatMessage {
     model?: string;
     timestamp: number;
     alternatives?: AlternativeResponse[];
+    sources?: SourceItem[];
 }
 interface ChatSession {
     id: string;
@@ -139,7 +148,7 @@ interface ChatUIProps {
     /** CSS 클래스 */
     className?: string;
     /** 메시지 전송 핸들러 (커스텀 API 사용 시) */
-    onSendMessage?: (params: SendMessageParams) => Promise<ReadableStream<Uint8Array> | string>;
+    onSendMessage?: (params: SendMessageParams) => Promise<SendMessageResponse>;
     /** 세션 변경 핸들러 */
     onSessionChange?: (session: ChatSession | null) => void;
     /** 에러 핸들러 */
@@ -155,6 +164,11 @@ interface SendMessageParams {
     apiKey?: string;
     systemPrompt?: string;
 }
+/** onSendMessage 반환 타입 */
+type SendMessageResponse = ReadableStream<Uint8Array> | string | {
+    content: string;
+    sources?: SourceItem[];
+};
 interface ChatUIComponents {
     /** 사이드바 컴포넌트 */
     Sidebar?: React.ComponentType<SidebarProps>;
@@ -322,7 +336,7 @@ interface UseChatUIOptions {
     /** 압축 후 유지할 메시지 수 */
     keepRecentMessages?: number;
     /** 커스텀 메시지 전송 핸들러 */
-    onSendMessage?: (params: SendMessageParams) => Promise<ReadableStream<Uint8Array> | string>;
+    onSendMessage?: (params: SendMessageParams) => Promise<SendMessageResponse>;
     /** 세션 변경 핸들러 */
     onSessionChange?: (session: ChatSession | null) => void;
     /** 에러 핸들러 */
@@ -493,4 +507,4 @@ interface SettingsModalProps {
 }
 declare const SettingsModal: React$1.FC<SettingsModalProps>;
 
-export { type ActionItem, type ActionMenuProps, type AlternativeResponse, ChatHeader, ChatInput, type ChatMessage, type ChatSession, ChatSidebar, ChatUI, type ChatUIComponents, type ChatUIProps, EmptyState, type EmptyStateProps, type HeaderProps, Icon, type IconName, type IconProps, IconSvg, type InputProps, LinkChip, type LinkChipProps, MarkdownRenderer, type MarkdownRendererProps, type MemoryItem, MemoryPanel, type MemoryPanelProps, MessageBubble, type MessageBubbleProps, MessageList, type MessageListProps, type ModelConfig, type ModelSelectorProps, type PersonalizationConfig, type PromptTemplate, type ProviderType, type ResponseStyle, type SendMessageParams, SettingsModal, type SettingsModalProps, type SettingsTab, type SidebarProps, type ThemeConfig, type ThemeMode, type UseChatUIOptions, type UseChatUIReturn, type UserProfile, useChatUI };
+export { type ActionItem, type ActionMenuProps, type AlternativeResponse, ChatHeader, ChatInput, type ChatMessage, type ChatSession, ChatSidebar, ChatUI, type ChatUIComponents, type ChatUIProps, EmptyState, type EmptyStateProps, type HeaderProps, Icon, type IconName, type IconProps, IconSvg, type InputProps, LinkChip, type LinkChipProps, MarkdownRenderer, type MarkdownRendererProps, type MemoryItem, MemoryPanel, type MemoryPanelProps, MessageBubble, type MessageBubbleProps, MessageList, type MessageListProps, type ModelConfig, type ModelSelectorProps, type PersonalizationConfig, type PromptTemplate, type ProviderType, type ResponseStyle, type SendMessageParams, type SendMessageResponse, SettingsModal, type SettingsModalProps, type SettingsTab, type SidebarProps, type SourceItem, type ThemeConfig, type ThemeMode, type UseChatUIOptions, type UseChatUIReturn, type UserProfile, useChatUI };

package/dist/react/index.js

@@ -408,6 +408,22 @@ ${contextSummary}` },
           );
           return;
         }
+        if (typeof result === "object" && "content" in result) {
+          setSessions(
+            (prev) => prev.map((s) => {
+              if (s.id === capturedSessionId) {
+                return {
+                  ...s,
+                  messages: s.messages.map(
+                    (m) => m.id === assistantMessageId ? { ...m, content: result.content, sources: result.sources } : m
+                  )
+                };
+              }
+              return s;
+            })
+          );
+          return;
+        }
         response = new Response(result);
       } else {
         response = await fetch(apiEndpoint, {
@@ -567,6 +583,7 @@ ${currentSession.contextSummary}` },
       const modelConfig = models.find((m) => m.id === targetModel);
       const provider = modelConfig?.provider || "ollama";
       let responseContent = "";
+      let responseSources;
       if (onSendMessage) {
         const result = await onSendMessage({
           messages: messagesForApi,
@@ -577,6 +594,9 @@ ${currentSession.contextSummary}` },
         });
         if (typeof result === "string") {
           responseContent = result;
+        } else if (typeof result === "object" && "content" in result) {
+          responseContent = result.content;
+          responseSources = result.sources;
         } else {
           const reader = result.getReader();
           const decoder = new TextDecoder();
@@ -640,7 +660,8 @@ ${currentSession.contextSummary}` },
         id: generateId("alt"),
         model: targetModel,
         content: responseContent,
-        timestamp: Date.now()
+        timestamp: Date.now(),
+        sources: responseSources
       };
       const capturedSessionId = currentSessionId;
       setSessions(
@@ -2195,6 +2216,7 @@ var MessageBubble = ({
   const otherModels = models?.filter((m) => m.id !== message.model) || [];
   const displayContent = alternatives && alternatives.length > 0 && activeAlternativeIndex > 0 ? alternatives[activeAlternativeIndex - 1]?.content || message.content : message.content;
   const displayModel = alternatives && alternatives.length > 0 && activeAlternativeIndex > 0 ? alternatives[activeAlternativeIndex - 1]?.model : message.model;
+  const displaySources = alternatives && alternatives.length > 0 && activeAlternativeIndex > 0 ? alternatives[activeAlternativeIndex - 1]?.sources : message.sources;
   const handleMouseUp = () => {
     if (!onQuote) return;
     const selection = window.getSelection();
@@ -2319,6 +2341,43 @@ var MessageBubble = ({
               ]
             }
           ),
+          displaySources && displaySources.length > 0 && /* @__PURE__ */ (0, import_jsx_runtime7.jsxs)(
+            "div",
+            {
+              style: {
+                display: "flex",
+                flexWrap: "wrap",
+                gap: "8px",
+                marginTop: "12px",
+                paddingTop: "12px",
+                borderTop: "1px solid var(--chatllm-border-light, #f3f4f6)"
+              },
+              children: [
+                /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(
+                  "span",
+                  {
+                    style: {
+                      fontSize: "12px",
+                      fontWeight: 500,
+                      color: "var(--chatllm-text-muted, #9ca3af)",
+                      marginRight: "4px"
+                    },
+                    children: "\uCD9C\uCC98:"
+                  }
+                ),
+                displaySources.map((source, index) => /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(
+                  LinkChip,
+                  {
+                    text: source.title,
+                    url: source.url,
+                    index: index + 1,
+                    showFavicon: true
+                  },
+                  source.id
+                ))
+              ]
+            }
+          ),
           alternatives && alternatives.length > 0 && /* @__PURE__ */ (0, import_jsx_runtime7.jsxs)(
             "div",
             {