AFFAANhandClaude Opus 5
19e1e84fb7
Independent FastAPI backend for the MaskanX agentic growth platform. Includes the agent runtime, MCP client integrations (Meta Ads, LinkedIn, HubSpot, Tavily, Exa, xAI, Citedy, image generation), PostgreSQL storage for chats and cron jobs, provider and secret management, and the CLI. Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
39 lines
1003 B
Python
39 lines
1003 B
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import annotations
|
|
|
|
from types import SimpleNamespace
|
|
|
|
from adclaw.providers import ollama_manager
|
|
|
|
|
|
def _make_fake_ollama(timeout_box: dict, response: dict):
|
|
class _FakeClient:
|
|
def __init__(self, *, timeout=None, **kwargs):
|
|
_ = kwargs
|
|
timeout_box["timeout"] = timeout
|
|
|
|
def list(self):
|
|
return response
|
|
|
|
return SimpleNamespace(Client=_FakeClient)
|
|
|
|
|
|
def test_list_models_uses_default_timeout(monkeypatch) -> None:
|
|
timeout_box: dict = {}
|
|
fake_ollama = _make_fake_ollama(
|
|
timeout_box,
|
|
{"models": [{"model": "qwen2:7b", "size": 1}]},
|
|
)
|
|
|
|
monkeypatch.delenv("ADCLAW_MODEL_PROVIDER_CHECK_TIMEOUT", raising=False)
|
|
monkeypatch.setattr(
|
|
ollama_manager,
|
|
"_ensure_ollama",
|
|
lambda: fake_ollama,
|
|
)
|
|
|
|
models = ollama_manager.OllamaModelManager.list_models()
|
|
|
|
assert timeout_box["timeout"] == 5.0
|
|
assert [m.name for m in models] == ["qwen2:7b"]
|