/*
 * Copyright 2024-2026 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.alibaba.cloud.ai.agent.nacos;

import io.micrometer.observation.ObservationRegistry;

import org.springframework.ai.chat.client.observation.ChatClientObservationConvention;
import org.springframework.ai.chat.observation.ChatModelObservationConvention;
import org.springframework.ai.model.tool.ToolCallingManager;

public class ObservationConfiguration {

	private ObservationRegistry observationRegistry;

	private ToolCallingManager toolCallingManager;

	private ChatModelObservationConvention chatModelObservationConvention;

	private ChatClientObservationConvention chatClientObservationConvention;

	public ObservationConfiguration(ObservationRegistry observationRegistry, ToolCallingManager toolCallingManager,
			ChatModelObservationConvention chatModelObservationConvention,
			ChatClientObservationConvention chatClientObservationConvention) {
		this.observationRegistry = observationRegistry;
		this.toolCallingManager = toolCallingManager;
		this.chatModelObservationConvention = chatModelObservationConvention;
		this.chatClientObservationConvention = chatClientObservationConvention;
	}

	public ObservationRegistry getObservationRegistry() {
		return observationRegistry;
	}

	public void setObservationRegistry(ObservationRegistry observationRegistry) {
		this.observationRegistry = observationRegistry;
	}

	public ToolCallingManager getToolCallingManager() {
		return toolCallingManager;
	}

	public void setToolCallingManager(ToolCallingManager toolCallingManager) {
		this.toolCallingManager = toolCallingManager;
	}

	public ChatModelObservationConvention getChatModelObservationConvention() {
		return chatModelObservationConvention;
	}

	public void setChatModelObservationConvention(ChatModelObservationConvention chatModelObservationConvention) {
		this.chatModelObservationConvention = chatModelObservationConvention;
	}

	public ChatClientObservationConvention getChatClientObservationConvention() {
		return chatClientObservationConvention;
	}

	public void setChatClientObservationConvention(ChatClientObservationConvention chatClientObservationConvention) {
		this.chatClientObservationConvention = chatClientObservationConvention;
	}
}
