Skip to content

Commit

Permalink
langchain[patch]: import nits (#14354)
Browse files Browse the repository at this point in the history
import from core instead of langchain.schema
  • Loading branch information
baskaryan committed Dec 6, 2023
1 parent ce4d81f commit cc76f0e
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 9 deletions.
3 changes: 2 additions & 1 deletion libs/langchain/langchain/llms/watsonxllm.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,10 +2,11 @@
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

Expand Down
Original file line number Diff line number Diff line change
@@ -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


Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit cc76f0e

Please sign in to comment.