import json
from collections.abc import Generator
from typing import Any

from llama_index.core.llms import CompletionResponse, CustomLLM, LLMMetadata
from pydantic import BaseModel, Field


class TicketSummary(BaseModel):
    team: str = Field(description="Team that should handle the ticket")
    priority: str = Field(description="Routing priority")
    next_action: str = Field(description="Next action for the team")