چت‌بات‌ها و ردیابی حالت گفت‌وگو (DST)

SLOT_EXTRACTORS = {"cuisine": extract_cuisine, "area": extract_area,
                   "price": extract_price, "people": extract_people}

def update_state(state, utterance):
    new_state = dict(state)          # هرگز جایگاه‌های دست‌نخورده را ریست نکنید
    for slot, extractor in SLOT_EXTRACTORS.items():
        value = extractor(utterance)
        if value is not None:
            new_state[slot] = value  # یک اصلاح جایگزین مقادیر قبلی می‌شود
            continue
        if is_negation(utterance, slot):
            new_state[slot] = None   # «نوع غذا مهم نیست»
    return new_state

class RestaurantState(BaseModel):
    cuisine: Optional[Literal["italian", "chinese", "indian", "thai", "any"]] = None
    area: Optional[Literal["north", "south", "east", "west", "center"]] = None
    price: Optional[Literal["cheap", "moderate", "expensive"]] = None
    people: Optional[int] = None
    day: Optional[str] = None                           # جایگاه آزاد (free-form)

def llm_dst(history, llm):
    prompt = f"""...{render(history)}...Output only the JSON state."""
    return llm(prompt, response_model=RestaurantState)  # کل حالت در هر نوبت گفت‌وگو