github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-22 14:49:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

langchain[patch]: import nits (#14354)

Browse Source 
import from core instead of langchain.schema
This commit is contained in:
Bagatur 2023-12-06 11:45:05 -08:00 committed by GitHub
parent ce4d81f88b
commit cc76f0e834
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 7 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
libs/langchain/langchain/llms/watsonxllm.py
View File

@ -2,10 +2,11 @@ import logging
import os import os
from typing import Any, Dict, Iterator, List, Mapping, Optional, Union from typing import Any, Dict, Iterator, List, Mapping, Optional, Union
from langchain_core.outputs import LLMResult
from langchain.callbacks.manager import CallbackManagerForLLMRun from langchain.callbacks.manager import CallbackManagerForLLMRun
from langchain.llms.base import BaseLLM from langchain.llms.base import BaseLLM
from langchain.pydantic_v1 import Extra, SecretStr, root_validator from langchain.pydantic_v1 import Extra, SecretStr, root_validator
from langchain.schema import LLMResult
from langchain.schema.output import Generation, GenerationChunk from langchain.schema.output import Generation, GenerationChunk
from langchain.utils import convert_to_secret_str, get_from_dict_or_env from langchain.utils import convert_to_secret_str, get_from_dict_or_env

10
libs/langchain/tests/integration_tests/chat_models/test_volcengine_maas.py
View File

@ -1,14 +1,10 @@
"""Test volc engine maas chat model.""" """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.callbacks.manager import CallbackManager
from langchain.chat_models.volcengine_maas import VolcEngineMaasChat 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 from tests.unit_tests.callbacks.fake_callback_handler import FakeCallbackHandler

3
libs/langchain/tests/integration_tests/llms/test_volcengine_maas.py
View File

@ -2,8 +2,9 @@
from typing import Generator from typing import Generator
from langchain_core.outputs import LLMResult
from langchain.llms.volcengine_maas import VolcEngineMaasLLM from langchain.llms.volcengine_maas import VolcEngineMaasLLM
from langchain.schema import LLMResult
def test_default_call() -> None: def test_default_call() -> None: