langgraph-api 0.0.37__py3-none-any.whl → 0.0.39__py3-none-any.whl

langgraph_api/js/src/graph.mts

@@ -26,13 +26,17 @@ export function filterValidGraphSpecs(specs: Record<string, string>) {
   );
 }
 
+export type CompiledGraphFactory<T extends string> = (config: {
+  configurable?: Record<string, unknown>;
+}) => Promise<CompiledGraph<T>>;
+
 export async function resolveGraph(
   spec: string,
   options?: { onlyFilePresence?: false }
 ): Promise<{
   sourceFile: string;
   exportSymbol: string;
-  resolved: CompiledGraph<string>;
+  resolved: CompiledGraph<string> | CompiledGraphFactory<string>;
 }>;
 
 export async function resolveGraph(
@@ -58,7 +62,9 @@ export async function resolveGraph(
   type GraphUnknown =
     | GraphLike
     | Promise<GraphLike>
-    | (() => GraphLike | Promise<GraphLike>)
+    | ((config: {
+        configurable?: Record<string, unknown>;
+      }) => GraphLike | Promise<GraphLike>)
     | undefined;
 
   const isGraph = (graph: GraphLike): graph is Graph<string> => {
@@ -71,13 +77,24 @@ export async function resolveGraph(
   );
 
   // obtain the graph, and if not compiled, compile it
-  const resolved: CompiledGraph<string> = await (async () => {
-    if (!graph) throw new Error("Failed to load graph: graph is nullush");
-    const graphLike = typeof graph === "function" ? await graph() : await graph;
-
-    if (isGraph(graphLike)) return graphLike.compile();
-    return graphLike;
-  })();
+  const resolved: CompiledGraph<string> | CompiledGraphFactory<string> =
+    await (async () => {
+      if (!graph) throw new Error("Failed to load graph: graph is nullush");
+
+      const afterResolve = (graphLike: GraphLike): CompiledGraph<string> => {
+        const graph = isGraph(graphLike) ? graphLike.compile() : graphLike;
+        return graph;
+      };
+
+      if (typeof graph === "function") {
+        return async (config: { configurable?: Record<string, unknown> }) => {
+          const graphLike = await graph(config);
+          return afterResolve(graphLike);
+        };
+      }
+
+      return afterResolve(await graph);
+    })();
 
   return { sourceFile, exportSymbol, resolved };
 }