mirror of
https://github.com/kennethreitz/langchain.git
synced 2026-06-05 23:00:18 +00:00
Richy Wang
45bb414be2
- Add langchain.llms.Tonyi for text completion, in examples into the Tonyi Text API, - Add system tests. Note async completion for the Text API is not yet supported and will be included in a future PR. Dependencies: dashscope. It will be installed manually cause it is not need by everyone. Happy for feedback on any aspect of this PR @hwchase17 @baskaryan.
28 lines
759 B
Python
28 lines
759 B
Python
"""Test Tongyi API wrapper."""
|
|
from langchain.llms.tongyi import Tongyi
|
|
from langchain.schema import LLMResult
|
|
|
|
|
|
def test_tongyi_call() -> None:
|
|
"""Test valid call to tongyi."""
|
|
llm = Tongyi()
|
|
output = llm("who are you")
|
|
assert isinstance(output, str)
|
|
|
|
|
|
def test_tongyi_generate() -> None:
|
|
"""Test valid call to tongyi."""
|
|
llm = Tongyi()
|
|
output = llm.generate(["who are you"])
|
|
assert isinstance(output, LLMResult)
|
|
assert isinstance(output.generations, list)
|
|
|
|
|
|
def test_tongyi_generate_stream() -> None:
|
|
"""Test valid call to tongyi."""
|
|
llm = Tongyi(streaming=True)
|
|
output = llm.generate(["who are you"])
|
|
print(output)
|
|
assert isinstance(output, LLMResult)
|
|
assert isinstance(output.generations, list)
|