Merge branch 'main' into brace/run-ci-prs

langchain-ai · Jan 17, 2024 · c14eb6b · c14eb6b
2 parents 92682c4 + 9e814f2
commit c14eb6b
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 29 deletions.
diff --git a/langgraph/src/graph/index.ts b/langgraph/src/graph/index.ts
@@ -149,7 +149,7 @@ export class Graph<
 
     const nodes: Record<
       string,
-      ChannelInvoke<RunInput, RunOutput> | ChannelBatch<RunInput, RunOutput>
+      ChannelInvoke<RunInput, RunOutput> | ChannelBatch
     > = {};
     for (const [key, node] of Object.entries(this.nodes)) {
       nodes[key] = Channel.subscribeTo(`${key}:inbox`)

diff --git a/langgraph/src/pregel/read.ts b/langgraph/src/pregel/read.ts
@@ -74,11 +74,15 @@ interface ChannelInvokeArgs<RunInput, RunOutput>
   when?: (args: any) => boolean;
 }
 
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+export type ChannelInvokeInputType = any;
+
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+export type ChannelInvokeOutputType = any;
+
 export class ChannelInvoke<
-  // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  RunInput = any,
-  // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  RunOutput = any
+  RunInput = ChannelInvokeInputType,
+  RunOutput = ChannelInvokeOutputType
 > extends RunnableBinding<RunInput, RunOutput, RunnableConfig> {
   lc_graph_name = "ChannelInvoke";
 
@@ -147,30 +151,33 @@ export class ChannelInvoke<
   }
 }
 
-interface ChannelBatchArgs<RunInput, RunOutput> {
+interface ChannelBatchArgs {
   channel: string;
   key?: string;
-  bound?: Runnable<RunInput, RunOutput, RunnableConfig>;
+  bound?: Runnable;
 }
 
-export class ChannelBatch<
-  // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  RunInput = any,
-  // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  RunOutput = any
-> extends RunnableEach<RunInput, RunOutput, RunnableConfig> {
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+export type ChannelBatchInputType = any;
+
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+export type ChannelBatchOutputType = any;
+
+export class ChannelBatch extends RunnableEach<
+  ChannelBatchInputType,
+  ChannelBatchOutputType,
+  RunnableConfig
+> {
   lc_graph_name = "ChannelBatch";
 
   channel: string;
 
   key?: string;
 
-  constructor(fields: ChannelBatchArgs<RunInput, RunOutput>) {
+  constructor(fields: ChannelBatchArgs) {
     super({
       ...fields,
-      bound:
-        fields.bound ??
-        (defaultRunnableBound as unknown as Runnable<RunInput, RunOutput>),
+      bound: fields.bound ?? defaultRunnableBound,
     });
 
     this.channel = fields.channel;
@@ -199,28 +206,22 @@ export class ChannelBatch<
       return new ChannelBatch({
         channel: this.channel,
         key: this.key,
-        bound: this.bound.pipe(
-          joiner as unknown as RunnableLike<
-            RunOutput,
-            Record<string, unknown> & { [x: string]: unknown }
-          >
-        ),
+        bound: this.bound.pipe(joiner),
       });
     }
   }
 
-  pipe<NewRunOutput>(
-    coerceable: RunnableLike<RunOutput, NewRunOutput>
-  ): ChannelBatch<RunInput, NewRunOutput> {
+  // @ts-expect-error TODO: fix later
+  pipe(coerceable: RunnableLike): ChannelBatch {
     if (this.bound === defaultRunnableBound) {
-      return new ChannelBatch<RunInput, RunOutput>({
+      return new ChannelBatch({
         channel: this.channel,
         key: this.key,
         bound: _coerceToRunnable(coerceable),
       });
     } else {
       // Delegate to `or` in `this.bound`
-      return new ChannelBatch<RunInput, RunOutput>({
+      return new ChannelBatch({
         channel: this.channel,
         key: this.key,
         bound: this.bound.pipe(coerceable),

diff --git a/langgraph/src/tests/pregel.test.ts b/langgraph/src/tests/pregel.test.ts
@@ -170,7 +170,9 @@ it("should invoke two processes and get correct output", async () => {
     nodes: { one, two },
   });
 
-  await expect(app.invoke(2, { recursionLimit: 1 })).rejects.toThrow(GraphRecursionError);
+  await expect(app.invoke(2, { recursionLimit: 1 })).rejects.toThrow(
+    GraphRecursionError
+  );
 
   expect(await app.invoke(2)).toEqual(4);