new feature: reasoning_content for compatible models in ainvoke_response()

agent_framework_oci/tests/unit/test_llm_rich_response.py

@@ -0,0 +1,66 @@
+from types import SimpleNamespace
+
+import pytest
+
+from agent_framework.llm.base import LLMProvider
+from agent_framework.llm.providers import MockLLMProvider, OCISDKProvider, _extract_reasoning_content
+from agent_framework.llm.types import LLMResponse
+
+
+class LegacyOnlyProvider(LLMProvider):
+    async def ainvoke(self, messages, **kwargs) -> str:
+        return "legacy-answer"
+
+
+@pytest.mark.asyncio
+async def test_legacy_provider_gets_rich_response_fallback_without_breaking_contract():
+    provider = LegacyOnlyProvider()
+
+    legacy = await provider.ainvoke([{"role": "user", "content": "hello"}])
+    rich = await provider.ainvoke_response([{"role": "user", "content": "hello"}])
+
+    assert legacy == "legacy-answer"
+    assert isinstance(legacy, str)
+    assert isinstance(rich, LLMResponse)
+    assert rich.content == legacy
+    assert rich.reasoning_content is None
+
+
+@pytest.mark.asyncio
+async def test_mock_ainvoke_remains_string_and_rich_api_is_opt_in():
+    provider = MockLLMProvider()
+    messages = [{"role": "user", "content": "hello"}]
+
+    legacy = await provider.ainvoke(messages)
+    rich = await provider.ainvoke_response(messages)
+
+    assert isinstance(legacy, str)
+    assert legacy == rich.content
+    assert rich.provider == "mock"
+    assert rich.model == "mock-llm"
+    assert rich.reasoning_content is None
+    assert rich.usage["total_tokens"] > 0
+
+
+def test_openai_compatible_reasoning_content_attribute_is_extracted():
+    message = SimpleNamespace(content="answer", reasoning_content="model reasoning")
+    assert _extract_reasoning_content(message) == "model reasoning"
+
+
+def test_openai_compatible_reasoning_content_model_extra_is_extracted():
+    message = SimpleNamespace(content="answer", model_extra={"reasoning_content": "extra reasoning"})
+    assert _extract_reasoning_content(message) == "extra reasoning"
+
+
+def test_missing_reasoning_content_is_none():
+    message = SimpleNamespace(content="answer")
+    assert _extract_reasoning_content(message) is None
+
+
+def test_oci_sdk_reasoning_is_extracted_from_choice_message():
+    message = SimpleNamespace(content="answer", reasoning_content="oci reasoning")
+    choice = SimpleNamespace(message=message)
+    chat_response = SimpleNamespace(choices=[choice])
+    response = SimpleNamespace(data=SimpleNamespace(chat_response=chat_response))
+
+    assert OCISDKProvider._extract_reasoning_content(response) == "oci reasoning"