mirror of
https://github.com/hwchase17/langchain.git
synced 2025-06-28 17:38:36 +00:00
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>
This commit is contained in:
parent
c5f6b828ad
commit
2709d3e5f2
@ -1,7 +1,7 @@
|
|||||||
"""Base callback handler that can be used to handle callbacks in langchain."""
|
"""Base callback handler that can be used to handle callbacks in langchain."""
|
||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
from langchain_core.callbacks.base import (
|
from langchain_core.callbacks import (
|
||||||
AsyncCallbackHandler,
|
AsyncCallbackHandler,
|
||||||
BaseCallbackHandler,
|
BaseCallbackHandler,
|
||||||
BaseCallbackManager,
|
BaseCallbackManager,
|
||||||
|
@ -2,10 +2,9 @@
|
|||||||
from typing import Any, Dict, Optional, TextIO, cast
|
from typing import Any, Dict, Optional, TextIO, cast
|
||||||
|
|
||||||
from langchain_core.agents import AgentAction, AgentFinish
|
from langchain_core.agents import AgentAction, AgentFinish
|
||||||
|
from langchain_core.callbacks import BaseCallbackHandler
|
||||||
from langchain_core.utils.input import print_text
|
from langchain_core.utils.input import print_text
|
||||||
|
|
||||||
from langchain.callbacks.base import BaseCallbackHandler
|
|
||||||
|
|
||||||
|
|
||||||
class FileCallbackHandler(BaseCallbackHandler):
|
class FileCallbackHandler(BaseCallbackHandler):
|
||||||
"""Callback Handler that writes to a file."""
|
"""Callback Handler that writes to a file."""
|
||||||
|
@ -3,10 +3,9 @@ from __future__ import annotations
|
|||||||
import asyncio
|
import asyncio
|
||||||
from typing import Any, AsyncIterator, Dict, List, Literal, Union, cast
|
from typing import Any, AsyncIterator, Dict, List, Literal, Union, cast
|
||||||
|
|
||||||
|
from langchain_core.callbacks import AsyncCallbackHandler
|
||||||
from langchain_core.outputs import LLMResult
|
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
|
# 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."""
|
"""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"]
|
__all__ = ["StreamingStdOutCallbackHandler"]
|
||||||
|
@ -2,7 +2,7 @@
|
|||||||
import sys
|
import sys
|
||||||
from typing import Any, Dict, List, Optional
|
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", ":"]
|
DEFAULT_ANSWER_PREFIX_TOKENS = ["Final", "Answer", ":"]
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user