add ai plugins

2025-12-10 07:48:37 +01:00 · 2024-12-19 16:45:53 +01:00
parent 7082b88523
commit 3f26f1ab68
5 changed files with 215 additions and 4 deletions
--- a/dashboard/server/ai/functions/AI_Billing.ts
+++ b/dashboard/server/ai/functions/AI_Billing.ts
@@ -0,0 +1,127 @@
+
+import { ProjectLimitModel } from "@schema/project/ProjectsLimits";
+import { AIPlugin } from "../Plugin";
+import { MAX_LOG_LIMIT_PERCENT } from "@data/broker/Limits";
+import { ProjectModel } from "@schema/project/ProjectSchema";
+import StripeService from "~/server/services/StripeService";
+import { InvoiceData } from "~/server/api/pay/invoices";
+
+export class AiBilling extends AIPlugin<[
+    'getBillingInfo',
+    'getLimits',
+    'getInvoices'
+]> {
+
+    constructor() {
+        super({
+
+            'getInvoices': {
+                handler: async (data: { project_id: string }) => {
+
+                    const project = await ProjectModel.findOne({ _id: data.project_id });
+                    if (!project) return { error: 'Project not found' };
+                    const invoices = await StripeService.getInvoices(project.customer_id);
+                    if (!invoices) return [];
+
+                    return invoices?.data.map(e => {
+                        const result: InvoiceData = {
+                            link: e.invoice_pdf || '',
+                            id: e.number || '',
+                            date: e.created * 1000,
+                            status: e.status || 'NO_STATUS',
+                            cost: e.amount_due
+                        }
+                        return result;
+                    });
+                },
+                tool: {
+                    type: 'function',
+                    function: {
+                        name: 'getInvoices',
+                        description: 'Gets the invoices of the user project',
+                        parameters: {}
+                    }
+                }
+            },
+
+            'getBillingInfo': {
+                handler: async (data: { project_id: string }) => {
+
+                    const project = await ProjectModel.findOne({ _id: data.project_id });
+                    if (!project) return { error: 'Project not found' };
+
+                    if (project.subscription_id === 'onetime') {
+
+                        const projectLimits = await ProjectLimitModel.findOne({ project_id: data.project_id });
+                        if (!projectLimits) return { error: 'Limits not found' }
+
+                        const result = {
+                            premium: project.premium,
+                            premium_type: project.premium_type,
+                            billing_start_at: projectLimits.billing_start_at,
+                            billing_expire_at: projectLimits.billing_expire_at,
+                            limit: projectLimits.limit,
+                            count: projectLimits.events + projectLimits.visits,
+                            subscription_status: StripeService.isDisabled() ? 'Disabled mode' : ('One time payment')
+                        }
+
+                        return result;
+                    }
+
+                    const subscription = await StripeService.getSubscription(project.subscription_id);
+
+                    const projectLimits = await ProjectLimitModel.findOne({ project_id: data.project_id });
+                    if (!projectLimits) return { error: 'Limits not found' }
+
+
+                    const result = {
+                        premium: project.premium,
+                        premium_type: project.premium_type,
+                        billing_start_at: projectLimits.billing_start_at,
+                        billing_expire_at: projectLimits.billing_expire_at,
+                        limit: projectLimits.limit,
+                        count: projectLimits.events + projectLimits.visits,
+                        subscription_status: StripeService.isDisabled() ? 'Disabled mode' : (subscription?.status ?? '?')
+                    }
+
+                    return result;
+                },
+                tool: {
+                    type: 'function',
+                    function: {
+                        name: 'getBillingInfo',
+                        description: 'Gets the informations about the billing of the user project, limits, count, subscription_status, is premium, premium type, billing start at, billing expire at',
+                        parameters: {}
+                    }
+                }
+            },
+
+            'getLimits': {
+                handler: async (data: { project_id: string }) => {
+                    const projectLimits = await ProjectLimitModel.findOne({ project_id: data.project_id });
+                    if (!projectLimits) return { error: 'Project limits not found' };
+                    const TOTAL_COUNT = projectLimits.events + projectLimits.visits;
+                    const COUNT_LIMIT = projectLimits.limit;
+                    return {
+                        total: TOTAL_COUNT,
+                        limit: COUNT_LIMIT,
+                        limited: TOTAL_COUNT > COUNT_LIMIT * MAX_LOG_LIMIT_PERCENT,
+                        percent: Math.round(100 / COUNT_LIMIT * TOTAL_COUNT)
+                    }
+                },
+                tool: {
+                    type: 'function',
+                    function: {
+                        name: 'getLimits',
+                        description: 'Gets the informations about the limits of the user project',
+                        parameters: {}
+                    }
+                }
+            },
+
+
+        })
+    }
+}
+
+export const AiBillingInstance = new AiBilling();