mirror of
https://github.com/hwchase17/langchain.git
synced 2025-06-21 14:18:52 +00:00
langchain[patch]: import nits (#14354)
import from core instead of langchain.schema
This commit is contained in:
parent
ce4d81f88b
commit
cc76f0e834
@ -2,10 +2,11 @@ import logging
|
||||
import os
|
||||
from typing import Any, Dict, Iterator, List, Mapping, Optional, Union
|
||||
|
||||
from langchain_core.outputs import LLMResult
|
||||
|
||||
from langchain.callbacks.manager import CallbackManagerForLLMRun
|
||||
from langchain.llms.base import BaseLLM
|
||||
from langchain.pydantic_v1 import Extra, SecretStr, root_validator
|
||||
from langchain.schema import LLMResult
|
||||
from langchain.schema.output import Generation, GenerationChunk
|
||||
from langchain.utils import convert_to_secret_str, get_from_dict_or_env
|
||||
|
||||
|
@ -1,14 +1,10 @@
|
||||
"""Test volc engine maas chat model."""
|
||||
|
||||
from langchain_core.messages import AIMessage, BaseMessage, HumanMessage
|
||||
from langchain_core.outputs import ChatGeneration, LLMResult
|
||||
|
||||
from langchain.callbacks.manager import CallbackManager
|
||||
from langchain.chat_models.volcengine_maas import VolcEngineMaasChat
|
||||
from langchain.schema import (
|
||||
AIMessage,
|
||||
BaseMessage,
|
||||
ChatGeneration,
|
||||
HumanMessage,
|
||||
LLMResult,
|
||||
)
|
||||
from tests.unit_tests.callbacks.fake_callback_handler import FakeCallbackHandler
|
||||
|
||||
|
||||
|
@ -2,8 +2,9 @@
|
||||
|
||||
from typing import Generator
|
||||
|
||||
from langchain_core.outputs import LLMResult
|
||||
|
||||
from langchain.llms.volcengine_maas import VolcEngineMaasLLM
|
||||
from langchain.schema import LLMResult
|
||||
|
||||
|
||||
def test_default_call() -> None:
|
||||
|
Loading…
Reference in New Issue
Block a user