clean

Signed-off-by: Dan Selman <[email protected]>

Author: dselman

src/demo/index.ts

@@ -1,7 +1,9 @@
 import { GraphModel, GraphModelOptions, getOpenAiEmbedding } from '../graphmodel.js'
 
+const NS = '[email protected]';
+
 const MODEL = `
-namespace [email protected]
+namespace ${NS}
 import [email protected].{GraphNode}
 
 // show how a complex type gets flattened
@@ -14,10 +16,6 @@ concept Address {
   o String country
 }
 
-concept PersonProperties extends GraphNode {
-  o Address address
-}
-
 concept Person extends GraphNode {
   o Address address optional
   @label("ACTED_IN")
@@ -50,7 +48,7 @@ concept Movie extends GraphNode {
 }
 `;
 
-async function run(): Promise<GraphModel> {
+async function run() {
   const options:GraphModelOptions = {
     NEO4J_USER: process.env.NEO4J_USER,
     NEO4J_PASS: process.env.NEO4J_PASS,
@@ -75,21 +73,39 @@ async function run(): Promise<GraphModel> {
       state: 'CO',
       country: 'USA'
     };
-    const summary = 'The film centres on Sam Lowry, a low-ranking bureaucrat trying to find a woman who appears in his dreams while he is working in a mind-numbing job and living in a small apartment, set in a dystopian world in which there is an over-reliance on poorly maintained (and rather whimsical) machines';
-    await graphModel.mergeNode(transaction, '[email protected]', {identifier: 'Brazil', summary} );
-    await graphModel.mergeNode(transaction, '[email protected]', {identifier: 'Terry Gilliam'} );
-    await graphModel.mergeRelationship(transaction, '[email protected]', 'Terry Gilliam', '[email protected]', 'Brazil', 'directed' );
-    await graphModel.mergeNode(transaction, '[email protected]', {identifier: 'Dan', address} );
-    await graphModel.mergeRelationship(transaction, '[email protected]', 'Dan', '[email protected]', 'Brazil', 'ratedMovies' );
-    await graphModel.mergeNode(transaction, '[email protected]', {identifier: 'Jonathan Pryce'} );
-    await graphModel.mergeRelationship(transaction, '[email protected]', 'Jonathan Pryce', '[email protected]', 'Brazil', 'actedIn' );
-    const search = 'working in a mind-numbing job and living in a small apartment';
-    const results = await graphModel.similarityQuery('[email protected]', 'embedding', search, 3);
+    await graphModel.mergeNode(transaction, `${NS}.Movie`, {identifier: 'Brazil', summary: 'The film centres on Sam Lowry, a low-ranking bureaucrat trying to find a woman who appears in his dreams while he is working in a mind-numbing job and living in a small apartment, set in a dystopian world in which there is an over-reliance on poorly maintained (and rather whimsical) machines'} );
+    await graphModel.mergeNode(transaction, `${NS}.Movie`, {identifier: 'The Man Who Killed Don Quixote', summary: 'Instead of a literal adaptation, Gilliam\'s film was about "an old, retired, and slightly kooky nobleman named Alonso Quixano".'} );
+    await graphModel.mergeNode(transaction, `${NS}.Movie`, {identifier: 'Fear and Loathing in Las Vegas', summary: 'Duke, under the influence of mescaline, complains of a swarm of giant bats, and inventories their drug stash. They pick up a young hitchhiker and explain their mission: Duke has been assigned by a magazine to cover the Mint 400 motorcycle race in Las Vegas. They bought excessive drugs for the trip, and rented a red Chevrolet Impala convertible.'} );
+
+    await graphModel.mergeNode(transaction, `${NS}.Genre`, {identifier: 'Comedy'} );
+    await graphModel.mergeNode(transaction, `${NS}.Genre`, {identifier: 'Science Fiction'} );
+
+    await graphModel.mergeRelationship(transaction, `${NS}.Movie`, 'Brazil', `${NS}.Genre`, 'Comedy', 'genres' );
+    await graphModel.mergeRelationship(transaction, `${NS}.Movie`, 'Brazil', `${NS}.Genre`, 'Science Fiction', 'genres' );
+    await graphModel.mergeRelationship(transaction, `${NS}.Movie`, 'The Man Who Killed Don Quixote', `${NS}.Genre`, 'Comedy', 'genres' );
+    await graphModel.mergeRelationship(transaction, `${NS}.Movie`, 'Fear and Loathing in Las Vegas', `${NS}.Genre`, 'Comedy', 'genres' );
+
+    await graphModel.mergeNode(transaction, `${NS}.Director`, {identifier: 'Terry Gilliam'} );
+    await graphModel.mergeRelationship(transaction, `${NS}.Director`, 'Terry Gilliam', `${NS}.Movie`, 'Brazil', 'directed' );
+    await graphModel.mergeRelationship(transaction, `${NS}.Director`, 'Terry Gilliam', `${NS}.Movie`, 'The Man Who Killed Don Quixote', 'directed' );
+    await graphModel.mergeRelationship(transaction, `${NS}.Director`, 'Terry Gilliam', `${NS}.Movie`, 'Fear and Loathing in Las Vegas', 'directed' );
+
+    await graphModel.mergeNode(transaction, `${NS}.User`, {identifier: 'Dan', address} );
+    await graphModel.mergeRelationship(transaction, `${NS}.User`, 'Dan', `${NS}.Movie`, 'Brazil', 'ratedMovies' );
+    
+    await graphModel.mergeNode(transaction, `${NS}.Actor`, {identifier: 'Jonathan Pryce'} );
+    await graphModel.mergeRelationship(transaction, `${NS}.Actor`, 'Jonathan Pryce', `${NS}.Movie`, 'Brazil', 'actedIn' );
+    await graphModel.mergeRelationship(transaction, `${NS}.Actor`, 'Jonathan Pryce', `${NS}.Movie`, 'The Man Who Killed Don Quixote', 'actedIn' );
+
+    await graphModel.mergeNode(transaction, `${NS}.Actor`, {identifier: 'Johnny Depp'} );
+    await graphModel.mergeRelationship(transaction, `${NS}.Actor`, 'Johnny Depp', `${NS}.Movie`, 'Fear and Loathing in Las Vegas', 'actedIn' );
+
+    const search = 'Working in a boring job and looking for love.';
+    const results = await graphModel.similarityQuery(`${NS}.Movie`, 'embedding', search, 3);
     console.log(results);
   });
-  await session.close();
+  await graphModel.closeSession(context);
   console.log('done');
-  return graphModel;
 }
 
 run();