Source code for langchain_experimental.tot.thought

from __future__ import annotations

from enum import Enum
from typing import Set

from langchain_experimental.pydantic_v1 import BaseModel, Field


class ThoughtValidity(Enum):
    """思想有效性的枚举。"""

    VALID_INTERMEDIATE = 0
    VALID_FINAL = 1
    INVALID = 2


class Thought(BaseModel):
    """在ToT中的一个想法。"""

    text: str
    validity: ThoughtValidity
    children: Set[Thought] = Field(default_factory=set)

    def __hash__(self) -> int:
        return id(self)