basedosdados · vrtornisiello · Mar 10, 2026 · Mar 10, 2026 · Mar 10, 2026 · Mar 10, 2026
diff --git a/.env.example b/.env.example
@@ -39,7 +39,6 @@ GOOGLE_SERVICE_ACCOUNT=/app/credentials/chatbot-sa.json
 # ============================================================
 MODEL_URI=google_genai:gemini-2.5-flash
 MODEL_TEMPERATURE=0.2
-MAX_TOKENS=524288
 
 # ============================================================
 # ==                   LangSmith settings                   ==

diff --git a/app/agent/__init__.py b/app/agent/__init__.py
@@ -1,3 +0,0 @@
-from .react_agent import ReActAgent
-
-__all__ = ["ReActAgent"]

diff --git a/app/agent/hooks.py b/app/agent/hooks.py
diff --git a/app/agent/react_agent.py b/app/agent/react_agent.py
diff --git a/app/agent/types.py b/app/agent/types.py
diff --git a/app/api/dependencies/agent.py b/app/api/dependencies/agent.py
@@ -1,12 +1,11 @@
 from typing import Annotated
 
 from fastapi import Depends, Request
+from langgraph.graph.state import CompiledStateGraph
 
-from app.agent import ReActAgent
 
-
-def get_agent(request: Request) -> ReActAgent:
+def get_agent(request: Request) -> CompiledStateGraph:
     return request.app.state.agent
 
 
-Agent = Annotated[ReActAgent, Depends(get_agent)]
+Agent = Annotated[CompiledStateGraph, Depends(get_agent)]
Original file line number	Diff line number	Diff line change
		@@ -1,3 +0,0 @@
		from .react_agent import ReActAgent

		__all__ = ["ReActAgent"]