--wip-- [skip ci]

kousun12 · kousun12 · commit 7b45ad5e3587 · 2024-08-01T15:48:23.000-04:00
diff --git a/examples/knowledge-graph/generate.ts b/examples/knowledge-graph/generate.ts
@@ -7,26 +7,32 @@ const sampleBook = "War and Peace by Tolstoy";
 const book = process.argv[2] || sampleBook;
 const substrate = new Substrate({ apiKey: process.env["SUBSTRATE_API_KEY"] });
 
+const opts = { cache_age: 60 * 60 * 24 * 7 };
 async function main() {
-  const initialList = new ComputeText({
-    prompt: `List all the main characters in "${book}", then for all the meaningful relationships between them, provide a short label for the relationship`,
-    model: "Llama3Instruct405B",
-  });
-  const graph = new ComputeJSON({
-    prompt: sb.interpolate`Make a JSON graph composed of the characters in ${book} that illustrates the relationships between them. 
+  const initialList = new ComputeText(
+    {
+      prompt: `List all the main characters in "${book}", then for all the meaningful relationships between them, provide a short label for the relationship`,
+      model: "Llama3Instruct405B",
+    },
+    opts,
+  );
+  const graph = new ComputeJSON(
+    {
+      prompt: sb.interpolate`Make a JSON graph composed of the characters in ${book} that illustrates the relationships between them. 
 Use this context:
 ${initialList.future.text}`,
-    json_schema: jsonSchema,
-    temperature: 0.2,
-    model: "Llama3Instruct8B",
-  });
+      json_schema: jsonSchema,
+      temperature: 0.2,
+      model: "Llama3Instruct8B",
+    },
+    opts,
+  );
   const res = await substrate.run(graph);
   const jsonOut = res.get(graph).json_object;
   const htmlTemplate = fs.readFileSync(`${currentDir}/index.html`, "utf8");
-  const html = htmlTemplate.replace(
-    '"{{ graphData }}"',
-    JSON.stringify(jsonOut, null, 2),
-  );
+  const html = htmlTemplate
+    .replace('"{{ graphData }}"', JSON.stringify(jsonOut, null, 2))
+    .replaceAll("{{ title }}", book);
   fs.writeFileSync("knowledge-graph.html", html);
 }
 main();
diff --git a/examples/knowledge-graph/index.html b/examples/knowledge-graph/index.html
@@ -3,7 +3,7 @@
   <head>
     <meta charset="UTF-8" />
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
-    <title>Character Relationship Graph</title>
+    <title>{{ title }} - Relationships</title>
     <script src="https://d3js.org/d3.v7.min.js"></script>
     <style>
       body {
@@ -41,11 +41,22 @@
         fill: #555;
         pointer-events: none;
       }
+      #book-title {
+        font-size: 20px;
+        font-weight: bold;
+        margin-bottom: 20px;
+        text-transform: uppercase;
+        position: absolute;
+        top: 18px;
+        color: darkgray;
+        text-align: center;
+        font-family: Menlo, Monaco, Consolas, "Courier New", monospace;
+      }
     </style>
   </head>
   <body>
+    <div id="book-title">{{ title }}</div>
     <div id="graph-container"></div>
-
     <script>
       const graphData = "{{ graphData }}";
 
