feat(api): integrate new AI engines and update chat routes for dynamic agent handling

Author: Davidson Gomes

pyproject.toml

@@ -51,6 +51,9 @@ dependencies = [
     "langgraph==0.4.1",
     "opentelemetry-sdk==1.33.0",
     "opentelemetry-exporter-otlp==1.33.0",
+    "mcp==1.9.0",
+    "crewai==0.120.1",
+    "crewai-tools==0.45.0",
 ]
 
 [project.optional-dependencies]

src/api/chat_routes.py

@@ -39,6 +39,7 @@
     Header,
 )
 from sqlalchemy.orm import Session
+from src.config.settings import settings
 from src.config.database import get_db
 from src.core.jwt_middleware import (
     get_jwt_token,
@@ -49,7 +50,8 @@
     agent_service,
 )
 from src.schemas.chat import ChatRequest, ChatResponse, ErrorResponse, FileData
-from src.services.agent_runner import run_agent, run_agent_stream
+from src.services.adk.agent_runner import run_agent as run_agent_adk, run_agent_stream
+from src.services.crewai.agent_runner import run_agent as run_agent_crewai
 from src.core.exceptions import AgentNotFoundError
 from src.services.service_providers import (
     session_service,
@@ -262,7 +264,7 @@ async def websocket_chat(
 
 
 @router.post(
-    "",
+    "/{agent_id}/{external_id}",
     response_model=ChatResponse,
     responses={
         400: {"model": ErrorResponse},
@@ -272,20 +274,32 @@ async def websocket_chat(
 )
 async def chat(
     request: ChatRequest,
+    agent_id: str,
+    external_id: str,
     _=Depends(get_agent_by_api_key),
     db: Session = Depends(get_db),
 ):
     try:
-        final_response = await run_agent(
-            request.agent_id,
-            request.external_id,
-            request.message,
-            session_service,
-            artifacts_service,
-            memory_service,
-            db,
-            files=request.files,
-        )
+        if settings.AI_ENGINE == "adk":
+            final_response = await run_agent_adk(
+                agent_id,
+                external_id,
+                request.message,
+                session_service,
+                artifacts_service,
+                memory_service,
+                db,
+                files=request.files,
+            )
+        elif settings.AI_ENGINE == "crewai":
+            final_response = await run_agent_crewai(
+                agent_id,
+                external_id,
+                request.message,
+                session_service,
+                db,
+                files=request.files,
+            )
 
         return {
             "response": final_response["final_response"],

src/config/__init__.py

@@ -0,0 +1,3 @@
+from src.config.settings import settings
+
+__all__ = ["settings"]

src/config/settings.py

@@ -57,6 +57,9 @@ class Settings(BaseSettings):
         "POSTGRES_CONNECTION_STRING", "postgresql://postgres:root@localhost:5432/evo_ai"
     )
 
+    # AI engine settings
+    AI_ENGINE: str = os.getenv("AI_ENGINE", "adk")
+
     # Logging settings
     LOG_LEVEL: str = os.getenv("LOG_LEVEL", "INFO")
     LOG_DIR: str = "logs"
@@ -83,11 +86,11 @@ class Settings(BaseSettings):
 
     # Email provider settings
     EMAIL_PROVIDER: str = os.getenv("EMAIL_PROVIDER", "sendgrid")
-    
+
     # SendGrid settings
     SENDGRID_API_KEY: str = os.getenv("SENDGRID_API_KEY", "")
     EMAIL_FROM: str = os.getenv("EMAIL_FROM", "[email protected]")
-    
+
     # SMTP settings
     SMTP_HOST: str = os.getenv("SMTP_HOST", "")
     SMTP_PORT: int = int(os.getenv("SMTP_PORT", 587))
@@ -96,7 +99,7 @@ class Settings(BaseSettings):
     SMTP_USE_TLS: bool = os.getenv("SMTP_USE_TLS", "true").lower() == "true"
     SMTP_USE_SSL: bool = os.getenv("SMTP_USE_SSL", "false").lower() == "true"
     SMTP_FROM: str = os.getenv("SMTP_FROM", "")
-    
+
     APP_URL: str = os.getenv("APP_URL", "http://localhost:8000")
 
     # Server settings

src/schemas/chat.py

@@ -43,9 +43,11 @@ class FileData(BaseModel):
 class ChatRequest(BaseModel):
     """Model to represent a chat request."""
 
-    agent_id: str = Field(..., description="Agent ID to process the message")
-    external_id: str = Field(..., description="External ID for user identification")
     message: str = Field(..., description="User message to the agent")
+    agent_id: Optional[str] = Field(None, description="Agent ID to process the message")
+    external_id: Optional[str] = Field(
+        None, description="External ID for user identification"
+    )
     files: Optional[List[FileData]] = Field(
         None, description="List of files attached to the message"
     )

src/services/__init__.py

@@ -1 +1 @@
-from .agent_runner import run_agent
+from .adk.agent_runner import run_agent

src/services/a2a_task_manager.py

@@ -45,7 +45,7 @@
 )
 from src.services.mcp_server_service import get_mcp_server
 
-from src.services.agent_runner import run_agent, run_agent_stream
+from src.services.adk.agent_runner import run_agent, run_agent_stream
 from src.services.service_providers import (
     session_service,
     artifacts_service,
@@ -388,7 +388,6 @@ async def _stream_task_process(
         self, request: SendTaskStreamingRequest, agent: Agent
     ) -> AsyncIterable[SendTaskStreamingResponse]:
         """Processes a task in streaming mode using the specified agent."""
-        # Extrair e processar arquivos da mesma forma que no método _process_task
         query = self._extract_user_query(request.params)
 
         try:
@@ -448,21 +447,19 @@ async def _stream_task_process(
                     ),
                 )
 
-            # Use os arquivos processados do _extract_user_query
             files = getattr(self, "_last_processed_files", None)
 
-            # Log sobre os arquivos processados
             if files:
                 logger.info(
-                    f"Streaming: Passando {len(files)} arquivos processados para run_agent_stream"
+                    f"Streaming: Uploading {len(files)} files to run_agent_stream"
                 )
                 for file_info in files:
                     logger.info(
-                        f"Streaming: Arquivo sendo enviado: {file_info.filename} ({file_info.content_type})"
+                        f"Streaming: File being sent: {file_info.filename} ({file_info.content_type})"
                     )
             else:
                 logger.warning(
-                    "Streaming: Nenhum arquivo processado disponível para enviar ao agente"
+                    "Streaming: No processed files available to send to the agent"
                 )
 
             async for chunk in run_agent_stream(
@@ -473,7 +470,7 @@ async def _stream_task_process(
                 artifacts_service=artifacts_service,
                 memory_service=memory_service,
                 db=self.db,
-                files=files,  # Passar os arquivos processados para o streaming
+                files=files,
             ):
                 try:
                     chunk_data = json.loads(chunk)

src/services/custom_agents/__init__.py src/services/adk/__init__.pysrc/services/custom_agents/__init__.py renamed to src/services/adk/__init__.py

@@ -36,11 +36,11 @@
 from src.utils.logger import setup_logger
 from src.core.exceptions import AgentNotFoundError
 from src.services.agent_service import get_agent
-from src.services.custom_tools import CustomToolBuilder
-from src.services.mcp_service import MCPService
-from src.services.custom_agents.a2a_agent import A2ACustomAgent
-from src.services.custom_agents.workflow_agent import WorkflowAgent
-from src.services.custom_agents.task_agent import TaskAgent
+from src.services.adk.custom_tools import CustomToolBuilder
+from src.services.adk.mcp_service import MCPService
+from src.services.adk.custom_agents.a2a_agent import A2ACustomAgent
+from src.services.adk.custom_agents.workflow_agent import WorkflowAgent
+from src.services.adk.custom_agents.task_agent import TaskAgent
 from src.services.apikey_service import get_decrypted_api_key
 from sqlalchemy.orm import Session
 from contextlib import AsyncExitStack

src/services/agent_builder.py src/services/adk/agent_builder.pysrc/services/agent_builder.py renamed to src/services/adk/agent_builder.py

@@ -35,7 +35,7 @@
 from src.utils.logger import setup_logger
 from src.core.exceptions import AgentNotFoundError, InternalServerError
 from src.services.agent_service import get_agent
-from src.services.agent_builder import AgentBuilder
+from src.services.adk.agent_builder import AgentBuilder
 from sqlalchemy.orm import Session
 from typing import Optional, AsyncGenerator
 import asyncio