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:
Leonid Ganeline 2024-01-17 10:06:59 -08:00 committed by GitHub
parent c5f6b828ad
commit 2709d3e5f2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
5 changed files with 5 additions and 7 deletions

View File

@ -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,

View File

@ -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."""

View 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

View File

@ -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"]

View File

@ -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", ":"]