fastapi-fullstack 0.1.7__py3-none-any.whl

fastapi_gen/template/{{cookiecutter.project_slug}}/backend/app/agents/langchain_assistant.py

@@ -0,0 +1,226 @@
+{%- if cookiecutter.enable_ai_agent and cookiecutter.use_langchain %}
+"""Assistant agent with LangChain.
+
+The main conversational agent that can be extended with custom tools.
+"""
+
+import logging
+from typing import Any, TypedDict
+
+from langchain.agents import create_agent
+from langchain.messages import AIMessage, HumanMessage, SystemMessage
+from langchain.tools import tool
+{%- if cookiecutter.use_openai %}
+from langchain_openai import ChatOpenAI
+{%- endif %}
+{%- if cookiecutter.use_anthropic %}
+from langchain_anthropic import ChatAnthropic
+{%- endif %}
+
+from app.agents.prompts import DEFAULT_SYSTEM_PROMPT
+from app.agents.tools import get_current_datetime
+from app.core.config import settings
+
+logger = logging.getLogger(__name__)
+
+
+class AgentContext(TypedDict, total=False):
+    """Runtime context for the agent.
+
+    Passed via context parameter to agent.invoke()/stream().
+    """
+
+    user_id: str | None
+    user_name: str | None
+    metadata: dict[str, Any]
+
+
+class AgentState(TypedDict):
+    """State for the LangChain agent.
+
+    This is what flows through the agent graph.
+    """
+
+    messages: list[Any]
+
+
+@tool
+def current_datetime() -> str:
+    """Get the current date and time.
+
+    Use this tool when you need to know the current date or time.
+    """
+    return get_current_datetime()
+
+
+class LangChainAssistant:
+    """Assistant agent wrapper for conversational AI using LangChain.
+
+    Encapsulates agent creation and execution with tool support.
+    """
+
+    def __init__(
+        self,
+        model_name: str | None = None,
+        temperature: float | None = None,
+        system_prompt: str | None = None,
+    ):
+        self.model_name = model_name or settings.AI_MODEL
+        self.temperature = temperature or settings.AI_TEMPERATURE
+        self.system_prompt = system_prompt or DEFAULT_SYSTEM_PROMPT
+        self._agent = None
+        self._tools = [current_datetime]
+
+    def _create_agent(self):
+        """Create and configure the LangChain agent."""
+{%- if cookiecutter.use_openai %}
+        model = ChatOpenAI(
+            model=self.model_name,
+            temperature=self.temperature,
+            api_key=settings.OPENAI_API_KEY,
+        )
+{%- endif %}
+{%- if cookiecutter.use_anthropic %}
+        model = ChatAnthropic(
+            model=self.model_name,
+            temperature=self.temperature,
+            api_key=settings.ANTHROPIC_API_KEY,
+        )
+{%- endif %}
+
+        agent = create_agent(
+            model=model,
+            tools=self._tools,
+            system_prompt=self.system_prompt,
+        )
+
+        return agent
+
+    @property
+    def agent(self):
+        """Get or create the agent instance."""
+        if self._agent is None:
+            self._agent = self._create_agent()
+        return self._agent
+
+    @staticmethod
+    def _convert_history(
+        history
+        : list[dict[str, str]] | None
+    ) -> list[HumanMessage | AIMessage | SystemMessage]:
+        """Convert conversation history to LangChain message format."""
+        messages: list[HumanMessage | AIMessage | SystemMessage] = []
+
+        for msg in history or []:
+            if msg["role"] == "user":
+                messages.append(HumanMessage(content=msg["content"]))
+            elif msg["role"] == "assistant":
+                messages.append(AIMessage(content=msg["content"]))
+            elif msg["role"] == "system":
+                messages.append(SystemMessage(content=msg["content"]))
+
+        return messages
+
+    async def run(
+        self,
+        user_input: str,
+        history: list[dict[str, str]] | None = None,
+        context: AgentContext | None = None,
+    ) -> tuple[str, list[Any], AgentContext]:
+        """Run agent and return the output along with tool call events.
+
+        Args:
+            user_input: User's message.
+            history: Conversation history as list of {"role": "...", "content": "..."}.
+            context: Optional runtime context with user info.
+
+        Returns:
+            Tuple of (output_text, tool_events, context).
+        """
+        messages = self._convert_history(history)
+        messages.append(HumanMessage(content=user_input))
+
+        agent_context: AgentContext = context if context is not None else {}
+
+        logger.info(f"Running agent with user input: {user_input[:100]}...")
+
+        result = self.agent.invoke(
+            {"messages": messages},
+            config={"configurable": agent_context} if agent_context else None,
+        )
+
+        # Extract the final response
+        output = ""
+        tool_events: list[Any] = []
+
+        for message in result.get("messages", []):
+            if hasattr(message, "content") and isinstance(message, AIMessage):
+                output = message.content
+            if hasattr(message, "tool_calls") and message.tool_calls:
+                tool_events.extend(message.tool_calls)
+
+        logger.info(f"Agent run complete. Output length: {len(output)} chars")
+
+        return output, tool_events, agent_context
+
+    def stream(
+        self,
+        user_input: str,
+        history: list[dict[str, str]] | None = None,
+        context: AgentContext | None = None,
+    ):
+        """Stream agent execution with token-level streaming.
+
+        Args:
+            user_input: User's message.
+            history: Conversation history.
+            context: Optional runtime context.
+
+        Yields:
+            Tuples of (stream_mode, data) for streaming responses.
+            - stream_mode="messages": (token, metadata) for LLM tokens
+            - stream_mode="updates": state updates after each step
+        """
+        messages = self._convert_history(history)
+        messages.append(HumanMessage(content=user_input))
+
+        agent_context: AgentContext = context if context is not None else {}
+
+        yield from self.agent.stream(
+            {"messages": messages},
+            stream_mode=["messages", "updates"],
+            config={"configurable": agent_context} if agent_context else None,
+        )
+
+
+def get_agent() -> LangChainAssistant:
+    """Factory function to create a LangChainAssistant.
+
+    Returns:
+        Configured LangChainAssistant instance.
+    """
+    return LangChainAssistant()
+
+
+async def run_agent(
+    user_input: str,
+    history: list[dict[str, str]],
+    context: AgentContext | None = None,
+) -> tuple[str, list[Any], AgentContext]:
+    """Run agent and return the output along with tool call events.
+
+    This is a convenience function for backwards compatibility.
+
+    Args:
+        user_input: User's message.
+        history: Conversation history.
+        context: Optional runtime context.
+
+    Returns:
+        Tuple of (output_text, tool_events, context).
+    """
+    agent = get_agent()
+    return await agent.run(user_input, history, context)
+{%- else %}
+"""LangChain Assistant agent - not configured."""
+{%- endif %}

fastapi_gen/template/{{cookiecutter.project_slug}}/backend/app/agents/prompts.py

@@ -0,0 +1,10 @@
+{%- if cookiecutter.enable_ai_agent %}
+"""System prompts for AI agents.
+
+Centralized location for all agent prompts to make them easy to find and modify.
+"""
+
+DEFAULT_SYSTEM_PROMPT = """You are a helpful assistant."""
+{%- else %}
+"""AI Agent prompts - not configured."""
+{%- endif %}

fastapi_gen/template/{{cookiecutter.project_slug}}/backend/app/agents/tools/__init__.py

@@ -0,0 +1,13 @@
+{%- if cookiecutter.enable_ai_agent %}
+"""Agent tools module.
+
+This module contains utility functions that can be used as agent tools.
+Tools are registered in the agent definition using @agent.tool decorator.
+"""
+
+from app.agents.tools.datetime_tool import get_current_datetime
+
+__all__ = ["get_current_datetime"]
+{%- else %}
+"""Agent tools - not configured."""
+{%- endif %}

fastapi_gen/template/{{cookiecutter.project_slug}}/backend/app/agents/tools/datetime_tool.py

@@ -0,0 +1,17 @@
+{%- if cookiecutter.enable_ai_agent %}
+"""Date and time utilities for agents."""
+
+from datetime import datetime
+
+
+def get_current_datetime() -> str:
+    """Get the current date and time.
+
+    Returns:
+        A string with the current date and time.
+    """
+    now = datetime.now()
+    return f"Current date: {now.strftime('%Y-%m-%d')}, Current time: {now.strftime('%H:%M:%S')}"
+{%- else %}
+"""Datetime tools - not configured."""
+{%- endif %}

fastapi_gen/template/{{cookiecutter.project_slug}}/backend/app/api/__init__.py

@@ -0,0 +1 @@
+"""API module."""