Spaces:
Paused
Paused
File size: 1,517 Bytes
287f3d3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 | """Database Design Agent."""
from __future__ import annotations
from pydantic import BaseModel
from ..prompts import PROMPTS, build_user_prompt
from ..schemas import DatabaseOutput, ProjectContext
from .base import (
BaseAgent,
RevisionInstruction,
project_context_payload,
revision_instruction_text,
)
from .digest import (
condense_context,
digest_architecture,
digest_requirements,
dumps,
summary_or_digest,
)
class DatabaseAgent(BaseAgent):
name = "database"
output_schema: type[BaseModel] = DatabaseOutput
system_prompt = PROMPTS["database"].system
async def _execute(
self, context: ProjectContext, revision: RevisionInstruction | None = None
) -> DatabaseOutput:
user_prompt = build_user_prompt(
"database",
project_context=dumps(
condense_context(project_context_payload(context), downstream=True)
),
requirements=summary_or_digest(
context, "requirements", digest_requirements(context.requirements or {})
),
architecture=summary_or_digest(
context, "architecture", digest_architecture(context.architecture or {})
),
)
if revision is not None:
user_prompt += revision_instruction_text(revision)
return await self._llm.generate(
self.system_prompt, user_prompt, DatabaseOutput, stats=self._stats
) |