chris/OSVauco
1
0
Fork 0
Code Issues 1 Pull Requests Actions 2 Packages Projects Releases Wiki Activity

feat(cg4): implement BillingAgent with BigQuery backend

Browse Source
This commit is contained in:
Chris Christiansen 2026-05-27 17:23:01 +00:00
parent a82c07e692
commit 33c811c4f3
1 changed files with 4 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
ml/billing_agent.py
View File

@ -1,7 +1,6 @@
import os import os
import datetime import datetime
from google.cloud import bigquery from google.cloud import bigquery
from google.cloud import billing
class BillingAgent: class BillingAgent:
def __init__(self): def __init__(self):
@ -9,13 +8,11 @@ class BillingAgent:
if not self.project_id: if not self.project_id:
raise ValueError("GOOGLE_CLOUD_PROJECT environment variable not set.") raise ValueError("GOOGLE_CLOUD_PROJECT environment variable not set.")
self.billing_account_id = os.environ.get("BILLING_ACCOUNT_ID") self.billing_table = os.environ.get("BILLING_TABLE")
if not self.billing_account_id: if not self.billing_table:
raise ValueError("BILLING_ACCOUNT_ID environment variable not set.") raise ValueError("BILLING_TABLE environment variable not set.")
self.bq_client = bigquery.Client(project=self.project_id) self.bq_client = bigquery.Client(project=self.project_id)
self.billing_client = billing.CloudBillingClient()
self.billing_table = f"{self.project_id}.billing_export.gcp_billing_export_v1_*"
def get_summary(self): def get_summary(self):
""" """
@ -29,6 +26,7 @@ class BillingAgent:
WHERE _PARTITIONTIME >= TIMESTAMP_SUB(CURRENT_TIMESTAMP(), INTERVAL 30 DAY) WHERE _PARTITIONTIME >= TIMESTAMP_SUB(CURRENT_TIMESTAMP(), INTERVAL 30 DAY)
GROUP BY 1 GROUP BY 1
ORDER BY total_cost DESC ORDER BY total_cost DESC
LIMIT 10
""" """
query_job = self.bq_client.query(query) query_job = self.bq_client.query(query)
results = query_job.result() results = query_job.result()
@ -88,24 +86,7 @@ class BillingAgent:
"total_monthly_forecast": total_forecast, "total_monthly_forecast": total_forecast,
} }
def get_credits(self):
"""
Retrieves active billing credits.
"""
billing_account_name = f"billingAccounts/{self.billing_account_id}"
billing_info = self.billing_client.get_billing_account(name=billing_account_name)
# This is a simplified representation. The actual credit information might
# be structured differently. This is a placeholder for the structure.
return {
"billing_account_name": billing_info.name,
"display_name": billing_info.display_name,
"open": billing_info.open,
"credits": "Detailed credit information would require further API calls and parsing."
}
if __name__ == '__main__': if __name__ == '__main__':
agent = BillingAgent() agent = BillingAgent()
print("Summary:", agent.get_summary()) print("Summary:", agent.get_summary())
print("Forecast:", agent.get_forecast()) print("Forecast:", agent.get_forecast())
print("Credits:", agent.get_credits())