diff --git a/src/api/agent_routes.py b/src/api/agent_routes.py index 8b82431a..19ab00e3 100644 --- a/src/api/agent_routes.py +++ b/src/api/agent_routes.py @@ -1,10 +1,10 @@ from datetime import datetime import asyncio import os -from fastapi import APIRouter, Depends, HTTPException, status, Header, Request +from fastapi import APIRouter, Depends, HTTPException, status, Header from sqlalchemy.orm import Session from src.config.database import get_db -from typing import List, Dict, Any, Optional +from typing import List, Dict, Any import uuid from src.core.jwt_middleware import ( get_jwt_token, diff --git a/src/api/chat_routes.py b/src/api/chat_routes.py index be8746d3..5062a2e7 100644 --- a/src/api/chat_routes.py +++ b/src/api/chat_routes.py @@ -28,7 +28,6 @@ from src.services.service_providers import ( from datetime import datetime import logging import json -from fastapi.responses import StreamingResponse logger = logging.getLogger(__name__) diff --git a/src/services/streaming_service.py b/src/services/streaming_service.py index 8ea3f67d..ff385b1f 100644 --- a/src/services/streaming_service.py +++ b/src/services/streaming_service.py @@ -6,7 +6,6 @@ from datetime import datetime from ..schemas.streaming import ( JSONRPCRequest, TaskStatusUpdateEvent, - TaskArtifactUpdateEvent, ) from ..services.agent_runner import run_agent from ..services.service_providers import ( diff --git a/src/utils/streaming.py b/src/utils/streaming.py index 46167178..b2013e0f 100644 --- a/src/utils/streaming.py +++ b/src/utils/streaming.py @@ -1,5 +1,5 @@ import asyncio -from typing import AsyncGenerator, Optional +from typing import AsyncGenerator from fastapi import HTTPException