Decouple extraction and save

RubricLab · Oct 10, 2024 · dce59e1 · dce59e1
1 parent da7e56e
commit dce59e1
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 5 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,3 +1,4 @@
+- [2024-10-10] [Decouple extraction and save](https://github.com/RubricLab/memory/commit/d5fd7c08eff9458b9775ef439b480cd38922759d)
 - [2024-10-09] [Add multi-turn examples](https://github.com/RubricLab/memory/commit/a0e32260e510a3ef1312030fcb5e2685ac2bb300)
 - [2024-10-09] [Scaffold multi-turn evals](https://github.com/RubricLab/memory/commit/ecb5531acef6a924b81684660150eeb71d93e704)
 - [2024-10-09] [Add TSConfig](https://github.com/RubricLab/memory/commit/ed521824cc492e46adff6d38a994e18cc08166b2)

diff --git a/env.ts b/env.ts
@@ -3,9 +3,11 @@ import z from 'zod'
 
 export default createEnv({
 	runtimeEnv: {
-		OPENAI_API_KEY: process.env.OPENAI_API_KEY
+		OPENAI_API_KEY: process.env.OPENAI_API_KEY,
+		DATABASE_URL: process.env.DATABASE_URL
 	},
 	server: {
-		OPENAI_API_KEY: z.string().min(1)
+		OPENAI_API_KEY: z.string().min(1),
+		DATABASE_URL: z.string().min(1)
 	}
 })
diff --git a/package.json b/package.json
@@ -1,7 +1,7 @@
 {
   "name": "@rubriclab/memory",
   "module": "src/index.ts",
-  "version": "0.0.11",
+  "version": "0.0.12",
   "private": false,
   "type": "module",
   "devDependencies": {

diff --git a/src/index.ts b/src/index.ts
@@ -82,6 +82,12 @@ export class Memory {
             "${content}"`
 		})
 
+		return { facts }
+	}
+
+	async insert({ content }: { content: string }) {
+		const { facts } = await this.extract({ content })
+
 		for await (const fact of facts) {
 			const { subject, relation, object } = fact
 
@@ -96,7 +102,5 @@ export class Memory {
 
 		const priorFacts = this.db.query('select * from facts').all()
 		console.log({ priorFacts })
-
-		return { facts }
 	}
 }