mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-05-20 04:45:12 +02:00
01aeede78b
* Tech spec * Python CLI utilities updated to use the API including streaming features * Added type safety to Python API * Completed missing auth token support in CLI
27 lines
729 B
Python
27 lines
729 B
Python
|
|
import requests
|
|
from typing import Optional, Dict
|
|
|
|
|
|
class Metrics:
|
|
"""Synchronous metrics client"""
|
|
|
|
def __init__(self, url: str, timeout: int, token: Optional[str]) -> None:
|
|
self.url: str = url
|
|
self.timeout: int = timeout
|
|
self.token: Optional[str] = token
|
|
|
|
def get(self) -> str:
|
|
"""Get Prometheus metrics as text"""
|
|
url: str = f"{self.url}/api/metrics"
|
|
|
|
headers: Dict[str, str] = {}
|
|
if self.token:
|
|
headers["Authorization"] = f"Bearer {self.token}"
|
|
|
|
resp = requests.get(url, timeout=self.timeout, headers=headers)
|
|
|
|
if resp.status_code != 200:
|
|
raise Exception(f"Status code {resp.status_code}")
|
|
|
|
return resp.text
|