21 lines
628 B
Python
21 lines
628 B
Python
from __future__ import annotations
|
|
|
|
from typing import Literal
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class DemoRuntimeConfig(BaseModel):
|
|
matcher_pipeline: Literal["classifier"]
|
|
classifier_backend: Literal["mock", "bert", "remote", "joint_bert"]
|
|
session_backend: Literal["memory", "redis"]
|
|
slot_extractor_backend: str
|
|
planner_backend: str
|
|
planner_model_name: str
|
|
|
|
|
|
class DemoRuntimeUpdateRequest(BaseModel):
|
|
matcher_pipeline: Literal["classifier"] = Field(default="classifier")
|
|
classifier_backend: Literal["mock", "bert", "remote", "joint_bert"]
|
|
session_backend: Literal["memory", "redis"]
|