Skip to content

Commit 05266bb

Browse files
Remove useless agent files (#22)
* Update av to version 16.0.1 - upgraded av from 10.0.0 to 16.0.1 * docs: update README with virtual environment setup - Added section for creating a virtual environment - Updated command for creating venv - Added instructions for activating venv * fix(docker): downgrade av package version - Changed av from 16.0.1 to 10.0.0 * Remove unused data agents and files - Deleted AdministratorAssistant class - Removed UserAgent class - Eliminated GroupChatManager class - Cleared out persistence files - Updated README installation instructions * refactor(agent): remove WebSocketManager and models - Delete WebSocketManager class - Remove connection management logic - Eliminate TaskResult and Message models - Clean up related dependency functions * feat(agent): remove WebSocket run endpoint - Delete WebSocket endpoint for runs - Remove related connection handling logic - Eliminate message processing for runs * chore: comment out agent imports and usage - Comment out agent router inclusion - Comment out manager initialization - Comment out manager cleanup during shutdown --------- Co-authored-by: Haihui.Wang <wanghh2000@163.com>
1 parent 057b392 commit 05266bb

2 files changed

Lines changed: 6 additions & 6 deletions

File tree

data_server/api/api_router.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from .endpoints import task_log
66
from .endpoints import op
77
from .endpoints import tool
8-
from .endpoints import agent
8+
# from .endpoints import agent
99
from .endpoints import datasource
1010
from .endpoints import formatify
1111
from .endpoints import op_pic_upload
@@ -19,7 +19,7 @@
1919
api_router.include_router(template.router, prefix="/templates", tags=["Templates"])
2020
api_router.include_router(op.router, prefix="/ops", tags=["Operators"])
2121
api_router.include_router(tool.router, prefix="/tools", tags=["Tools"])
22-
api_router.include_router(agent.router, prefix="/agent", tags=["Agent"])
22+
# api_router.include_router(agent.router, prefix="/agent", tags=["Agent"])
2323
api_router.include_router(datasource.router, prefix="/datasource", tags=["Datasource"])
2424
api_router.include_router(formatify.router, prefix="/formatify", tags=["Formatify"])
2525
api_router.include_router(celery_server.router, prefix="/celery", tags=["Celery 相关接口"])

data_server/main.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from contextlib import asynccontextmanager
77
from typing import AsyncGenerator
88
from loguru import logger
9-
from data_server.agent.deps import init_managers, cleanup_managers
9+
# from data_server.agent.deps import init_managers, cleanup_managers
1010
from apscheduler.schedulers.background import BackgroundScheduler
1111
from apscheduler.triggers.interval import IntervalTrigger
1212
from data_celery.main import celery_app
@@ -43,8 +43,8 @@ async def lifespan(app: FastAPI) -> AsyncGenerator[None, None]:
4343
# logger.info("S3 storage initialized successfully")
4444

4545
# Initialize managers (DB, Connection, Team)
46-
await init_managers()
47-
logger.info("Managers initialized successfully")
46+
# await init_managers()
47+
# logger.info("Managers initialized successfully")
4848

4949
# _scheduler = BackgroundScheduler()
5050
# _scheduler.add_job(
@@ -94,7 +94,7 @@ async def lifespan(app: FastAPI) -> AsyncGenerator[None, None]:
9494

9595

9696
logger.info("Cleaning up application resources...")
97-
await cleanup_managers()
97+
# await cleanup_managers()
9898
logger.info("Application shutdown complete")
9999
except Exception as e:
100100
logger.error(f"Error during shutdown: {str(e)}")

0 commit comments

Comments
 (0)