langchain[patch]: import nits (#14354)

import from core instead of langchain.schema
pull/14331/head
Bagatur 6 months ago committed by GitHub
parent ce4d81f88b
commit cc76f0e834
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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…
Cancel
Save