mirror of
https://github.com/hwchase17/langchain
synced 2024-10-29 17:07:25 +00:00
28 lines
761 B
Python
28 lines
761 B
Python
|
import langchain.callbacks.tracers.schemas as schemas
|
||
|
|
from langchain.callbacks.tracers.schemas import __all__ as schemas_all
|
||
|
|
|
||
|
|
|
||
|
|
def test_public_api() -> None:
|
||
|
|
"""Test for changes in the public API."""
|
||
|
|
expected_all = [
|
||
|
|
"BaseRun",
|
||
|
|
"ChainRun",
|
||
|
|
"LLMRun",
|
||
|
|
"Run",
|
||
|
|
"RunTypeEnum",
|
||
|
|
"ToolRun",
|
||
|
|
"TracerSession",
|
||
|
|
"TracerSessionBase",
|
||
|
|
"TracerSessionV1",
|
||
|
|
"TracerSessionV1Base",
|
||
|
|
"TracerSessionV1Create",
|
||
|
|
]
|
||
|
|
|
||
|
|
assert sorted(schemas_all) == expected_all
|
||
|
|
|
||
|
|
# Assert that the object is actually present in the schema module
|
||
|
|
for module_name in expected_all:
|
||
|
|
assert (
|
||
|
|
hasattr(schemas, module_name) and getattr(schemas, module_name) is not None
|
||
|
|
)
|