24 lines
631 B
Python
24 lines
631 B
Python
import unittest
|
|
from unittest.mock import patch
|
|
|
|
from cli.utils import select_llm_provider
|
|
|
|
|
|
class CliProviderSelectionTests(unittest.TestCase):
|
|
@patch("cli.utils.questionary.select")
|
|
def test_select_llm_provider_returns_internal_provider_key(self, mock_select):
|
|
mock_select.return_value.ask.return_value = (
|
|
"ollama",
|
|
"http://localhost:4000/v1",
|
|
"Ollama / llama.cpp",
|
|
)
|
|
|
|
provider, url = select_llm_provider()
|
|
|
|
self.assertEqual(provider, "ollama")
|
|
self.assertEqual(url, "http://localhost:4000/v1")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|