mirror of
https://github.com/langchain-ai/langgraph.git
synced 2026-08-21 07:02:25 +02:00
31 lines
519 B
Python
31 lines
519 B
Python
from dataclasses import dataclass
|
|
from typing import runtime_checkable
|
|
|
|
from pydantic import BaseModel
|
|
|
|
from langgraph_sdk.schema import (
|
|
_BaseModelLike,
|
|
_DataclassLike,
|
|
)
|
|
|
|
|
|
def rc(cls: type) -> type:
|
|
return runtime_checkable(cls)
|
|
|
|
|
|
class MyModel(BaseModel):
|
|
foo: str
|
|
|
|
|
|
def test_base_model_like():
|
|
assert isinstance(MyModel(foo="test"), rc(_BaseModelLike))
|
|
|
|
|
|
@dataclass
|
|
class MyDataclass:
|
|
foo: str
|
|
|
|
|
|
def test_dataclass_like():
|
|
assert isinstance(MyDataclass(foo="test"), rc(_DataclassLike))
|