@gendive/chatllm 0.4.0 → 0.5.0

package/dist/react/index.mjs

@@ -359,6 +359,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, {
@@ -518,6 +534,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,
@@ -528,6 +545,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();
@@ -591,7 +611,8 @@ ${currentSession.contextSummary}` },
         id: generateId("alt"),
         model: targetModel,
         content: responseContent,
-        timestamp: Date.now()
+        timestamp: Date.now(),
+        sources: responseSources
       };
       const capturedSessionId = currentSessionId;
       setSessions(
@@ -2146,6 +2167,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();
@@ -2270,6 +2292,43 @@ var MessageBubble = ({
               ]
             }
           ),
+          displaySources && displaySources.length > 0 && /* @__PURE__ */ jsxs6(
+            "div",
+            {
+              style: {
+                display: "flex",
+                flexWrap: "wrap",
+                gap: "8px",
+                marginTop: "12px",
+                paddingTop: "12px",
+                borderTop: "1px solid var(--chatllm-border-light, #f3f4f6)"
+              },
+              children: [
+                /* @__PURE__ */ jsx7(
+                  "span",
+                  {
+                    style: {
+                      fontSize: "12px",
+                      fontWeight: 500,
+                      color: "var(--chatllm-text-muted, #9ca3af)",
+                      marginRight: "4px"
+                    },
+                    children: "\uCD9C\uCC98:"
+                  }
+                ),
+                displaySources.map((source, index) => /* @__PURE__ */ jsx7(
+                  LinkChip,
+                  {
+                    text: source.title,
+                    url: source.url,
+                    index: index + 1,
+                    showFavicon: true
+                  },
+                  source.id
+                ))
+              ]
+            }
+          ),
           alternatives && alternatives.length > 0 && /* @__PURE__ */ jsxs6(
             "div",
             {