ai-char/ai-char-bot
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

feat: Add created_by and cost fields to generation models, populate created_by from the authenticated user, and implement cost calculation.

Browse Source
This commit is contained in:
xds
2026-02-09 01:52:23 +03:00
parent fa3e1bb05f
commit 4461964791
11 changed files with 20 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
api/service/generation_service.py
View File

@@ -109,12 +109,15 @@ class GenerationService:
async def get_running_generations(self) -> List[Generation]:
return await self.dao.generations.get_generations(status=GenerationStatus.RUNNING)
async def create_generation_task(self, generation_request: GenerationRequest) -> GenerationResponse:
async def create_generation_task(self, generation_request: GenerationRequest, user_id: Optional[str] = None) -> GenerationResponse:
gen_id = None
generation_model = None
try:
generation_model = Generation(**generation_request.model_dump())
if user_id:
generation_model.created_by = user_id
gen_id = await self.dao.generations.create_generation(generation_model)
generation_model.id = gen_id