main.py aktualisiert

This commit is contained in:
2026-03-05 11:19:31 +00:00
parent 88a0d1a734
commit 60d36d0f4b

13
main.py
View File

@@ -26,7 +26,13 @@ templates = Jinja2Templates(directory="templates")
SSH_KEY = os.path.expanduser("~/.ssh/id_rsa")
DB_PATH = "cluster.db"
chat_history = []
SYSTEM_PROMPT_TEMPLATE = os.getenv("SYSTEM_PROMPT", "Du bist ein Helfer. Nodes: {node_info}")
# Debug: Teste beim Start, ob der Prompt geladen wurde
RAW_PROMPT = os.getenv("SYSTEM_PROMPT")
if not RAW_PROMPT:
print("❌ FEHLER: SYSTEM_PROMPT konnte nicht aus .env geladen werden!")
else:
print("✅ SYSTEM_PROMPT erfolgreich geladen.")
# --- KI KONFIGURATION (Werte aus .env laden) ---
AI_PROVIDER = os.getenv("AI_PROVIDER", "google").lower()
@@ -48,8 +54,9 @@ def get_system_prompt():
for n in nodes:
node_info += f"- Name: {n['name']}, IP: {n['ip']}, User: {n['user']}\n"
# Ersetzt den Platzhalter im Template mit den echten Node-Daten
return SYSTEM_PROMPT_TEMPLATE.replace("{node_info}", node_info)
# Falls RAW_PROMPT leer ist, nutzen wir einen Sicherheits-Fallback
template = RAW_PROMPT if RAW_PROMPT else "Du bist ein Assistent. Nodes: {node_info}. Nutze <EXECUTE target=\"IP\">cmd</EXECUTE>"
return template.replace("{node_info}", node_info)
# --- KI FUNKTIONEN ---