diff --git a/ts/src/runtime/core/graph_memory/graph_database.ts b/ts/src/runtime/core/graph_memory/graph_database.ts
index 4bc6d79..fc81c29 100644
--- a/ts/src/runtime/core/graph_memory/graph_database.ts
+++ b/ts/src/runtime/core/graph_memory/graph_database.ts
@@ -2,7 +2,7 @@ import Database from 'better-sqlite3';
 import type {
   Entity, Relation, RecallParams, CommitParams, PurgeParams,
   RecallResult, CommitResult, PurgeResult, MemoryStats
-} from './types';
+} from './types.js';
 
 export class GraphDatabase {
   private db: Database.Database;
diff --git a/ts/src/runtime/core/graph_memory/index.ts b/ts/src/runtime/core/graph_memory/index.ts
index 770ece9..7fd6fd8 100644
--- a/ts/src/runtime/core/graph_memory/index.ts
+++ b/ts/src/runtime/core/graph_memory/index.ts
@@ -1,3 +1,3 @@
-export * from './types';
-export * from './graph_database';
-export { MemoryService } from './memory_service';
+export * from './types.js';
+export * from './graph_database.js';
+export { MemoryService } from './memory_service.js';
diff --git a/ts/src/runtime/core/graph_memory/memory_service.ts b/ts/src/runtime/core/graph_memory/memory_service.ts
index 8d5d26d..bfba4e0 100644
--- a/ts/src/runtime/core/graph_memory/memory_service.ts
+++ b/ts/src/runtime/core/graph_memory/memory_service.ts
@@ -1,5 +1,5 @@
-import { GraphDatabase } from './graph_database';
-import { TaskNodeStore } from './task_node_store';
+import { GraphDatabase } from './graph_database.js';
+import { TaskNodeStore } from './task_node_store.js';
 import * as fs from 'fs';
 import * as path from 'path';
 import type {
@@ -8,7 +8,7 @@ import type {
   ContextRewriteParams, ContextRewriteResult,
   WorkingMemoryChainParams, WorkingMemoryChainResult,
   TaskNodeCreateParams, TaskNodeChainResult
-} from './types';
+} from './types.js';
 
 export class MemoryService {
   private db: GraphDatabase;
diff --git a/ts/src/runtime/core/graph_memory/task_node_store.ts b/ts/src/runtime/core/graph_memory/task_node_store.ts
index 952979f..d51119c 100644
--- a/ts/src/runtime/core/graph_memory/task_node_store.ts
+++ b/ts/src/runtime/core/graph_memory/task_node_store.ts
@@ -1,7 +1,7 @@
 import Database from 'better-sqlite3';
 import * as fs from 'fs';
 import * as path from 'path';
-import type { TaskNodeData } from './types';
+import type { TaskNodeData } from './types.js';
 
 export { TaskNodeData as TaskNode };
 
diff --git a/ts/src/runtime/core/graph_memory/types.ts b/ts/src/runtime/core/graph_memory/types.ts
index ec4ae2c..63de215 100644
--- a/ts/src/runtime/core/graph_memory/types.ts
+++ b/ts/src/runtime/core/graph_memory/types.ts
@@ -7,6 +7,8 @@ export interface Entity {
   updatedAt: Date;
 }
 
+export const __types = true;
+
 export type RelationStatus = 'active' | 'deleted' | 'archived' | 'superseded';
 
 export interface Relation {