terghalin/metalcheck
1
0
Fork 0
mirror of https://ak-git.vectorsigma.ru/terghalin/metalcheck.git synced 2026-03-24 04:51:08 +09:00
Code Issues Actions Packages Projects Releases 1 Wiki Activity

fix(think): correctly get list

Browse Source
This commit is contained in:
Aleksandr Tcitlionok
2024-12-05 08:33:48 +00:00
parent f0617da839
commit f0c1458cde
1 changed files with 4 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
app/routes/think.py
View File

@@ -1,40 +1,3 @@
from fastapi import APIRouter, HTTPException
from kubernetes import client, config
from typing import List, Dict
import requests
import os
router = APIRouter()
def fetch_ai_summary(cluster_name: str, nodes: List[Dict], api_url: str, auth_token: str) -> str:
"""
Sends node data to an external AI API endpoint and retrieves a summary response.
"""
payload = {
"data": {
"cluster_name": cluster_name,
"nodes": nodes
}
}
headers = {
"Authorization": f"Bearer {auth_token}",
"Content-Type": "application/json"
}
try:
response = requests.post(api_url, json=payload, headers=headers)
if response.status_code == 200:
result = response.json()
if result.get("success"):
return result.get("result", "No result provided by the API.")
else:
return "API responded with success=false."
else:
return f"API request failed with status code {response.status_code}: {response.text}"
except requests.RequestException as e:
return f"An error occurred while contacting the API: {str(e)}"
@router.get("/think/k8s")
def think_k8s():
"""
@@ -58,15 +21,15 @@ def think_k8s():
"memory": f"{round(int(node.status.capacity.get('memory').replace('Ki', '')) / 1024 / 1024, 2)} GB",
"storage": f"{round(int(ephemeral_storage.replace('Ki', '')) / (1024 ** 2), 2)} GB",
"type": instance_type,
"namespaces": [ns.metadata.name for ns in v1.list_namespace().items()]
"namespaces": [ns.metadata.name for ns in v1.list_namespace().items]
})
# Fetch token from environment
# Fetch environment variables
api_url = os.getenv("AI_API_URL")
auth_token = os.getenv("AI_API_TOKEN")
if not auth_token:
raise HTTPException(status_code=500, detail="AI API token is not set. Please set the AI_API_TOKEN environment variable.")
if not api_url or not auth_token:
raise HTTPException(status_code=500, detail="AI API URL or token is not set. Please set AI_API_URL and AI_API_TOKEN environment variables.")
# Call AI API
summary = fetch_ai_summary(cluster_name, nodes, api_url, auth_token)