apunkt/rowboat
1
0
Fork 0
mirror of https://github.com/rowboatlabs/rowboat.git synced 2026-05-22 18:45:19 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

log billing usage

Browse source
This commit is contained in:
Ramnique Singh 2025-08-05 17:17:42 +05:30
parent 7592df8068
commit b3a245484f
1 changed files with 13 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

17
apps/rowboat/src/application/use-cases/conversations/run-conversation-turn.use-case.ts
View file

@ -1,6 +1,6 @@
import { Turn, TurnEvent } from "@/src/entities/models/turn";
import { USE_BILLING } from "@/app/lib/feature_flags";
import { authorize, getCustomerIdForProject } from "@/app/lib/billing";
import { authorize, getCustomerIdForProject, logUsage } from "@/app/lib/billing";
import { NotFoundError } from '@/src/entities/errors/common';
import { IConversationsRepository } from "@/src/application/repositories/conversations.repository.interface";
import { streamResponse } from "@/app/lib/agents";
@ -63,14 +63,15 @@ export class RunConversationTurnUseCase implements IRunConversationTurnUseCase {
await this.usageQuotaPolicy.assertAndConsume(projectId);
// Check billing auth
let billingCustomerId: string | null = null;
if (USE_BILLING) {
// get billing customer id for project
const customerId = await getCustomerIdForProject(projectId);
billingCustomerId = await getCustomerIdForProject(projectId);
const agentModels = conversation.workflow.agents.reduce((acc, agent) => {
acc.push(agent.model);
return acc;
}, [] as string[]);
const response = await authorize(customerId, {
const response = await authorize(billingCustomerId, {
type: 'agent_response',
data: {
agentModels,
@ -130,7 +131,7 @@ export class RunConversationTurnUseCase implements IRunConversationTurnUseCase {
const turn = await this.conversationsRepository.addTurn(data.conversationId, {
trigger: data.trigger,
input: data.input,
output: outputMessages,
output: outputMessages,
});
// yield event
@ -141,5 +142,13 @@ export class RunConversationTurnUseCase implements IRunConversationTurnUseCase {
}
}
}
// Log billing usage
if (USE_BILLING && billingCustomerId) {
await logUsage(billingCustomerId, {
type: "agent_messages",
amount: outputMessages.length,
});
}
}
}