fix(think): correctly get list
This commit is contained in:
@@ -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")
|
@router.get("/think/k8s")
|
||||||
def 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",
|
"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",
|
"storage": f"{round(int(ephemeral_storage.replace('Ki', '')) / (1024 ** 2), 2)} GB",
|
||||||
"type": instance_type,
|
"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")
|
api_url = os.getenv("AI_API_URL")
|
||||||
auth_token = os.getenv("AI_API_TOKEN")
|
auth_token = os.getenv("AI_API_TOKEN")
|
||||||
|
|
||||||
if not auth_token:
|
if not api_url or not auth_token:
|
||||||
raise HTTPException(status_code=500, detail="AI API token is not set. Please set the AI_API_TOKEN environment variable.")
|
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
|
# Call AI API
|
||||||
summary = fetch_ai_summary(cluster_name, nodes, api_url, auth_token)
|
summary = fetch_ai_summary(cluster_name, nodes, api_url, auth_token)
|
||||||
|
|||||||
Reference in New Issue
Block a user