arsam
621f4f7b32
Import the existing local project snapshot. This commit organizes existing files; it does not represent newly developed functionality.
81 lines
2.5 KiB
Python
81 lines
2.5 KiB
Python
import pytest
|
|
from unittest.mock import AsyncMock
|
|
from axon.models import SearchResult
|
|
|
|
|
|
def make_results(n=3):
|
|
return [
|
|
SearchResult(title=f"T{i}", url=f"https://ex{i}.com", content=f"Content {i}", score=0.9)
|
|
for i in range(n)
|
|
]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_synthesize_answer_returns_string(monkeypatch):
|
|
fake_response = {
|
|
"choices": [{"message": {"content": "The answer is 42."}}]
|
|
}
|
|
|
|
class FakeResponse:
|
|
def raise_for_status(self): pass
|
|
def json(self): return fake_response
|
|
|
|
class FakeClient:
|
|
async def __aenter__(self): return self
|
|
async def __aexit__(self, *_): pass
|
|
async def post(self, *args, **kwargs): return FakeResponse()
|
|
|
|
monkeypatch.setattr("axon.services.synthesizer.httpx.AsyncClient", FakeClient)
|
|
|
|
from axon.services.synthesizer import synthesize_answer
|
|
answer = await synthesize_answer("What is the answer?", make_results())
|
|
assert answer == "The answer is 42."
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_synthesize_answer_empty_results():
|
|
from axon.services.synthesizer import synthesize_answer
|
|
answer = await synthesize_answer("query", [])
|
|
assert "No search results" in answer
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_generate_sub_queries_parses_lines(monkeypatch):
|
|
fake_response = {
|
|
"choices": [{"message": {"content": "Query one\nQuery two\nQuery three"}}]
|
|
}
|
|
|
|
class FakeResponse:
|
|
def raise_for_status(self): pass
|
|
def json(self): return fake_response
|
|
|
|
class FakeClient:
|
|
async def __aenter__(self): return self
|
|
async def __aexit__(self, *_): pass
|
|
async def post(self, *args, **kwargs): return FakeResponse()
|
|
|
|
monkeypatch.setattr("axon.services.synthesizer.httpx.AsyncClient", FakeClient)
|
|
|
|
from axon.services.synthesizer import generate_sub_queries
|
|
queries = await generate_sub_queries("main topic", count=3)
|
|
assert len(queries) == 3
|
|
assert queries[0] == "Query one"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_generate_sub_queries_fallback_on_error(monkeypatch):
|
|
import httpx
|
|
|
|
class FakeClient:
|
|
async def __aenter__(self): return self
|
|
async def __aexit__(self, *_): pass
|
|
async def post(self, *args, **kwargs):
|
|
raise httpx.ConnectError("LLM down")
|
|
|
|
monkeypatch.setattr("axon.services.synthesizer.httpx.AsyncClient", FakeClient)
|
|
|
|
from axon.services.synthesizer import generate_sub_queries
|
|
queries = await generate_sub_queries("main topic", count=3)
|
|
assert len(queries) == 3
|
|
assert "main topic" in queries[0]
|