wehub-resource-sync
42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
from __future__ import annotations
|
|
|
|
import pytest
|
|
from omnigent_client._client import OmnigentClient
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_client_query_threads_reasoning_to_temporary_session(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
captured: dict[str, object] = {}
|
|
|
|
class _FakeSession:
|
|
def __init__(self) -> None:
|
|
self.reasoning_effort = None
|
|
|
|
def set_reasoning_effort(self, effort: str | None) -> None:
|
|
captured["effort"] = effort
|
|
self.reasoning_effort = effort
|
|
|
|
async def query(self, input: object, *, files: object = None, stream: bool = False) -> str:
|
|
captured["input"] = input
|
|
captured["files"] = files
|
|
captured["stream"] = stream
|
|
return "ok"
|
|
|
|
client = OmnigentClient("http://example.invalid")
|
|
fake_session = _FakeSession()
|
|
monkeypatch.setattr(client, "session", lambda **kwargs: fake_session)
|
|
try:
|
|
result = await client.query(
|
|
model="agent",
|
|
input="hi",
|
|
reasoning={"effort": "xhigh"},
|
|
)
|
|
finally:
|
|
await client.close()
|
|
|
|
assert result == "ok"
|
|
assert captured["effort"] == "xhigh"
|
|
assert captured["input"] == "hi"
|