Method Edit

module/ai-api/iml.go:160–211

(ctx context.Context, serviceId string, apiId string, input *ai_api_dto.EditAPI)

}

func (i *imlAPIModule) Edit(ctx context.Context, serviceId string, apiId string, input *ai_api_dto.EditAPI) error {
	info, err := i.serviceService.Get(ctx, serviceId)
	if err != nil {
		return err
	}
	if info.Kind != service.AIService {
		return fmt.Errorf("service kind is not ai service")
	}

	return i.transaction.Transaction(ctx, func(txCtx context.Context) error {
		apiInfo, err := i.aiAPIService.Get(ctx, apiId)
		if err != nil {
			return err
		}
		orgPath := apiInfo.Path
		if input.Path != nil {
			apiInfo.Path = *input.Path
		}
		if input.Description != nil {
			apiInfo.Description = *input.Description
		}
		err = i.updateAPIDoc(ctx, serviceId, info.Name, orgPath, apiInfo.Path, apiInfo.Name, apiInfo.Description, input.AiPrompt)
		if err != nil {
			return err
		}
		var modelId *string
		var providerId *string
		if input.AiModel != nil {
			modelId = &input.AiModel.Id
			providerId = &input.AiModel.Provider
		}
		if input.AiPrompt != nil {
			apiInfo.AdditionalConfig["ai_prompt"] = input.AiPrompt
		}
		if input.AiModel != nil {
			apiInfo.AdditionalConfig["ai_model"] = input.AiModel
		}
		typ := ai_api_dto.ModelType(input.AiModel.Type).Int()
		return i.aiAPIService.Save(ctx, apiId, &ai_api.Edit{
			Name:             input.Name,
			Path:             input.Path,
			Description:      input.Description,
			Timeout:          input.Timeout,
			Retry:            input.Retry,
			Model:            modelId,
			Provider:         providerId,
			Type:             &typ,
			AdditionalConfig: &apiInfo.AdditionalConfig,
			Disable:          input.Disable,
		})
	})
}

func (i *imlAPIModule) Delete(ctx context.Context, serviceId string, apiId string) error {
	info, err := i.serviceService.Get(ctx, serviceId)