9e4c7ad07f
Co-authored-by: Shubham Naik <shubham.naik10@gmail.com> Co-authored-by: Matthew Zhou <mattzh1314@gmail.com> Co-authored-by: Matt Zhou <mattzhou@Matts-MacBook-Pro.local> Co-authored-by: Shubham Naik <shub@memgpt.ai>
44 lines
944 B
Python
44 lines
944 B
Python
import os
|
|
|
|
from letta.providers import AnthropicProvider, OpenAIProvider
|
|
|
|
|
|
def test_openai():
|
|
|
|
provider = OpenAIProvider(api_key=os.getenv("OPENAI_API_KEY"))
|
|
models = provider.list_llm_models()
|
|
print(models)
|
|
|
|
|
|
def test_anthropic():
|
|
if os.getenv("ANTHROPIC_API_KEY") is None:
|
|
return
|
|
provider = AnthropicProvider(api_key=os.getenv("ANTHROPIC_API_KEY"))
|
|
models = provider.list_llm_models()
|
|
print(models)
|
|
|
|
|
|
# def test_groq():
|
|
# provider = GroqProvider(api_key=os.getenv("GROQ_API_KEY"))
|
|
# models = provider.list_llm_models()
|
|
# print(models)
|
|
#
|
|
#
|
|
# def test_ollama():
|
|
# provider = OllamaProvider()
|
|
# models = provider.list_llm_models()
|
|
# print(models)
|
|
#
|
|
#
|
|
# def test_googleai():
|
|
# provider = GoogleAIProvider(api_key=os.getenv("GEMINI_API_KEY"))
|
|
# models = provider.list_llm_models()
|
|
# print(models)
|
|
#
|
|
#
|
|
# test_googleai()
|
|
# test_ollama()
|
|
# test_groq()
|
|
# test_openai()
|
|
# test_anthropic()
|