Skip to content

Commit 3282776

Browse files
authored
core[patch]: mark usage callback handler as beta (#30486)
1 parent 9f345d6 commit 3282776

File tree

1 file changed

+3
-0
lines changed
  • libs/core/langchain_core/callbacks

1 file changed

+3
-0
lines changed

libs/core/langchain_core/callbacks/usage.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,12 +6,14 @@
66
from contextvars import ContextVar
77
from typing import Any, Optional
88

9+
from langchain_core._api import beta
910
from langchain_core.callbacks import BaseCallbackHandler
1011
from langchain_core.messages import AIMessage
1112
from langchain_core.messages.ai import UsageMetadata, add_usage
1213
from langchain_core.outputs import ChatGeneration, LLMResult
1314

1415

16+
@beta()
1517
class UsageMetadataCallbackHandler(BaseCallbackHandler):
1618
"""Callback Handler that tracks AIMessage.usage_metadata.
1719
@@ -67,6 +69,7 @@ def on_llm_end(self, response: LLMResult, **kwargs: Any) -> None:
6769

6870

6971
@contextmanager
72+
@beta()
7073
def get_usage_metadata_callback(
7174
name: str = "usage_metadata_callback",
7275
) -> Generator[UsageMetadataCallbackHandler, None, None]:

0 commit comments

Comments
 (0)