pi-farm/PiDoBot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

main.py aktualisiert

Browse Source
This commit is contained in:
info@pi-farm.de
2026-03-05 11:19:31 +00:00
parent 88a0d1a734
commit 60d36d0f4b
1 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
main.py
View File

@@ -26,7 +26,13 @@ templates = Jinja2Templates(directory="templates")
SSH_KEY = os.path.expanduser("~/.ssh/id_rsa") SSH_KEY = os.path.expanduser("~/.ssh/id_rsa")
DB_PATH = "cluster.db" DB_PATH = "cluster.db"
chat_history = [] 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) --- # --- KI KONFIGURATION (Werte aus .env laden) ---
AI_PROVIDER = os.getenv("AI_PROVIDER", "google").lower() AI_PROVIDER = os.getenv("AI_PROVIDER", "google").lower()
@@ -48,8 +54,9 @@ def get_system_prompt():
for n in nodes: for n in nodes:
node_info += f"- Name: {n['name']}, IP: {n['ip']}, User: {n['user']}\n" node_info += f"- Name: {n['name']}, IP: {n['ip']}, User: {n['user']}\n"
# Ersetzt den Platzhalter im Template mit den echten Node-Daten # Falls RAW_PROMPT leer ist, nutzen wir einen Sicherheits-Fallback
return SYSTEM_PROMPT_TEMPLATE.replace("{node_info}", node_info) 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 --- # --- KI FUNKTIONEN ---