qanswer-sdk 0.1.0.dev0__py3-none-any.whl

qanswer_sdk/models/dataset_default_values_model.py

@@ -0,0 +1,129 @@
+# coding: utf-8
+
+"""
+    free-text-client
+
+    QAnswer 2.0 API
+
+    The version of the OpenAPI document: 2.0.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictFloat, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional, Union
+from typing_extensions import Annotated
+from qanswer_sdk.models.data_storage_preference import DataStoragePreference
+from typing import Optional, Set
+from typing_extensions import Self
+
+class DatasetDefaultValuesModel(BaseModel):
+    """
+    DatasetDefaultValuesModel
+    """ # noqa: E501
+    language: Optional[StrictStr] = Field(default=None, description="Language preference")
+    split_by: Optional[StrictStr] = Field(default=None, description="Split method")
+    split_length: Optional[Annotated[int, Field(strict=True, ge=1)]] = Field(default=None, description="Split length")
+    split_overlap: Optional[Annotated[int, Field(strict=True, ge=0)]] = Field(default=None, description="Split overlap")
+    bot_name: Optional[StrictStr] = Field(default=None, description="Bot name")
+    bot_description: Optional[StrictStr] = Field(default=None, description="Bot description")
+    bot_mode: Optional[StrictStr] = Field(default=None, description="Bot mode")
+    bot_knowledge: Optional[StrictStr] = Field(default=None, description="Bot knowledge settings")
+    bot_answer_length: Optional[StrictStr] = Field(default=None, description="Bot answer length")
+    bot_temperature: Optional[Union[Annotated[float, Field(le=2.0, strict=True, ge=0.0)], Annotated[int, Field(le=2, strict=True, ge=0)]]] = Field(default=None, description="Temperature for bot responses")
+    bot_seed: Optional[StrictInt] = Field(default=None, description="Seed for bot randomness")
+    prompt: Optional[StrictStr] = Field(default=None, description="Prompt to give the bot")
+    number_of_references: Optional[Annotated[int, Field(strict=True, ge=1)]] = Field(default=None, description="Number of references used in answering questions")
+    data_storage_preference: Optional[DataStoragePreference] = None
+    stream_speed: Optional[Union[StrictFloat, StrictInt]] = Field(default=0.0, description="Stream speed")
+    top_k_retriever: Optional[Annotated[int, Field(strict=True, ge=1)]] = Field(default=128, description="Number of retriever results")
+    weight_title: Optional[Union[StrictFloat, StrictInt]] = Field(default=1.0, description="Weight of title in ranking")
+    weight_text: Optional[Union[StrictFloat, StrictInt]] = Field(default=1.0, description="Weight of text in ranking")
+    track_geo: Optional[StrictBool] = Field(default=None, description="Flag to track geo location")
+    use_training: Optional[StrictBool] = Field(default=True, description="Flag to use training data")
+    embedding_model: Optional[StrictStr] = Field(default=None, description="Embedding model to use")
+    __properties: ClassVar[List[str]] = ["language", "split_by", "split_length", "split_overlap", "bot_name", "bot_description", "bot_mode", "bot_knowledge", "bot_answer_length", "bot_temperature", "bot_seed", "prompt", "number_of_references", "data_storage_preference", "stream_speed", "top_k_retriever", "weight_title", "weight_text", "track_geo", "use_training", "embedding_model"]
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of DatasetDefaultValuesModel from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of DatasetDefaultValuesModel from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "language": obj.get("language"),
+            "split_by": obj.get("split_by"),
+            "split_length": obj.get("split_length"),
+            "split_overlap": obj.get("split_overlap"),
+            "bot_name": obj.get("bot_name"),
+            "bot_description": obj.get("bot_description"),
+            "bot_mode": obj.get("bot_mode"),
+            "bot_knowledge": obj.get("bot_knowledge"),
+            "bot_answer_length": obj.get("bot_answer_length"),
+            "bot_temperature": obj.get("bot_temperature"),
+            "bot_seed": obj.get("bot_seed"),
+            "prompt": obj.get("prompt"),
+            "number_of_references": obj.get("number_of_references"),
+            "data_storage_preference": obj.get("data_storage_preference"),
+            "stream_speed": obj.get("stream_speed") if obj.get("stream_speed") is not None else 0.0,
+            "top_k_retriever": obj.get("top_k_retriever") if obj.get("top_k_retriever") is not None else 128,
+            "weight_title": obj.get("weight_title") if obj.get("weight_title") is not None else 1.0,
+            "weight_text": obj.get("weight_text") if obj.get("weight_text") is not None else 1.0,
+            "track_geo": obj.get("track_geo"),
+            "use_training": obj.get("use_training") if obj.get("use_training") is not None else True,
+            "embedding_model": obj.get("embedding_model")
+        })
+        return _obj
+
+

qanswer_sdk/models/dataset_detail.py

@@ -0,0 +1,128 @@
+# coding: utf-8
+
+"""
+    free-text-client
+
+    QAnswer 2.0 API
+
+    The version of the OpenAPI document: 2.0.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictInt, StrictStr, field_validator
+from typing import Any, ClassVar, Dict, List, Optional
+from qanswer_sdk.models.access_type_shared import AccessTypeShared
+from qanswer_sdk.models.dataset_detail_kg import DatasetDetailKg
+from qanswer_sdk.models.dataset_schema import DatasetSchema
+from typing import Optional, Set
+from typing_extensions import Self
+
+class DatasetDetail(BaseModel):
+    """
+    DatasetDetail
+    """ # noqa: E501
+    id: Optional[StrictInt] = None
+    dataset: Optional[StrictStr] = None
+    username: Optional[StrictStr] = None
+    name: Optional[StrictStr] = None
+    description: Optional[StrictStr] = None
+    access_type_open: Optional[StrictStr] = Field(default=None, alias="accessTypeOpen")
+    access_type_shared: Optional[AccessTypeShared] = Field(default=None, alias="accessTypeShared")
+    free_text_details: Optional[DatasetSchema] = Field(default=None, alias="freeTextDetails")
+    kg_details: Optional[DatasetDetailKg] = Field(default=None, alias="kgDetails")
+    question_quota_available: Optional[StrictBool] = Field(default=None, alias="questionQuotaAvailable")
+    can_remove_branding: Optional[StrictBool] = Field(default=None, alias="canRemoveBranding")
+    favorite: Optional[StrictBool] = None
+    __properties: ClassVar[List[str]] = ["id", "dataset", "username", "name", "description", "accessTypeOpen", "accessTypeShared", "freeTextDetails", "kgDetails", "questionQuotaAvailable", "canRemoveBranding", "favorite"]
+
+    @field_validator('access_type_open')
+    def access_type_open_validate_enum(cls, value):
+        """Validates the enum"""
+        if value is None:
+            return value
+
+        if value not in set(['LoggedIn', 'Open', 'None']):
+            raise ValueError("must be one of enum values ('LoggedIn', 'Open', 'None')")
+        return value
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of DatasetDetail from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of free_text_details
+        if self.free_text_details:
+            _dict['freeTextDetails'] = self.free_text_details.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of kg_details
+        if self.kg_details:
+            _dict['kgDetails'] = self.kg_details.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of DatasetDetail from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "id": obj.get("id"),
+            "dataset": obj.get("dataset"),
+            "username": obj.get("username"),
+            "name": obj.get("name"),
+            "description": obj.get("description"),
+            "accessTypeOpen": obj.get("accessTypeOpen"),
+            "accessTypeShared": obj.get("accessTypeShared"),
+            "freeTextDetails": DatasetSchema.from_dict(obj["freeTextDetails"]) if obj.get("freeTextDetails") is not None else None,
+            "kgDetails": DatasetDetailKg.from_dict(obj["kgDetails"]) if obj.get("kgDetails") is not None else None,
+            "questionQuotaAvailable": obj.get("questionQuotaAvailable"),
+            "canRemoveBranding": obj.get("canRemoveBranding"),
+            "favorite": obj.get("favorite")
+        })
+        return _obj
+
+

qanswer_sdk/models/dataset_detail_kg.py

@@ -0,0 +1,88 @@
+# coding: utf-8
+
+"""
+    free-text-client
+
+    QAnswer 2.0 API
+
+    The version of the OpenAPI document: 2.0.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, Field
+from typing import Any, ClassVar, Dict, List, Optional
+from qanswer_sdk.models.assistant_task import AssistantTask
+from typing import Optional, Set
+from typing_extensions import Self
+
+class DatasetDetailKg(BaseModel):
+    """
+    DatasetDetailKg
+    """ # noqa: E501
+    assistant_task: Optional[AssistantTask] = Field(default=None, alias="assistantTask")
+    __properties: ClassVar[List[str]] = ["assistantTask"]
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of DatasetDetailKg from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of DatasetDetailKg from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "assistantTask": obj.get("assistantTask")
+        })
+        return _obj
+
+

qanswer_sdk/models/dataset_schema.py

@@ -0,0 +1,193 @@
+# coding: utf-8
+
+"""
+    free-text-client
+
+    QAnswer 2.0 API
+
+    The version of the OpenAPI document: 2.0.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from datetime import datetime
+from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictFloat, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional, Union
+from typing_extensions import Annotated
+from uuid import UUID
+from qanswer_sdk.models.assistant_category_color_item import AssistantCategoryColorItem
+from qanswer_sdk.models.assistant_task import AssistantTask
+from qanswer_sdk.models.data_storage_preference import DataStoragePreference
+from qanswer_sdk.models.dataset_type import DatasetType
+from qanswer_sdk.models.extraction_strategy import ExtractionStrategy
+from qanswer_sdk.models.logs_visibility_policy import LogsVisibilityPolicy
+from qanswer_sdk.models.widget_configs import WidgetConfigs
+from typing import Optional, Set
+from typing_extensions import Self
+
+class DatasetSchema(BaseModel):
+    """
+    DatasetSchema
+    """ # noqa: E501
+    id: StrictInt
+    dataset: StrictStr
+    username: StrictStr
+    assistant_id: Optional[UUID] = None
+    index_hash: Optional[StrictStr] = None
+    language: Optional[StrictStr] = None
+    teams_app_id: Optional[StrictStr] = Field(default=None, alias="teamsAppId")
+    teams_bot_lookup_key: Optional[StrictStr] = Field(default=None, alias="teamsBotLookupKey")
+    is_updating: Optional[StrictBool] = None
+    top_k_retriever: Optional[Annotated[int, Field(strict=True, ge=1)]] = None
+    top_k_ranker: Optional[Annotated[int, Field(strict=True, ge=1)]] = None
+    weight_title: Optional[Union[StrictFloat, StrictInt]] = None
+    weight_text: Optional[Union[StrictFloat, StrictInt]] = None
+    split_by: Optional[StrictStr] = None
+    split_len: Optional[Annotated[int, Field(strict=True, ge=0)]] = 512
+    split_overlap: Optional[Annotated[int, Field(strict=True, ge=0)]] = 32
+    bot_name: Optional[StrictStr] = None
+    bot_description: Optional[StrictStr] = None
+    bot_mode: Optional[StrictStr] = None
+    bot_knowledge: Optional[StrictStr] = None
+    bot_answer_length: Optional[StrictStr] = None
+    bot_seed: Optional[StrictInt] = None
+    bot_temperature: Optional[Union[StrictFloat, StrictInt]] = None
+    prompt: Optional[StrictStr] = None
+    initial_message: Optional[StrictStr] = None
+    track_geo: Optional[StrictBool] = None
+    creation_date: Optional[datetime] = None
+    use_training: Optional[StrictBool] = None
+    embedding_model: Optional[StrictStr] = None
+    reindexing_failed: Optional[StrictBool] = False
+    reindexing_in_progress: Optional[StrictBool] = False
+    data_storage_preference: Optional[DataStoragePreference] = None
+    logs_visibility_policy: Optional[LogsVisibilityPolicy] = None
+    dataset_type: Optional[DatasetType] = None
+    categories: Optional[List[StrictStr]] = None
+    is_template: Optional[StrictBool] = False
+    incognito: Optional[StrictBool] = False
+    task: Optional[AssistantTask] = None
+    charge_owner: Optional[StrictBool] = None
+    extraction_strategy: Optional[ExtractionStrategy] = Field(default=None, description="Override file extraction routing. When unset, the indexing pipeline auto-selects based on file type (docling / readables / vanilla).")
+    categories_with_color: Optional[List[AssistantCategoryColorItem]] = None
+    widget_configs: Optional[WidgetConfigs] = None
+    n_tokens: Optional[StrictInt] = None
+    embedding_model_id: Optional[StrictInt] = -1
+    __properties: ClassVar[List[str]] = ["id", "dataset", "username", "assistant_id", "index_hash", "language", "teamsAppId", "teamsBotLookupKey", "is_updating", "top_k_retriever", "top_k_ranker", "weight_title", "weight_text", "split_by", "split_len", "split_overlap", "bot_name", "bot_description", "bot_mode", "bot_knowledge", "bot_answer_length", "bot_seed", "bot_temperature", "prompt", "initial_message", "track_geo", "creation_date", "use_training", "embedding_model", "reindexing_failed", "reindexing_in_progress", "data_storage_preference", "logs_visibility_policy", "dataset_type", "categories", "is_template", "incognito", "task", "charge_owner", "extraction_strategy", "categories_with_color", "widget_configs", "n_tokens", "embedding_model_id"]
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of DatasetSchema from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of each item in categories_with_color (list)
+        _items = []
+        if self.categories_with_color:
+            for _item_categories_with_color in self.categories_with_color:
+                if _item_categories_with_color:
+                    _items.append(_item_categories_with_color.to_dict())
+            _dict['categories_with_color'] = _items
+        # override the default output from pydantic by calling `to_dict()` of widget_configs
+        if self.widget_configs:
+            _dict['widget_configs'] = self.widget_configs.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of DatasetSchema from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "id": obj.get("id"),
+            "dataset": obj.get("dataset"),
+            "username": obj.get("username"),
+            "assistant_id": obj.get("assistant_id"),
+            "index_hash": obj.get("index_hash"),
+            "language": obj.get("language"),
+            "teamsAppId": obj.get("teamsAppId"),
+            "teamsBotLookupKey": obj.get("teamsBotLookupKey"),
+            "is_updating": obj.get("is_updating"),
+            "top_k_retriever": obj.get("top_k_retriever"),
+            "top_k_ranker": obj.get("top_k_ranker"),
+            "weight_title": obj.get("weight_title"),
+            "weight_text": obj.get("weight_text"),
+            "split_by": obj.get("split_by"),
+            "split_len": obj.get("split_len") if obj.get("split_len") is not None else 512,
+            "split_overlap": obj.get("split_overlap") if obj.get("split_overlap") is not None else 32,
+            "bot_name": obj.get("bot_name"),
+            "bot_description": obj.get("bot_description"),
+            "bot_mode": obj.get("bot_mode"),
+            "bot_knowledge": obj.get("bot_knowledge"),
+            "bot_answer_length": obj.get("bot_answer_length"),
+            "bot_seed": obj.get("bot_seed"),
+            "bot_temperature": obj.get("bot_temperature"),
+            "prompt": obj.get("prompt"),
+            "initial_message": obj.get("initial_message"),
+            "track_geo": obj.get("track_geo"),
+            "creation_date": obj.get("creation_date"),
+            "use_training": obj.get("use_training"),
+            "embedding_model": obj.get("embedding_model"),
+            "reindexing_failed": obj.get("reindexing_failed") if obj.get("reindexing_failed") is not None else False,
+            "reindexing_in_progress": obj.get("reindexing_in_progress") if obj.get("reindexing_in_progress") is not None else False,
+            "data_storage_preference": obj.get("data_storage_preference"),
+            "logs_visibility_policy": obj.get("logs_visibility_policy"),
+            "dataset_type": obj.get("dataset_type"),
+            "categories": obj.get("categories"),
+            "is_template": obj.get("is_template") if obj.get("is_template") is not None else False,
+            "incognito": obj.get("incognito") if obj.get("incognito") is not None else False,
+            "task": obj.get("task"),
+            "charge_owner": obj.get("charge_owner"),
+            "extraction_strategy": obj.get("extraction_strategy"),
+            "categories_with_color": [AssistantCategoryColorItem.from_dict(_item) for _item in obj["categories_with_color"]] if obj.get("categories_with_color") is not None else None,
+            "widget_configs": WidgetConfigs.from_dict(obj["widget_configs"]) if obj.get("widget_configs") is not None else None,
+            "n_tokens": obj.get("n_tokens"),
+            "embedding_model_id": obj.get("embedding_model_id") if obj.get("embedding_model_id") is not None else -1
+        })
+        return _obj
+
+

qanswer_sdk/models/dataset_type.py

@@ -0,0 +1,38 @@
+# coding: utf-8
+
+"""
+    free-text-client
+
+    QAnswer 2.0 API
+
+    The version of the OpenAPI document: 2.0.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import json
+from enum import Enum
+from typing_extensions import Self
+
+
+class DatasetType(str, Enum):
+    """
+    DatasetType
+    """
+
+    """
+    allowed enum values
+    """
+    DEFAULT = 'default'
+    CHATBOT_PLAYGROUND = 'chatbot_playground'
+    DOCUMENT_PLAYGROUND = 'document_playground'
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of DatasetType from a JSON string"""
+        return cls(json.loads(json_str))
+
+