Initial commit: add .gitignore and README

fusionagi/adapters/stub_adapter.py

@@ -0,0 +1,67 @@
+"""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