aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/client/views/nodes/chatbot/chatboxcomponents/ChatBox.tsx13
-rw-r--r--src/client/views/nodes/chatbot/utils/AgentDocumentManager.ts20
2 files changed, 15 insertions, 18 deletions
diff --git a/src/client/views/nodes/chatbot/chatboxcomponents/ChatBox.tsx b/src/client/views/nodes/chatbot/chatboxcomponents/ChatBox.tsx
index 867e78860..00077d68d 100644
--- a/src/client/views/nodes/chatbot/chatboxcomponents/ChatBox.tsx
+++ b/src/client/views/nodes/chatbot/chatboxcomponents/ChatBox.tsx
@@ -970,14 +970,6 @@ export class ChatBox extends ViewBoxAnnotatableComponent<FieldViewProps>() {
componentWillUnmount() {
this.removeScrollListener();
}
- /**
- * Getter that retrieves document IDs of linked documents that have PDF_chunker–parsed content.
- */
- @computed
- get docIds(): string[] {
- // Use the document manager to get all document IDs
- return Array.from(this.docManager.listDocs);
- }
/**
* Getter that retrieves all linked CSV files for analysis.
@@ -1004,11 +996,6 @@ export class ChatBox extends ViewBoxAnnotatableComponent<FieldViewProps>() {
// Other helper methods for retrieving document data and processing
- retrieveSummaries = (): string => {
- console.log(this.docManager.listDocs);
- return JSON.stringify(this.docManager.listDocs);
- };
-
retrieveCSVData = () => {
return this.linkedCSVs;
};
diff --git a/src/client/views/nodes/chatbot/utils/AgentDocumentManager.ts b/src/client/views/nodes/chatbot/utils/AgentDocumentManager.ts
index 82b7ed3df..e9d41efbd 100644
--- a/src/client/views/nodes/chatbot/utils/AgentDocumentManager.ts
+++ b/src/client/views/nodes/chatbot/utils/AgentDocumentManager.ts
@@ -965,11 +965,21 @@ export class AgentDocumentManager {
* @returns An array of document IDs (strings).
*/
@computed
- public get listDocs(): string[] {
- console.log(
- Array.from(this.documentsById.entries()).map(([id, agentDoc]) => JSON.stringify({ id, title: agentDoc.layoutDoc.title, type: agentDoc.layoutDoc.type, summary: agentDoc.layoutDoc.summary || 'No summary available for this document.' }))
- );
- return Array.from(this.documentsById.entries()).map(([id, agentDoc]) => JSON.stringify({ id, title: agentDoc.layoutDoc.title, type: agentDoc.layoutDoc.type, summary: agentDoc.layoutDoc.summary || 'No summary available for this document.' }));
+ public get listDocs(): string {
+ const xmlDocs = Array.from(this.documentsById.entries()).map(([id, agentDoc]) => {
+ return `<document>
+ <id>${id}</id>
+ <title>${this.escapeXml(StrCast(agentDoc.layoutDoc.title))}</title>
+ <type>${this.escapeXml(StrCast(agentDoc.layoutDoc.type))}</type>
+ <summary>${this.escapeXml(StrCast(agentDoc.layoutDoc.summary))}</summary>
+</document>`;
+ });
+
+ return xmlDocs.join('\n');
+ }
+
+ private escapeXml(str: string): string {
+ return str.replace(/&/g, '&amp;').replace(/</g, '&lt;').replace(/>/g, '&gt;').replace(/"/g, '&quot;').replace(/'/g, '&apos;');
}
@computed