Napredne Prompt Engineering Tehnike

"""
it_triage.py
Kompletan IT Problem Triage sistem sa Ollama i Lokalnim LLM-om.

Funkcionira kao: IT Helpdesk → AI Triage → JSON Report → Akcioni Plan
Kombinuje: Few-shot, CoT, System Prompts, Structured Output i Temperature kontrolu.
"""

import ollama
import json
import time
from datetime import datetime

# ================================================================
# KONFIGURACIJA
# ================================================================
MODEL = "llama3.2"

# ================================================================
# KORAK 1: SISTEM PROMPT ZA TRIAGE AGENTA
# Napisan na engleskom (efikasnost tokena), output na bosanskom
# ================================================================
TRIAGE_SYSTEM = """You are an expert IT incident triage agent for a government institution in Bosnia-Herzegovina.
You analyze IT problems and return structured JSON reports.

CLASSIFICATION RULES:
- Priority P1 (Critical): System down, security breach, data loss, >50 users affected
- Priority P2 (High): Significant degradation, <50 users affected, workaround available
- Priority P3 (Medium): Single user, non-critical system, workaround available
- Priority P4 (Low): Cosmetic issues, questions, enhancement requests

CATEGORIES: NETWORK, SERVER, APPLICATION, SECURITY, DATABASE, USER_DEVICE, GENERAL

ALWAYS respond with ONLY valid JSON, no other text.
Response language for descriptions: Bosnian (bosanski).

JSON Response Schema:
{
  "ticket_id": "auto-generated",
  "timestamp": "ISO 8601",
  "problem_summary": "Kratki sažetak na bosanskom (max 100 char)",
  "category": "CATEGORY_NAME",
  "priority": "P1|P2|P3|P4",
  "affected_users": "estimated count or 'single'",
  "immediate_actions": ["lista konkretnih koraka za odmah"],
  "diagnostic_commands": ["PowerShell/CMD komande za dijagnozu"],
  "escalation_required": true|false,
  "escalate_to": "null ili '[email protected]'",
  "estimated_resolution_time": "15min|1h|4h|1d|unknown",
  "similar_incidents_hint": "Savjet baziran na tipičnim uzrocima"
}"""

# ================================================================
# KORAK 2: FEW-SHOT PRIMJERI ZA TRIAGE
# ================================================================
FEW_SHOT_USER = """Analiziraj i klasificiraj IT problem koristeći few-shot primjere kao vodič:

PRIMJER 1:
Problem: "Korisnik ne može otvoriti Excel fajlove, dobija grešku 'File is locked'"
Odgovor:
{
  "ticket_id": "IT-2024-001",
  "timestamp": "2024-01-15T09:00:00",
  "problem_summary": "Korisnik ne može otvoriti Excel - file locked greška",
  "category": "APPLICATION",
  "priority": "P3",
  "affected_users": "single",
  "immediate_actions": ["Provjeriti da li neko drugi ima otvoren fajl", "Zatvoriti sve Excel instance: taskkill /IM excel.exe /F"],
  "diagnostic_commands": ["Get-Process -Name EXCEL", "net files"],
  "escalation_required": false,
  "escalate_to": null,
  "estimated_resolution_time": "15min",
  "similar_incidents_hint": "Najčešće uzrokovano neispravnim zatvaranjem Excel-a ili mrežnim prekidom tokom edita"
}

PRIMJER 2:
Problem: "Sve WEB stranice institucije su nedostupne, web server ne odgovara - baza korisnika 200+"
Odgovor:
{
  "ticket_id": "IT-2024-002",
  "timestamp": "2024-01-15T09:00:00",
  "problem_summary": "KRITIČNO: Web server nedostupan, 200+ korisnika pogođeno",
  "category": "SERVER",
  "priority": "P1",
  "affected_users": "200+",
  "immediate_actions": ["Odmah obavijestiti IT management", "Provjeriti status IIS servisa", "Pregledati Windows Event Log za CRITICAL greške"],
  "diagnostic_commands": ["Get-Service W3SVC", "Get-EventLog -LogName System -EntryType Error -Newest 20", "Test-NetConnection -ComputerName localhost -Port 80"],
  "escalation_required": true,
  "escalate_to": "[email protected]",
  "estimated_resolution_time": "1h",
  "similar_incidents_hint": "Provjeriti: disk space (iisreset može pasti ako je disk pun), App Pool crash, certifikat expired"
}

Sada klasificiraj sljedeći problem (vrati SAMO JSON):
Problem: "{problem}"
"""

# ================================================================
# KORAK 3: TRIAGE FUNKCIJA
# ================================================================
def triage_problem(opis_problema: str) -> dict:
    """
    Vrši AI triage IT problema.
    Vraća Python dict parsiran iz AI JSON odgovora.
    """
    print(f"\n🔍 Triaging problem: {opis_problema[:80]}...")

    # Ubacujemo problem u few-shot template
    user_prompt = FEW_SHOT_USER.format(problem=opis_problema)

    # Generišemo ticket ID
    ticket_id = f"IT-{datetime.now().strftime('%Y%m%d-%H%M%S')}"

    odgovor = ollama.chat(
        model=MODEL,
        messages=[
            {'role': 'system', 'content': TRIAGE_SYSTEM},
            {'role': 'user', 'content': user_prompt}
        ],
        options={
            # Niska temperatura za konzistentne, faktične klasifikacije
            'temperature': 0.1,
            'num_predict': 600,
        }
    )

    raw_text = odgovor['message']['content'].strip()

    # Parsiranje JSON-a (čišćenje eventualnih markdown code fences)
    try:
        cisti = raw_text
        if '```' in cisti:
            dijelovi = cisti.split('```')
            for d in dijelovi:
                if d.startswith('json'):
                    cisti = d[4:]
                    break
                elif '{' in d:
                    cisti = d
                    break

        parsed = json.loads(cisti)
        # Dopunjujemo ticket_id sa generisanim
        parsed['ticket_id'] = ticket_id
        parsed['timestamp'] = datetime.now().isoformat()
        return parsed

    except json.JSONDecodeError:
        # Fallback ako JSON parsiranje ne uspije
        print(f"   ⚠️ JSON parsiranje nije uspjelo. Raw odgovor:")
        print(f"   {raw_text[:200]}")
        return {"error": "JSON parse failed", "raw": raw_text, "ticket_id": ticket_id}

# ================================================================
# KORAK 4: AKCIONI PLAN GENERATOR
# ================================================================
def generiraj_akcioni_plan(triage_result: dict) -> str:
    """
    Na osnovu triage JSON-a, generiše detaljni akcioni plan koristeći CoT.
    """
    if 'error' in triage_result:
        return "Greška u triage-u — nije moguće generisati plan."

    print(f"\n📋 Generišem akcioni plan za {triage_result.get('priority', 'P?')} incident...")

    cot_prompt = f"""Na osnovu triage analize IT incidenta, generiši detaljni akcioni plan.

TRIAGE PODACI:
- Problem: {triage_result.get('problem_summary', 'N/A')}
- Prioritet: {triage_result.get('priority', 'N/A')}
- Kategorija: {triage_result.get('category', 'N/A')}
- Pogođenih korisnika: {triage_result.get('affected_users', 'N/A')}
- Potrebna eskalacija: {triage_result.get('escalation_required', False)}
- Procijenjeno rješavanje: {triage_result.get('estimated_resolution_time', 'N/A')}

Razmisli korak po korak i generiši:
1. IMMEDIATE (sljedećih 15 minuta): Šta odmah uraditi?
2. SHORT-TERM (sljedećih 2 sata): Dijagnoza i mitigation
3. RESOLUTION: Permanentno rješenje
4. COMMUNICATION: Koga obavijestiti i kada?
5. PREVENTION: Kako spriječiti ponavljanje?

Budi konkretan — navedi specifične komande, kontakte i procedure naše institucije.
Odgovaraj na bosanskom jeziku, formalan ton."""

    odgovor = ollama.chat(
        model=MODEL,
        messages=[{'role': 'user', 'content': cot_prompt}],
        options={'temperature': 0.4, 'num_predict': 700}
    )
    return odgovor['message']['content']

# ================================================================
# KORAK 5: POKRETANJE DEMONSTRACIJE
# ================================================================
def ispisi_triage_report(triage: dict, plan: str, problem: str):
    """Formatirani ispis triage reporta."""
    print("\n" + "="*65)
    print(f"🎫 TRIAGE REPORT — {triage.get('ticket_id', 'N/A')}")
    print("="*65)
    print(f"📋 Problem: {problem}")
    print(f"🏷️  Sažetak: {triage.get('problem_summary', 'N/A')}")
    print(f"🚨 Prioritet: {triage.get('priority', 'N/A')} | Kategorija: {triage.get('category', 'N/A')}")
    print(f"👥 Pogođeni korisnici: {triage.get('affected_users', 'N/A')}")
    print(f"⏱️  Procijenjeno: {triage.get('estimated_resolution_time', 'N/A')}")
    print(f"📧 Eskalacija: {'✅ POTREBNA → ' + str(triage.get('escalate_to', '')) if triage.get('escalation_required') else '❌ Nije potrebna'}")

    if triage.get('immediate_actions'):
        print(f"\n⚡ ODMAH URADITI:")
        for i, akcija in enumerate(triage['immediate_actions'], 1):
            print(f"   {i}. {akcija}")

    if triage.get('diagnostic_commands'):
        print(f"\n🔧 DIJAGNOSTIČKE KOMANDE:")
        for cmd in triage['diagnostic_commands']:
            print(f"   $ {cmd}")

    print(f"\n💡 Savjet: {triage.get('similar_incidents_hint', 'N/A')}")

    print(f"\n{'─'*65}")
    print(f"📄 DETALJNI AKCIONI PLAN:")
    print(f"{'─'*65}")
    print(plan)
    print("="*65)

if __name__ == "__main__":
    print("="*65)
    print("🤖 IT TRIAGE SISTEM — Powered by Ollama + Llama 3.2")
    print("="*65)

    # Lista test problema
    test_problemi = [
        "Korisnik account-a Ana Kovač ne može se prijaviti na Windows domensku mrežu od jutros, a ostali na istom odjelu rade normalno.",
        "Firewall je blokirao sav internet promet nakon what je admin greškom obrisao permit all pravilo - svi korisnici institucije pogođeni."
    ]

    for i, problem in enumerate(test_problemi, 1):
        print(f"\n\n🔄 TEST {i}/{len(test_problemi)}")
        print(f"📝 Problem: {problem}")

        # Triage
        triage = triage_problem(problem)

        # Akcioni plan
        plan = generiraj_akcioni_plan(triage)

        # Prikaz
        ispisi_triage_report(triage, plan, problem)

        if i < len(test_problemi):
            print("\nNastavljam sa sljedećim problemom...")
            time.sleep(2)

    print("\n\n✅ IT Triage sistem demonstracija završena!")
    print("Ovaj sistem možete integrirati sa JIRA, email ili Slack za produkcijsku upotrebu.")

cd C:\Users\$env:USERNAME\Desktop\AI_Kurs
.\venv\Scripts\Activate.ps1
python it_triage.py

# Napomena: Sistem procesira svaki problem 2x (triage + akcioni plan)
# Ukupno trajanje: 5-15 minuta za 2 test problema

# Eksperiment: Poredite isti prompt sa različitim temperaturama
# Savjet: Kopirajte benchmark_temperature.py i modificirajte:

# Temperatura 0.0 → Uvijek isti JSON (deterministički)
# Temperatura 0.5 → Malo različit tekst, isti JSON format
# Temperatura 1.0 → Veća varijabilnost u tekstu i potencijalne JSON greške

# Zaključak koji ćete vidjeti:
# Za JSON structured output: UVIJEK koristite temperature ≤ 0.2
# Za narativne opise/planove: 0.3-0.6 daje bogatiji tekst
# Za kreativne zadatke: 0.7-1.0