[UXE-0000] - Fix/ai agent

templates/typescript/azion-react-agent/setup/setup.ts

@@ -54,11 +54,11 @@ export default async function azionVectorStoreSetup() {
   const vectorStore = new AzionVectorStore(embeddingModel, {dbName:process.env.VECTOR_STORE_DB_NAME, tableName:process.env.VECTOR_STORE_TABLE_NAME})
 
   // Add documents to the vector store
-  // await vectorStore.addDocuments(productDocs)
+  await vectorStore.addDocuments(productDocs)
 
   // Use the vector store as retriever
-  const docs = await vectorStore.AzionHybridSearch("what is good for headache?",{kfts:1,kvector:1})
-  console.log(JSON.stringify(docs, null, 2))
+  // const docs = await vectorStore.AzionHybridSearch("what is good for headache?",{kfts:1,kvector:1})
+  // console.log(JSON.stringify(docs, null, 2))
 }
 
 azionVectorStoreSetup()

templates/typescript/azion-react-agent/src/helper/tracer.ts

@@ -107,13 +107,24 @@ export class AzionEdgeTracer {
     }
   }
 
+  /**
+   * Remove the single quotes from input, output and run metadata
+   */
+  removeSingleQuotesFromStrings(): void {
+    this.inputMessages = this.inputMessages.map((message:string) => message.replace(/'/g, ""))
+    this.outputMessages = this.outputMessages.replace(/'/g, "")
+    this.runMetadata = this.runMetadata?.map((message:string) => message.replace(/'/g, ""))
+  }
+
   /**
    * Save the trace into the database
    */
   async save() {
     console.log("Saving trace into ", this.databaseName, "for session ", this.sessionId)
     const createdAt = new Date().toISOString()
+    this.removeSingleQuotesFromStrings()
     const statements = [`INSERT INTO ${this.tableName} (session_id, run_id, input_messages, output_messages, run_metadata, created_at) VALUES ('${this.sessionId}', '${this.runId}', '${this.inputMessages}', '${this.outputMessages}', '${this.runMetadata}', '${createdAt}')`]
+
     const { data, error } = await useExecute(this.databaseName,statements)
     if (error) {
       console.error("Error saving trace:", error)

templates/typescript/azion-react-agent/src/langchain-components/AzionRetriever.ts

@@ -348,7 +348,24 @@ export class AzionRetriever extends BaseRetriever {
         entityExtractionPrompt,
         new HumanMessage(query),
       ]);
-      return entityQuery.content.toString().replace(/[^a-zA-Z0-9\s]/g, ' ').split(' ').join(' OR ')
+
+      return this.convert2FTSQuery(entityQuery.content.toString())
+  }
+
+  /**
+   * Converts a query to a FTS query.
+   * @param query The user query
+   * @returns The converted FTS query
+   */
+  protected convert2FTSQuery(
+    query: string
+  ): string {
+    return query
+      .replace(/[^a-záàâãéèêíïóôõöúçñA-ZÁÀÂÃÉÈÊÍÏÓÔÕÖÚÇÑ0-9\s]/g, '') // Remove special chars keeping accents
+      .replace(/\s+/g, ' ') // Remove multiple spaces
+      .trim() // Remove leading/trailing spaces
+      .split(' ')
+      .join(' OR ');
   }
 
   /**