fix: make OpenAI optional

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

Author: dselman

README.md

@@ -65,8 +65,9 @@ Runtime result:
 
 ### GraphDB
 
-- NEO4J_URL: <optional> the NEO4J URL. If not set, no data will be written to the agreement graph.
-- NEO4J_PASS: <optional> the neo4j password.
+- NEO4J_URL: the NEO4J URL. E.g. `neo4j+s://<DB_NAME>.databases.neo4j.io` if you are using AuraDB.
+- NEO4J_PASS: your neo4j password.
+- NEO4J_USER: <optional> defaults to `neo4j`
 
 ### Text Embeddings
 - OPENAI_API_KEY: <optional> the OpenAI API key. If not set embeddings are not computed and written to the agreement graph and similarity search is not possible.

src/demo/index.ts

@@ -55,7 +55,7 @@ async function run() {
     NEO4J_URL: process.env.NEO4J_URL,
     logger: console,
     logQueries: false,
-    embeddingFunction: getOpenAiEmbedding
+    embeddingFunction: process.env.OPENAI_API_KEY ? getOpenAiEmbedding : undefined
   }
   const graphModel = new GraphModel(MODEL, options);  
   await graphModel.connect();
@@ -100,9 +100,12 @@ async function run() {
     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);
+  if(process.env.OPENAI_API_KEY) {
+    const search = 'Working in a boring job and looking for love.';
+    console.log(`Searching for movies related to: '${search}'`);
+    const results = await graphModel.similarityQuery(`${NS}.Movie`, 'embedding', search, 3);
+    console.log(results);  
+  }
   await graphModel.closeSession(context);
   console.log('done');
 }