forked from Archives/langchain
13 lines
390 B
Python
13 lines
390 B
Python
"""Test manifest integration."""
|
|
from langchain.llms.manifest import ManifestWrapper
|
|
|
|
|
|
def test_manifest_wrapper() -> None:
|
|
"""Test manifest wrapper."""
|
|
from manifest import Manifest
|
|
|
|
manifest = Manifest(client_name="openai")
|
|
llm = ManifestWrapper(client=manifest, llm_kwargs={"temperature": 0})
|
|
output = llm("The capital of New York is:")
|
|
assert output == "Albany"
|