Add Writer, Banana, Modal, StochasticAI (#1270)

Add LLM wrappers and examples for Banana, Writer, Modal, Stochastic AI

Added rigid json format for Banana and Modal
This commit is contained in:
Enrico Shippole
2023-02-24 09:58:58 -05:00
committed by GitHub
parent 5457d48416
commit 9becdeaadf
20 changed files with 1071 additions and 2 deletions

View File

@@ -0,0 +1,10 @@
"""Test BananaDev API wrapper."""
from langchain.llms.bananadev import Banana
def test_banana_call() -> None:
"""Test valid call to BananaDev."""
llm = Banana()
output = llm("Say foo:")
assert isinstance(output, str)

View File

@@ -0,0 +1,10 @@
"""Test Modal API wrapper."""
from langchain.llms.modal import Modal
def test_modal_call() -> None:
"""Test valid call to Modal."""
llm = Modal()
output = llm("Say foo:")
assert isinstance(output, str)

View File

@@ -0,0 +1,10 @@
"""Test StochasticAI API wrapper."""
from langchain.llms.stochasticai import StochasticAI
def test_stochasticai_call() -> None:
"""Test valid call to StochasticAI."""
llm = StochasticAI()
output = llm("Say foo:")
assert isinstance(output, str)

View File

@@ -0,0 +1,10 @@
"""Test Writer API wrapper."""
from langchain.llms.writer import Writer
def test_writer_call() -> None:
"""Test valid call to Writer."""
llm = Writer()
output = llm("Say foo:")
assert isinstance(output, str)