langgraph-api 0.0.1__py3-none-any.whl

langgraph_api/js/src/schema/types.template.mts

@@ -0,0 +1,74 @@
+import type { BaseMessage } from "@langchain/core/messages";
+import type {
+  StateType,
+  UpdateType,
+  StateDefinition,
+} from "@langchain/langgraph";
+import type { Graph } from "@langchain/langgraph";
+import type { Pregel } from "@langchain/langgraph/pregel";
+
+type AnyPregel = Pregel<any, any>;
+type AnyGraph = Graph<any, any, any, any, any>;
+
+type Wrap<T> = (a: T) => void;
+type MatchBaseMessage<T> = T extends BaseMessage ? BaseMessage : never;
+type MatchBaseMessageArray<T> =
+  T extends Array<infer C>
+    ? Wrap<MatchBaseMessage<C>> extends Wrap<BaseMessage>
+      ? BaseMessage[]
+      : never
+    : never;
+
+type Defactorify<T> = T extends (...args: any[]) => infer R
+  ? Awaited<R>
+  : Awaited<T>;
+
+type Inspect<T> = T extends unknown
+  ? {
+      [K in keyof T]: 0 extends 1 & T[K]
+        ? T[K]
+        : Wrap<MatchBaseMessageArray<T[K]>> extends Wrap<BaseMessage[]>
+          ? BaseMessage[]
+          : Wrap<MatchBaseMessage<T[K]>> extends Wrap<BaseMessage>
+            ? BaseMessage
+            : Inspect<T[K]>;
+    }
+  : never;
+
+type ReflectCompiled<T> = T extends { RunInput: infer S; RunOutput: infer U }
+  ? { state: S; update: U }
+  : never;
+
+type Reflect<T> =
+  Defactorify<T> extends infer DT
+    ? DT extends {
+        compile(...args: any[]): infer Compiled;
+      }
+      ? ReflectCompiled<Compiled>
+      : ReflectCompiled<DT>
+    : never;
+
+type BuilderReflectCompiled<T> = T extends {
+  builder: {
+    _inputDefinition: infer I extends StateDefinition;
+    _outputDefinition: infer O extends StateDefinition;
+    _configSchema?: infer C extends StateDefinition | undefined;
+  };
+}
+  ? {
+      input: UpdateType<I>;
+      output: StateType<O>;
+      config: UpdateType<Exclude<C, undefined>>;
+    }
+  : never;
+
+type BuilderReflect<T> =
+  Defactorify<T> extends infer DT
+    ? DT extends {
+        compile(...args: any[]): infer Compiled;
+      }
+      ? BuilderReflectCompiled<Compiled>
+      : BuilderReflectCompiled<DT>
+    : never;
+
+type FilterAny<T> = 0 extends 1 & T ? never : T;

langgraph_api/js/src/utils/importMap.mts

@@ -0,0 +1,85 @@
+import {
+  PromptTemplate,
+  AIMessagePromptTemplate,
+  ChatMessagePromptTemplate,
+  ChatPromptTemplate,
+  HumanMessagePromptTemplate,
+  MessagesPlaceholder,
+  SystemMessagePromptTemplate,
+  ImagePromptTemplate,
+  PipelinePromptTemplate,
+} from "@langchain/core/prompts";
+import {
+  AIMessage,
+  AIMessageChunk,
+  BaseMessage,
+  BaseMessageChunk,
+  ChatMessage,
+  ChatMessageChunk,
+  FunctionMessage,
+  FunctionMessageChunk,
+  HumanMessage,
+  HumanMessageChunk,
+  SystemMessage,
+  SystemMessageChunk,
+  ToolMessage,
+  ToolMessageChunk,
+} from "@langchain/core/messages";
+import { StringPromptValue } from "@langchain/core/prompt_values";
+
+export const prompts__prompt = {
+  PromptTemplate,
+};
+
+export const schema__messages = {
+  AIMessage,
+  AIMessageChunk,
+  BaseMessage,
+  BaseMessageChunk,
+  ChatMessage,
+  ChatMessageChunk,
+  FunctionMessage,
+  FunctionMessageChunk,
+  HumanMessage,
+  HumanMessageChunk,
+  SystemMessage,
+  SystemMessageChunk,
+  ToolMessage,
+  ToolMessageChunk,
+};
+
+export const schema = {
+  AIMessage,
+  AIMessageChunk,
+  BaseMessage,
+  BaseMessageChunk,
+  ChatMessage,
+  ChatMessageChunk,
+  FunctionMessage,
+  FunctionMessageChunk,
+  HumanMessage,
+  HumanMessageChunk,
+  SystemMessage,
+  SystemMessageChunk,
+  ToolMessage,
+  ToolMessageChunk,
+};
+
+export const prompts__chat = {
+  AIMessagePromptTemplate,
+  ChatMessagePromptTemplate,
+  ChatPromptTemplate,
+  HumanMessagePromptTemplate,
+  MessagesPlaceholder,
+  SystemMessagePromptTemplate,
+};
+export const prompts__image = {
+  ImagePromptTemplate,
+};
+export const prompts__pipeline = {
+  PipelinePromptTemplate,
+};
+
+export const prompts__base = {
+  StringPromptValue,
+};

langgraph_api/js/src/utils/pythonSchemas.mts

@@ -0,0 +1,28 @@
+/**
+ * Represents a stored item with metadata.
+ */
+export interface PyItem {
+  /**
+   * The stored data as an object. Keys are filterable.
+   */
+  value: Record<string, any>;
+  /**
+   * Unique identifier within the namespace.
+   */
+  key: string;
+  /**
+   * Hierarchical path defining the collection in which this document resides.
+   * Represented as an array of strings, allowing for nested categorization.
+   * For example: ["documents", "user123"]
+   */
+  namespace: string[];
+  /**
+   * Timestamp of item creation.
+   */
+  created_at: Date;
+  /**
+   * Timestamp of last update.
+   */
+  updated_at: Date;
+}
+export type PyResult = PyItem | PyItem[] | string[][] | null | undefined;

langgraph_api/js/src/utils/serde.mts

@@ -0,0 +1,21 @@
+export const serialiseAsDict = (obj: unknown) => {
+  return JSON.stringify(
+    obj,
+    function (key: string | number, value: unknown) {
+      const rawValue = this[key];
+      if (
+        rawValue != null &&
+        typeof rawValue === "object" &&
+        "toDict" in rawValue &&
+        typeof rawValue.toDict === "function"
+      ) {
+        // TODO: we need to upstream this to LangChainJS
+        const { type, data } = rawValue.toDict();
+        return { ...data, type };
+      }
+
+      return value;
+    },
+    2
+  );
+};