mirror of
https://github.com/hwchase17/langchain.git
synced 2025-06-20 05:43:55 +00:00
37b72023fe
removing Sambaverse llm model and references given is not available after Sep/10/2024 <img width="1781" alt="image" src="https://github.com/user-attachments/assets/4dcdb5f7-5264-4a03-b8e5-95c88304e059">
19 lines
583 B
Python
19 lines
583 B
Python
"""Test sambanova API wrapper.
|
|
|
|
In order to run this test, you need to have a sambastudio base url,
|
|
project id, endpoint id, and api key.
|
|
You'll then need to set SAMBASTUDIO_BASE_URL, SAMBASTUDIO_BASE_URI
|
|
SAMBASTUDIO_PROJECT_ID, SAMBASTUDIO_ENDPOINT_ID, and SAMBASTUDIO_API_KEY
|
|
environment variables.
|
|
"""
|
|
|
|
from langchain_community.llms.sambanova import SambaStudio
|
|
|
|
|
|
def test_sambastudio_call() -> None:
|
|
"""Test simple non-streaming call to sambastudio."""
|
|
llm = SambaStudio()
|
|
output = llm.invoke("What is LangChain")
|
|
assert output
|
|
assert isinstance(output, str)
|