445865e429
Some checks failed
- Integrate GPU scoring inline into reasoning/multi_path.py (auto-uses GPU when available) - Integrate GPU deduplication into multi_agent/consensus_engine.py - Add semantic_search() method to memory/semantic_graph.py with GPU acceleration - Integrate GPU training into self_improvement/training.py AutoTrainer - Fix all 758 ruff lint issues (whitespace, import sorting, unused imports, ambiguous vars, undefined names) - Fix all 40 mypy type errors across the codebase (no-any-return, union-attr, arg-type, etc.) - Fix deprecated ruff config keys (select/ignore -> [tool.ruff.lint]) - Add .dockerignore to exclude .venv/, tests/, docs/ from Docker builds - Add type hints and docstrings to verification/outcome.py - Fix E402 import ordering in witness_agent.py - Fix F821 undefined names in vector_pgvector.py and native.py - Fix E741 ambiguous variable names in reflective.py and recommender.py All 276 tests pass. 0 ruff errors. 0 mypy errors. Co-Authored-By: Nakamoto, S <defi@defi-oracle.io>
68 lines
2.0 KiB
Python
68 lines
2.0 KiB
Python
"""Stub LLM adapter for tests; returns fixed responses."""
|
|
|
|
import json
|
|
from typing import Any
|
|
|
|
from fusionagi.adapters.base import LLMAdapter
|
|
|
|
|
|
class StubAdapter(LLMAdapter):
|
|
"""
|
|
Returns configurable fixed responses; no API calls.
|
|
|
|
Useful for testing without making actual LLM API calls.
|
|
Supports both text and structured (JSON) responses.
|
|
"""
|
|
|
|
def __init__(
|
|
self,
|
|
response: str = "Stub response",
|
|
structured_response: dict[str, Any] | list[Any] | None = None,
|
|
) -> None:
|
|
"""
|
|
Initialize the stub adapter.
|
|
|
|
Args:
|
|
response: Fixed text response for complete().
|
|
structured_response: Fixed structured response for complete_structured().
|
|
"""
|
|
self._response = response
|
|
self._structured_response = structured_response
|
|
|
|
def complete(
|
|
self,
|
|
messages: list[dict[str, str]],
|
|
**kwargs: Any,
|
|
) -> str:
|
|
"""Return the configured stub response."""
|
|
return self._response
|
|
|
|
def complete_structured(
|
|
self,
|
|
messages: list[dict[str, str]],
|
|
schema: dict[str, Any] | None = None,
|
|
**kwargs: Any,
|
|
) -> Any:
|
|
"""
|
|
Return the configured structured response.
|
|
|
|
If no structured_response was configured, attempts to parse
|
|
the text response as JSON, or returns None.
|
|
"""
|
|
if self._structured_response is not None:
|
|
return self._structured_response
|
|
|
|
# Try to parse text response as JSON
|
|
try:
|
|
return json.loads(self._response)
|
|
except json.JSONDecodeError:
|
|
return None
|
|
|
|
def set_response(self, response: str) -> None:
|
|
"""Update the text response (useful for test scenarios)."""
|
|
self._response = response
|
|
|
|
def set_structured_response(self, response: dict[str, Any] | list[Any] | None) -> None:
|
|
"""Update the structured response (useful for test scenarios)."""
|
|
self._structured_response = response
|