Spaces:
Build error
Build error
| import uuid | |
| import yaml | |
| from agentctl.agent import K8sAgent | |
| from agentctl.config import settings | |
| from agentctl.llm import ask_llm, HF_MODEL # <--- import model name | |
| def randomize_name(yaml_text: str) -> str: | |
| try: | |
| obj = yaml.safe_load(yaml_text) | |
| if not isinstance(obj, dict): | |
| return yaml_text | |
| kind = obj.get("kind") | |
| if kind not in ("Job", "Deployment", "CronJob"): | |
| return yaml_text | |
| metadata = obj.setdefault("metadata", {}) | |
| name = metadata.get("name") | |
| if not name: | |
| return yaml_text | |
| suffix = str(uuid.uuid4())[:5] | |
| metadata["name"] = f"{name}-{suffix}" | |
| return yaml.dump(obj, sort_keys=False) | |
| except Exception: | |
| return yaml_text | |
| class AgentService: | |
| def __init__(self): | |
| self.agent = K8sAgent() | |
| async def generate_yaml(self, prompt: str, namespace: str | None, kind: str | None): | |
| ns = namespace or settings.k8s_namespace | |
| kind_sel = None if (kind is None or kind == "Auto") else kind | |
| # ------------------------------- | |
| # 1) LLM PLAN STEP | |
| # ------------------------------- | |
| plan_prompt = f""" | |
| Convert this plan into valid Kubernetes YAML. | |
| HARD RULES: | |
| - kind: Job | |
| - image: busybox or python:3.10-slim only | |
| - restartPolicy: Never | |
| - No GPUs, no ML, no training. | |
| - Commands MUST be simple (echo/sleep/ls) | |
| - No markdown | |
| - No commentary | |
| Plan: | |
| {prompt} | |
| Output: | |
| - workload type | |
| - image | |
| - command | |
| - args | |
| """ | |
| try: | |
| plan = await ask_llm(plan_prompt) | |
| except Exception as e: | |
| print(f"[LLM ERROR: plan] Fallback. Error: {e}") | |
| _, yaml_text = self.agent.nl_to_resource_yaml( | |
| prompt, namespace=ns, kind=kind_sel | |
| ) | |
| return { | |
| "yaml": randomize_name(yaml_text), | |
| "mode": "fallback", | |
| "model": None | |
| } | |
| # ------------------------------- | |
| # 2) LLM YAML STEP | |
| # ------------------------------- | |
| yaml_prompt = f""" | |
| Convert this Kubernetes plan into valid Kubernetes YAML. | |
| Output ONLY YAML: | |
| {plan} | |
| """ | |
| try: | |
| yaml_text = await ask_llm(yaml_prompt) | |
| if not yaml_text or yaml_text.strip() == "": | |
| raise ValueError("Empty YAML from LLM") | |
| return { | |
| "yaml": randomize_name(yaml_text), | |
| "mode": "llm", | |
| "model": HF_MODEL | |
| } | |
| except Exception as e: | |
| print(f"[LLM ERROR: yaml] Fallback. Error: {e}") | |
| _, yaml_text = self.agent.nl_to_resource_yaml( | |
| prompt, namespace=ns, kind=kind_sel | |
| ) | |
| return { | |
| "yaml": randomize_name(yaml_text), | |
| "mode": "fallback", | |
| "model": None | |
| } | |
| agent_service = AgentService() | |