8ae1e64987
Co-authored-by: Charles Packer <packercharles@gmail.com> Co-authored-by: Shubham Naik <shubham.naik10@gmail.com> Co-authored-by: Shubham Naik <shub@memgpt.ai>
18 lines
677 B
Python
18 lines
677 B
Python
from letta.cli.cli import QuickstartChoice, quickstart
|
|
from letta.config import LettaConfig
|
|
|
|
|
|
def test_quickstart():
|
|
|
|
# openai
|
|
quickstart(QuickstartChoice.openai, debug=True, terminal=False)
|
|
config = LettaConfig.load()
|
|
assert config.default_llm_config.model_endpoint_type == "openai"
|
|
assert config.default_embedding_config.embedding_endpoint_type == "openai"
|
|
|
|
# letta
|
|
quickstart(QuickstartChoice.letta_hosted, debug=True, terminal=False)
|
|
config = LettaConfig.load()
|
|
assert config.default_llm_config.model_endpoint == "https://inference.letta.ai"
|
|
assert config.default_embedding_config.embedding_endpoint == "https://embeddings.letta.ai"
|