Method update

backend/app/operators/model/model.py:159–205

(self, update_dict: Dict, **kwargs)

        return model

    async def update(self, update_dict: Dict, **kwargs) -> ModelEntity:
        model_id = kwargs["model_id"]
        model: Model = await super().get(model_id=model_id)

        new_update_dict = {}
        if update_dict.get("name") is not None:
            new_update_dict["name"] = update_dict.get("name")

        model_schema_id = update_dict.get("model_schema_id")
        provider_model_id = update_dict.get("provider_model_id")
        model_type = update_dict.get("type")
        credentials = update_dict.get("credentials")
        properties = update_dict.get("properties")

        if model_schema_id or provider_model_id or model_type or credentials or properties:
            # verify model credentials
            (
                new_model_schema_id,
                new_provider_id,
                new_provider_model_id,
                new_model_type,
                new_encrypted_credentials,
                new_display_credentials,
                new_properties,
            ) = await verify_model_credentials(
                model_schema_id=model_schema_id or model.model_schema_id,
                provider_model_id=provider_model_id or model.provider_model_id,
                properties=properties or model.properties,
                model_type=model_type or model.type,
                credentials=credentials,
                encrypted_credentials=model.encrypted_credentials if not credentials else None,
            )

            new_update_dict["model_schema_id"] = new_model_schema_id
            new_update_dict["provider_id"] = new_provider_id
            new_update_dict["provider_model_id"] = new_provider_model_id
            new_update_dict["type"] = new_model_type
            new_update_dict["properties"] = new_properties
            if credentials:
                new_update_dict["encrypted_credentials"] = new_encrypted_credentials
                new_update_dict["display_credentials"] = new_display_credentials

        model = await super().update(
            model_id=model_id,
            update_dict=new_update_dict,
        )
        return model


model_ops = ModelOperator(