# coding: utf-8

"""
    Hindsight HTTP API

    HTTP API for Hindsight

    The version of the OpenAPI document: 0.5.3
    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 typing import Optional, Set
from typing_extensions import Self

class BankTemplateConfig(BaseModel):
    """
    Bank configuration fields within a template manifest.  Only includes configurable (per-bank) fields. Credential fields (API keys, base URLs) are intentionally excluded for security.
    """ # noqa: E501
    reflect_mission: Optional[StrictStr] = None
    retain_mission: Optional[StrictStr] = None
    retain_extraction_mode: Optional[StrictStr] = None
    retain_custom_instructions: Optional[StrictStr] = None
    retain_chunk_size: Optional[StrictInt] = None
    enable_observations: Optional[StrictBool] = None
    observations_mission: Optional[StrictStr] = None
    disposition_skepticism: Optional[Annotated[int, Field(le=5, strict=True, ge=1)]] = None
    disposition_literalism: Optional[Annotated[int, Field(le=5, strict=True, ge=1)]] = None
    disposition_empathy: Optional[Annotated[int, Field(le=5, strict=True, ge=1)]] = None
    entity_labels: Optional[List[Dict[str, Any]]] = None
    entities_allow_free_form: Optional[StrictBool] = None
    retain_default_strategy: Optional[StrictStr] = None
    retain_strategies: Optional[Dict[str, Any]] = None
    retain_chunk_batch_size: Optional[StrictInt] = None
    mcp_enabled_tools: Optional[List[StrictStr]] = None
    consolidation_llm_batch_size: Optional[StrictInt] = None
    consolidation_source_facts_max_tokens: Optional[StrictInt] = None
    consolidation_source_facts_max_tokens_per_observation: Optional[StrictInt] = None
    max_observations_per_scope: Optional[StrictInt] = None
    reflect_source_facts_max_tokens: Optional[StrictInt] = None
    llm_gemini_safety_settings: Optional[List[Any]] = None
    recall_budget_function: Optional[StrictStr] = None
    recall_budget_fixed_low: Optional[StrictInt] = None
    recall_budget_fixed_mid: Optional[StrictInt] = None
    recall_budget_fixed_high: Optional[StrictInt] = None
    recall_budget_adaptive_low: Optional[Union[StrictFloat, StrictInt]] = None
    recall_budget_adaptive_mid: Optional[Union[StrictFloat, StrictInt]] = None
    recall_budget_adaptive_high: Optional[Union[StrictFloat, StrictInt]] = None
    recall_budget_min: Optional[StrictInt] = None
    recall_budget_max: Optional[StrictInt] = None
    __properties: ClassVar[List[str]] = ["reflect_mission", "retain_mission", "retain_extraction_mode", "retain_custom_instructions", "retain_chunk_size", "enable_observations", "observations_mission", "disposition_skepticism", "disposition_literalism", "disposition_empathy", "entity_labels", "entities_allow_free_form", "retain_default_strategy", "retain_strategies", "retain_chunk_batch_size", "mcp_enabled_tools", "consolidation_llm_batch_size", "consolidation_source_facts_max_tokens", "consolidation_source_facts_max_tokens_per_observation", "max_observations_per_scope", "reflect_source_facts_max_tokens", "llm_gemini_safety_settings", "recall_budget_function", "recall_budget_fixed_low", "recall_budget_fixed_mid", "recall_budget_fixed_high", "recall_budget_adaptive_low", "recall_budget_adaptive_mid", "recall_budget_adaptive_high", "recall_budget_min", "recall_budget_max"]

    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 BankTemplateConfig 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,
        )
        # set to None if reflect_mission (nullable) is None
        # and model_fields_set contains the field
        if self.reflect_mission is None and "reflect_mission" in self.model_fields_set:
            _dict['reflect_mission'] = None

        # set to None if retain_mission (nullable) is None
        # and model_fields_set contains the field
        if self.retain_mission is None and "retain_mission" in self.model_fields_set:
            _dict['retain_mission'] = None

        # set to None if retain_extraction_mode (nullable) is None
        # and model_fields_set contains the field
        if self.retain_extraction_mode is None and "retain_extraction_mode" in self.model_fields_set:
            _dict['retain_extraction_mode'] = None

        # set to None if retain_custom_instructions (nullable) is None
        # and model_fields_set contains the field
        if self.retain_custom_instructions is None and "retain_custom_instructions" in self.model_fields_set:
            _dict['retain_custom_instructions'] = None

        # set to None if retain_chunk_size (nullable) is None
        # and model_fields_set contains the field
        if self.retain_chunk_size is None and "retain_chunk_size" in self.model_fields_set:
            _dict['retain_chunk_size'] = None

        # set to None if enable_observations (nullable) is None
        # and model_fields_set contains the field
        if self.enable_observations is None and "enable_observations" in self.model_fields_set:
            _dict['enable_observations'] = None

        # set to None if observations_mission (nullable) is None
        # and model_fields_set contains the field
        if self.observations_mission is None and "observations_mission" in self.model_fields_set:
            _dict['observations_mission'] = None

        # set to None if disposition_skepticism (nullable) is None
        # and model_fields_set contains the field
        if self.disposition_skepticism is None and "disposition_skepticism" in self.model_fields_set:
            _dict['disposition_skepticism'] = None

        # set to None if disposition_literalism (nullable) is None
        # and model_fields_set contains the field
        if self.disposition_literalism is None and "disposition_literalism" in self.model_fields_set:
            _dict['disposition_literalism'] = None

        # set to None if disposition_empathy (nullable) is None
        # and model_fields_set contains the field
        if self.disposition_empathy is None and "disposition_empathy" in self.model_fields_set:
            _dict['disposition_empathy'] = None

        # set to None if entity_labels (nullable) is None
        # and model_fields_set contains the field
        if self.entity_labels is None and "entity_labels" in self.model_fields_set:
            _dict['entity_labels'] = None

        # set to None if entities_allow_free_form (nullable) is None
        # and model_fields_set contains the field
        if self.entities_allow_free_form is None and "entities_allow_free_form" in self.model_fields_set:
            _dict['entities_allow_free_form'] = None

        # set to None if retain_default_strategy (nullable) is None
        # and model_fields_set contains the field
        if self.retain_default_strategy is None and "retain_default_strategy" in self.model_fields_set:
            _dict['retain_default_strategy'] = None

        # set to None if retain_strategies (nullable) is None
        # and model_fields_set contains the field
        if self.retain_strategies is None and "retain_strategies" in self.model_fields_set:
            _dict['retain_strategies'] = None

        # set to None if retain_chunk_batch_size (nullable) is None
        # and model_fields_set contains the field
        if self.retain_chunk_batch_size is None and "retain_chunk_batch_size" in self.model_fields_set:
            _dict['retain_chunk_batch_size'] = None

        # set to None if mcp_enabled_tools (nullable) is None
        # and model_fields_set contains the field
        if self.mcp_enabled_tools is None and "mcp_enabled_tools" in self.model_fields_set:
            _dict['mcp_enabled_tools'] = None

        # set to None if consolidation_llm_batch_size (nullable) is None
        # and model_fields_set contains the field
        if self.consolidation_llm_batch_size is None and "consolidation_llm_batch_size" in self.model_fields_set:
            _dict['consolidation_llm_batch_size'] = None

        # set to None if consolidation_source_facts_max_tokens (nullable) is None
        # and model_fields_set contains the field
        if self.consolidation_source_facts_max_tokens is None and "consolidation_source_facts_max_tokens" in self.model_fields_set:
            _dict['consolidation_source_facts_max_tokens'] = None

        # set to None if consolidation_source_facts_max_tokens_per_observation (nullable) is None
        # and model_fields_set contains the field
        if self.consolidation_source_facts_max_tokens_per_observation is None and "consolidation_source_facts_max_tokens_per_observation" in self.model_fields_set:
            _dict['consolidation_source_facts_max_tokens_per_observation'] = None

        # set to None if max_observations_per_scope (nullable) is None
        # and model_fields_set contains the field
        if self.max_observations_per_scope is None and "max_observations_per_scope" in self.model_fields_set:
            _dict['max_observations_per_scope'] = None

        # set to None if reflect_source_facts_max_tokens (nullable) is None
        # and model_fields_set contains the field
        if self.reflect_source_facts_max_tokens is None and "reflect_source_facts_max_tokens" in self.model_fields_set:
            _dict['reflect_source_facts_max_tokens'] = None

        # set to None if llm_gemini_safety_settings (nullable) is None
        # and model_fields_set contains the field
        if self.llm_gemini_safety_settings is None and "llm_gemini_safety_settings" in self.model_fields_set:
            _dict['llm_gemini_safety_settings'] = None

        # set to None if recall_budget_function (nullable) is None
        # and model_fields_set contains the field
        if self.recall_budget_function is None and "recall_budget_function" in self.model_fields_set:
            _dict['recall_budget_function'] = None

        # set to None if recall_budget_fixed_low (nullable) is None
        # and model_fields_set contains the field
        if self.recall_budget_fixed_low is None and "recall_budget_fixed_low" in self.model_fields_set:
            _dict['recall_budget_fixed_low'] = None

        # set to None if recall_budget_fixed_mid (nullable) is None
        # and model_fields_set contains the field
        if self.recall_budget_fixed_mid is None and "recall_budget_fixed_mid" in self.model_fields_set:
            _dict['recall_budget_fixed_mid'] = None

        # set to None if recall_budget_fixed_high (nullable) is None
        # and model_fields_set contains the field
        if self.recall_budget_fixed_high is None and "recall_budget_fixed_high" in self.model_fields_set:
            _dict['recall_budget_fixed_high'] = None

        # set to None if recall_budget_adaptive_low (nullable) is None
        # and model_fields_set contains the field
        if self.recall_budget_adaptive_low is None and "recall_budget_adaptive_low" in self.model_fields_set:
            _dict['recall_budget_adaptive_low'] = None

        # set to None if recall_budget_adaptive_mid (nullable) is None
        # and model_fields_set contains the field
        if self.recall_budget_adaptive_mid is None and "recall_budget_adaptive_mid" in self.model_fields_set:
            _dict['recall_budget_adaptive_mid'] = None

        # set to None if recall_budget_adaptive_high (nullable) is None
        # and model_fields_set contains the field
        if self.recall_budget_adaptive_high is None and "recall_budget_adaptive_high" in self.model_fields_set:
            _dict['recall_budget_adaptive_high'] = None

        # set to None if recall_budget_min (nullable) is None
        # and model_fields_set contains the field
        if self.recall_budget_min is None and "recall_budget_min" in self.model_fields_set:
            _dict['recall_budget_min'] = None

        # set to None if recall_budget_max (nullable) is None
        # and model_fields_set contains the field
        if self.recall_budget_max is None and "recall_budget_max" in self.model_fields_set:
            _dict['recall_budget_max'] = None

        return _dict

    @classmethod
    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
        """Create an instance of BankTemplateConfig from a dict"""
        if obj is None:
            return None

        if not isinstance(obj, dict):
            return cls.model_validate(obj)

        _obj = cls.model_validate({
            "reflect_mission": obj.get("reflect_mission"),
            "retain_mission": obj.get("retain_mission"),
            "retain_extraction_mode": obj.get("retain_extraction_mode"),
            "retain_custom_instructions": obj.get("retain_custom_instructions"),
            "retain_chunk_size": obj.get("retain_chunk_size"),
            "enable_observations": obj.get("enable_observations"),
            "observations_mission": obj.get("observations_mission"),
            "disposition_skepticism": obj.get("disposition_skepticism"),
            "disposition_literalism": obj.get("disposition_literalism"),
            "disposition_empathy": obj.get("disposition_empathy"),
            "entity_labels": obj.get("entity_labels"),
            "entities_allow_free_form": obj.get("entities_allow_free_form"),
            "retain_default_strategy": obj.get("retain_default_strategy"),
            "retain_strategies": obj.get("retain_strategies"),
            "retain_chunk_batch_size": obj.get("retain_chunk_batch_size"),
            "mcp_enabled_tools": obj.get("mcp_enabled_tools"),
            "consolidation_llm_batch_size": obj.get("consolidation_llm_batch_size"),
            "consolidation_source_facts_max_tokens": obj.get("consolidation_source_facts_max_tokens"),
            "consolidation_source_facts_max_tokens_per_observation": obj.get("consolidation_source_facts_max_tokens_per_observation"),
            "max_observations_per_scope": obj.get("max_observations_per_scope"),
            "reflect_source_facts_max_tokens": obj.get("reflect_source_facts_max_tokens"),
            "llm_gemini_safety_settings": obj.get("llm_gemini_safety_settings"),
            "recall_budget_function": obj.get("recall_budget_function"),
            "recall_budget_fixed_low": obj.get("recall_budget_fixed_low"),
            "recall_budget_fixed_mid": obj.get("recall_budget_fixed_mid"),
            "recall_budget_fixed_high": obj.get("recall_budget_fixed_high"),
            "recall_budget_adaptive_low": obj.get("recall_budget_adaptive_low"),
            "recall_budget_adaptive_mid": obj.get("recall_budget_adaptive_mid"),
            "recall_budget_adaptive_high": obj.get("recall_budget_adaptive_high"),
            "recall_budget_min": obj.get("recall_budget_min"),
            "recall_budget_max": obj.get("recall_budget_max")
        })
        return _obj


