vivekvish2004's picture
fix: add /health, /metadata, /schema, /mcp endpoints + per-task graders
62fbd09
from pydantic import BaseModel
from typing import Any, Optional, Dict
class Action(BaseModel):
action_type: str
payload: Dict[str, Any]
class Observation(BaseModel):
state: Dict[str, Any]
info: Optional[Dict[str, Any]] = None
class Reward(BaseModel):
value: float
is_terminal: bool
# --- AI Configuration & Prompts ---
SYSTEM_PROMPT = """
You are an Enterprise AI Customer Support agent resolving a ticket pipeline.
For each ticket, you must:
1. classify_ticket: {"classification": "refund" | "general_inquiry" | "login_issue" | "feedback" | "technical_issue"}
2. assign_priority: {"priority": "low" | "medium" | "high"}
3. generate_response: {"response": "<empathetic_text>"}
4. resolve: {}
Your goal is to process the ticket efficiently and move to the next one in the queue.
You MUST return ONLY a fully valid JSON object:
{"action_type": "<name>", "payload": {...}}
""".strip()
DEFAULT_MODEL = "meta-llama/Meta-Llama-3-8B-Instruct"
DEFAULT_API_BASE = "https://router.huggingface.co/v1"