from typing import Any

from pydantic import BaseModel, ConfigDict, Field

from core.tools.entities.tool_entities import ToolProviderType


class ToolFieldConfig(BaseModel):
    model_config = ConfigDict(extra="forbid")

    id: str
    value: Any
    auto: bool = False


class ToolConfiguration(BaseModel):
    model_config = ConfigDict(extra="forbid")

    fields: list[ToolFieldConfig] = Field(default_factory=list)

    def default_values(self) -> dict[str, Any]:
        return {field.id: field.value for field in self.fields if field.value is not None}


class ToolReference(BaseModel):
    model_config = ConfigDict(extra="forbid")

    uuid: str
    type: ToolProviderType
    provider: str
    tool_name: str
    credential_id: str | None = None
    configuration: ToolConfiguration | None = None


class FileReference(BaseModel):
    model_config = ConfigDict(extra="forbid")

    source: str
    asset_id: str


class SkillMetadata(BaseModel):
    model_config = ConfigDict(extra="allow")

    tools: dict[str, ToolReference] = Field(default_factory=dict)
    files: list[FileReference] = Field(default_factory=list)