mirror of
https://github.com/HKUDS/nanobot.git
synced 2026-04-02 17:32:39 +00:00
263069583d
Handle string and dict-shaped responses from OpenAI-compatible backends so non-standard providers no longer crash on missing choices fields. Add regression tests to keep SDK, dict, and plain-text parsing paths aligned.
56 lines
1.7 KiB
Python
56 lines
1.7 KiB
Python
"""Tests for OpenAICompatProvider handling custom/direct endpoints."""
|
|
|
|
from types import SimpleNamespace
|
|
from unittest.mock import patch
|
|
|
|
from nanobot.providers.openai_compat_provider import OpenAICompatProvider
|
|
|
|
|
|
def test_custom_provider_parse_handles_empty_choices() -> None:
|
|
with patch("nanobot.providers.openai_compat_provider.AsyncOpenAI"):
|
|
provider = OpenAICompatProvider()
|
|
response = SimpleNamespace(choices=[])
|
|
|
|
result = provider._parse(response)
|
|
|
|
assert result.finish_reason == "error"
|
|
assert "empty choices" in result.content
|
|
|
|
|
|
def test_custom_provider_parse_accepts_plain_string_response() -> None:
|
|
with patch("nanobot.providers.openai_compat_provider.AsyncOpenAI"):
|
|
provider = OpenAICompatProvider()
|
|
|
|
result = provider._parse("hello from backend")
|
|
|
|
assert result.finish_reason == "stop"
|
|
assert result.content == "hello from backend"
|
|
|
|
|
|
def test_custom_provider_parse_accepts_dict_response() -> None:
|
|
with patch("nanobot.providers.openai_compat_provider.AsyncOpenAI"):
|
|
provider = OpenAICompatProvider()
|
|
|
|
result = provider._parse({
|
|
"choices": [{
|
|
"message": {"content": "hello from dict"},
|
|
"finish_reason": "stop",
|
|
}],
|
|
"usage": {
|
|
"prompt_tokens": 1,
|
|
"completion_tokens": 2,
|
|
"total_tokens": 3,
|
|
},
|
|
})
|
|
|
|
assert result.finish_reason == "stop"
|
|
assert result.content == "hello from dict"
|
|
assert result.usage["total_tokens"] == 3
|
|
|
|
|
|
def test_custom_provider_parse_chunks_accepts_plain_text_chunks() -> None:
|
|
result = OpenAICompatProvider._parse_chunks(["hello ", "world"])
|
|
|
|
assert result.finish_reason == "stop"
|
|
assert result.content == "hello world"
|