fix: logging

Signed-off-by: Dan Selman <danscode@selman.org>
accordproject · Jun 28, 2024 · f5a246b · f5a246b
1 parent 4012ac0
commit f5a246b
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/src/demo/index.ts b/src/demo/index.ts
@@ -88,7 +88,7 @@ async function run() {
   // await graphModel.loadConcertoModels(); // loads models from graph
   const context = await graphModel.openSession();
 
-  // const tools = graphModel.getTools({chatWithData: true, fullTextSearch: true, getById: true, similaritySearch: true});
+  // const tools = graphModel.getTools({currentDateTime: true});
   // console.log(JSON.stringify(tools, null , 2));
 
   const { session } = context;

diff --git a/src/graphmodel.ts b/src/graphmodel.ts
@@ -811,7 +811,7 @@ export class GraphModel {
                                     })
                                     return result;
                                 }) : [];
-                                this.options.logger?.info(`get_${node.getName().toLowerCase()}_by_id`, result);
+                                this.options.logger?.info(`get_${node.getName().toLowerCase()}_by_id`, ret);
                                 return ret;
                             }
                             catch (err) {
@@ -882,9 +882,10 @@ export class GraphModel {
                                 try {
                                     await session.executeWrite(async transaction => {
                                         await this.mergeRelationship(transaction, args.sourceType, args.sourceIdentfiier, args.targetType, args.targetIdentfiier, args.sourcePropertyName);
-                                        this.options.logger?.info(`merge_relationship_${node.getName().toLowerCase()}_${relationship.getName().toLowerCase()}`, result);
                                     });
-                                    return `created relationship ${relationship.getName()} between source node ${args.sourceIdentfiier} and target node ${args.targetIdentfiier}`;
+                                    const ret = `created relationship ${relationship.getName()} between source node ${args.sourceIdentfiier} and target node ${args.targetIdentfiier}`;
+                                    this.options.logger?.info(ret);
+                                    return ret;
                                 }
                                 catch (err) {
                                     return `An error occurred: ${err}`;
@@ -913,7 +914,7 @@ export class GraphModel {
                                 "required": ["sourceType", "sourceIdentfiier", "targetType", "targetIdentfiier", "sourcePropertyName"]
                             }
                         }
-                    })
+                    });
                 }
 
                 const parameters = this.getToolParameters(node);
@@ -929,9 +930,10 @@ export class GraphModel {
                             try {
                                 await session.executeWrite(async transaction => {
                                     await this.mergeNode(transaction, node.getName(), { identifier });
-                                    this.options.logger?.info(`add_${node.getName().toLowerCase()}`, result);
                                 });
-                                return `created node ${node.getName()} with identifier ${identifier}`;
+                                const ret = `created node ${node.getName()} with identifier ${identifier}`;
+                                this.options.logger?.info(ret);
+                                return ret;
                             }
                             catch (err) {
                                 return `An error occurred: ${err}`;