mirror of
https://github.com/hwchase17/langchain.git
synced 2025-09-02 19:47:13 +00:00
community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463)
Moved the following modules to new package langchain-community in a backwards compatible fashion: ``` mv langchain/langchain/adapters community/langchain_community mv langchain/langchain/callbacks community/langchain_community/callbacks mv langchain/langchain/chat_loaders community/langchain_community mv langchain/langchain/chat_models community/langchain_community mv langchain/langchain/document_loaders community/langchain_community mv langchain/langchain/docstore community/langchain_community mv langchain/langchain/document_transformers community/langchain_community mv langchain/langchain/embeddings community/langchain_community mv langchain/langchain/graphs community/langchain_community mv langchain/langchain/llms community/langchain_community mv langchain/langchain/memory/chat_message_histories community/langchain_community mv langchain/langchain/retrievers community/langchain_community mv langchain/langchain/storage community/langchain_community mv langchain/langchain/tools community/langchain_community mv langchain/langchain/utilities community/langchain_community mv langchain/langchain/vectorstores community/langchain_community mv langchain/langchain/agents/agent_toolkits community/langchain_community mv langchain/langchain/cache.py community/langchain_community mv langchain/langchain/adapters community/langchain_community mv langchain/langchain/callbacks community/langchain_community/callbacks mv langchain/langchain/chat_loaders community/langchain_community mv langchain/langchain/chat_models community/langchain_community mv langchain/langchain/document_loaders community/langchain_community mv langchain/langchain/docstore community/langchain_community mv langchain/langchain/document_transformers community/langchain_community mv langchain/langchain/embeddings community/langchain_community mv langchain/langchain/graphs community/langchain_community mv langchain/langchain/llms community/langchain_community mv langchain/langchain/memory/chat_message_histories community/langchain_community mv langchain/langchain/retrievers community/langchain_community mv langchain/langchain/storage community/langchain_community mv langchain/langchain/tools community/langchain_community mv langchain/langchain/utilities community/langchain_community mv langchain/langchain/vectorstores community/langchain_community mv langchain/langchain/agents/agent_toolkits community/langchain_community mv langchain/langchain/cache.py community/langchain_community ``` Moved the following to core ``` mv langchain/langchain/utils/json_schema.py core/langchain_core/utils mv langchain/langchain/utils/html.py core/langchain_core/utils mv langchain/langchain/utils/strings.py core/langchain_core/utils cat langchain/langchain/utils/env.py >> core/langchain_core/utils/env.py rm langchain/langchain/utils/env.py ``` See .scripts/community_split/script_integrations.sh for all changes
This commit is contained in:
@@ -0,0 +1,46 @@
|
||||
"""Test Replicate API wrapper."""
|
||||
|
||||
from langchain_core.callbacks import CallbackManager
|
||||
|
||||
from langchain_community.llms.replicate import Replicate
|
||||
from tests.unit_tests.callbacks.fake_callback_handler import FakeCallbackHandler
|
||||
|
||||
TEST_MODEL = "replicate/dolly-v2-12b:ef0e1aefc61f8e096ebe4db6b2bacc297daf2ef6899f0f7e001ec445893500e5" # noqa: E501
|
||||
|
||||
|
||||
def test_replicate_call() -> None:
|
||||
"""Test simple non-streaming call to Replicate."""
|
||||
llm = Replicate(model=TEST_MODEL)
|
||||
output = llm("What is LangChain")
|
||||
assert output
|
||||
assert isinstance(output, str)
|
||||
|
||||
|
||||
def test_replicate_streaming_call() -> None:
|
||||
"""Test streaming call to Replicate."""
|
||||
callback_handler = FakeCallbackHandler()
|
||||
callback_manager = CallbackManager([callback_handler])
|
||||
|
||||
llm = Replicate(streaming=True, callback_manager=callback_manager, model=TEST_MODEL)
|
||||
output = llm("What is LangChain")
|
||||
assert output
|
||||
assert isinstance(output, str)
|
||||
|
||||
|
||||
def test_replicate_model_kwargs() -> None:
|
||||
"""Test simple non-streaming call to Replicate."""
|
||||
llm = Replicate(
|
||||
model=TEST_MODEL, model_kwargs={"max_length": 100, "temperature": 0.01}
|
||||
)
|
||||
long_output = llm("What is LangChain")
|
||||
llm = Replicate(
|
||||
model=TEST_MODEL, model_kwargs={"max_length": 10, "temperature": 0.01}
|
||||
)
|
||||
short_output = llm("What is LangChain")
|
||||
assert len(short_output) < len(long_output)
|
||||
assert llm.model_kwargs == {"max_length": 10, "temperature": 0.01}
|
||||
|
||||
|
||||
def test_replicate_input() -> None:
|
||||
llm = Replicate(model=TEST_MODEL, input={"max_length": 10})
|
||||
assert llm.model_kwargs == {"max_length": 10}
|
Reference in New Issue
Block a user