langchain[patch]: updated imports for `langchain.callbacks` (#16060)

Updated imports from 'langchain` to `core` where it is possible

---------

Co-authored-by: Bagatur <baskaryan@gmail.com>
pull/16153/head
Leonid Ganeline 5 months ago committed by GitHub
parent c5f6b828ad
commit 2709d3e5f2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -1,7 +1,7 @@
"""Base callback handler that can be used to handle callbacks in langchain."""
from __future__ import annotations
from langchain_core.callbacks.base import (
from langchain_core.callbacks import (
AsyncCallbackHandler,
BaseCallbackHandler,
BaseCallbackManager,

@ -2,10 +2,9 @@
from typing import Any, Dict, Optional, TextIO, cast
from langchain_core.agents import AgentAction, AgentFinish
from langchain_core.callbacks import BaseCallbackHandler
from langchain_core.utils.input import print_text
from langchain.callbacks.base import BaseCallbackHandler
class FileCallbackHandler(BaseCallbackHandler):
"""Callback Handler that writes to a file."""

@ -3,10 +3,9 @@ from __future__ import annotations
import asyncio
from typing import Any, AsyncIterator, Dict, List, Literal, Union, cast
from langchain_core.callbacks import AsyncCallbackHandler
from langchain_core.outputs import LLMResult
from langchain.callbacks.base import AsyncCallbackHandler
# TODO If used by two LLM runs in parallel this won't work as expected

@ -1,4 +1,4 @@
"""Callback Handler streams to stdout on new llm token."""
from langchain_core.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
from langchain_core.callbacks import StreamingStdOutCallbackHandler
__all__ = ["StreamingStdOutCallbackHandler"]

@ -2,7 +2,7 @@
import sys
from typing import Any, Dict, List, Optional
from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
from langchain_core.callbacks import StreamingStdOutCallbackHandler
DEFAULT_ANSWER_PREFIX_TOKENS = ["Final", "Answer", ":"]

Loading…
Cancel
Save