refactor: 移除 crewai agent 架构相关代码并更新 LLM 配置

2026-03-04 11:37:09 +08:00
parent 87399f74c8
commit b02a322bf3
71 changed files with 1045 additions and 7499 deletions
@@ -1,27 +0,0 @@
-from __future__ import annotations
-
-from pathlib import Path
-
-
-def test_initial_migration_exists_and_creates_expected_tables() -> None:
-    versions_dir = Path(__file__).resolve().parents[3] / "alembic" / "versions"
-    migration_files = sorted(versions_dir.glob("20260226_*.py"))
-    assert len(migration_files) == 5, "split initial migrations should exist"
-
-    content = "\n".join(m.read_text(encoding="utf-8") for m in migration_files)
-
-    # New tables from social data model redesign
-    assert "create_table(" in content and "automation_jobs" in content
-    assert "create_table(" in content and "user_agents" in content
-    assert "create_table(" in content and "memories" in content
-    assert "create_table(" in content and "friendships" in content
-    assert "create_table(" in content and "groups" in content
-    assert "create_table(" in content and "group_members" in content
-    assert "create_table(" in content and "schedule_items" in content
-    assert "create_table(" in content and "schedule_subscriptions" in content
-    assert "create_table(" in content and "inbox_messages" in content
-    assert "create_table(" in content and "todos" in content
-    assert "create_table(" in content and "todo_sources" in content
-    assert "create_table(" in content and "profiles" in content
-    assert "create_table(" in content and "sessions" in content
-    assert "create_table(" in content and "messages" in content
@@ -1,119 +0,0 @@
-from __future__ import annotations
-
-from uuid import uuid4
-
-import pytest
-from sqlalchemy import Column, String, Table, select
-from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker, create_async_engine
-
-from core.db.base import Base
-from models.agent_chat_message import AgentChatMessage
-from models.agent_chat_session import AgentChatSession, AgentChatSessionStatus
-from models.llm import Llm
-from models.llm_factory import LlmFactory
-
-
-@pytest.fixture
-async def db_engine():
-    users_table = Table(
-        "users",
-        Base.metadata,
-        Column("id", String, primary_key=True),
-        schema="auth",
-        extend_existing=True,
-    )
-    engine = create_async_engine("sqlite+aiosqlite:///:memory:", echo=False)
-    async with engine.begin() as conn:
-        await conn.exec_driver_sql("ATTACH DATABASE ':memory:' AS auth")
-        await conn.run_sync(Base.metadata.create_all)
-    yield engine
-    Base.metadata.remove(users_table)
-    await engine.dispose()
-
-
-@pytest.fixture
-async def db_session(db_engine):
-    async_session = async_sessionmaker(
-        bind=db_engine,
-        class_=AsyncSession,
-        expire_on_commit=False,
-    )
-    async with async_session() as session:
-        yield session
-        await session.rollback()
-
-
-@pytest.mark.asyncio
-async def test_llm_factory_and_llm_relationship(db_session: AsyncSession) -> None:
-    factory = LlmFactory(
-        name="qwen",
-        request_url="https://dashscope.aliyuncs.com/compatible-mode/v1",
-        avatar="https://cdn.example.com/qwen.png",
-    )
-    db_session.add(factory)
-    await db_session.flush()
-
-    llm = Llm(
-        factory_id=factory.id,
-        model_code="qwen3.5-flash",
-    )
-    db_session.add(llm)
-    await db_session.commit()
-
-    found_llm = await db_session.get(Llm, llm.id)
-    assert found_llm is not None
-    assert found_llm.factory_id == factory.id
-
-
-@pytest.mark.asyncio
-async def test_session_status_supports_required_values(
-    db_session: AsyncSession,
-) -> None:
-    user_id = uuid4()
-    session = AgentChatSession(
-        user_id=user_id,
-        title="test",
-        status="pending",
-    )
-    db_session.add(session)
-    await db_session.commit()
-
-    statuses = [
-        AgentChatSessionStatus.PENDING,
-        AgentChatSessionStatus.RUNNING,
-        AgentChatSessionStatus.COMPLETED,
-        AgentChatSessionStatus.FAILED,
-    ]
-    for status in statuses:
-        session.status = status
-        await db_session.commit()
-        await db_session.refresh(session)
-        assert session.status == status
-
-
-@pytest.mark.asyncio
-async def test_messages_role_supports_tool(db_session: AsyncSession) -> None:
-    user_id = uuid4()
-    session = AgentChatSession(
-        user_id=user_id,
-        title="tool test",
-        status="pending",
-    )
-    db_session.add(session)
-    await db_session.flush()
-
-    message = AgentChatMessage(
-        session_id=session.id,
-        seq=1,
-        role="tool",
-        content="tool output",
-        cost=0,
-    )
-    db_session.add(message)
-    await db_session.commit()
-
-    result = await db_session.execute(
-        select(AgentChatMessage).where(AgentChatMessage.session_id == session.id)
-    )
-    found = result.scalar_one()
-    assert found.role == "tool"