| import sys, os | |
| import traceback | |
| from dotenv import load_dotenv | |
| load_dotenv() | |
| import os, io | |
| sys.path.insert( | |
| 0, os.path.abspath("../..") | |
| ) # Adds the parent directory to the system path | |
| import pytest | |
| import litellm | |
| def test_get_llm_provider(): | |
| _, response, _, _ = litellm.get_llm_provider(model="anthropic.claude-v2:1") | |
| assert response == "bedrock" | |
| test_get_llm_provider() | |