23图引擎的中断 / Checkpoint / 续跑

/*
 * Copyright 2024 CloudWeGo Authors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package compose

import (
  "context"
  "errors"
  "fmt"

  "github.com/google/uuid"

  "github.com/cloudwego/eino/internal/core"
  "github.com/cloudwego/eino/schema"
)

// WithInterruptBeforeNodes instructs to interrupt before the given nodes.
func WithInterruptBeforeNodes(nodes []string) GraphCompileOption {
  return func(options *graphCompileOptions) {
    options.interruptBeforeNodes = nodes
  }
}

// WithInterruptAfterNodes instructs to interrupt after the given nodes.
func WithInterruptAfterNodes(nodes []string) GraphCompileOption {
  return func(options *graphCompileOptions) {
    options.interruptAfterNodes = nodes
// … 这只是文件开头 40 行，并非完整声明；点击上方「浏览完整文件」

/*
 * Copyright 2024 CloudWeGo Authors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package compose

import (
  "context"
  "fmt"
  "reflect"

  "github.com/cloudwego/eino/internal/core"
  "github.com/cloudwego/eino/internal/serialization"
  "github.com/cloudwego/eino/schema"
)

func init() {
  schema.RegisterName[*checkpoint]("_eino_checkpoint")
  schema.RegisterName[*dagChannel]("_eino_dag_channel")
  schema.RegisterName[*pregelChannel]("_eino_pregel_channel")
  schema.RegisterName[dependencyState]("_eino_dependency_state")
  _ = serialization.GenericRegister[channel]("_eino_channel")
}

// RegisterSerializableType registers a custom type for eino serialization.
// This allows eino to properly serialize and deserialize custom types.
// Both custom interfaces and structs need to be registered using this function.
// Types only need to be registered once - pointers and other references will be handled automatically.
// … 这只是文件开头 40 行，并非完整声明；点击上方「浏览完整文件」

/*
 * Copyright 2025 CloudWeGo Authors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package compose

import (
  "context"

  "github.com/cloudwego/eino/internal/core"
)

// GetInterruptState provides a type-safe way to check for and retrieve the persisted state from a previous interruption.
// It is the primary function a component should use to understand its past state.
//
// It returns three values:
//   - wasInterrupted (bool): True if the node was part of a previous interruption, regardless of whether state was provided.
//   - state (T): The typed state object, if it was provided and matches type `T`.
//   - hasState (bool): True if state was provided during the original interrupt and successfully cast to type `T`.
func GetInterruptState[T any](ctx context.Context) (wasInterrupted bool, hasState bool, state T) {
  return core.GetInterruptState[T](ctx)
}

// GetResumeContext checks if the current component is the target of a resume operation
// and retrieves any data provided by the user for that resumption.
//
// This function is typically called *after* a component has already determined it is in a
// resumed state by calling GetInterruptState.
// … 这只是文件开头 40 行，并非完整声明；点击上方「浏览完整文件」