mirror of
https://github.com/hwchase17/langchain
synced 2024-10-29 17:07:25 +00:00
0c52275bdb
Update the Run object in the tracer to extend that in the SDK to include the parameters necessary for tracking/tracing
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
|
|
)
|