package com.smtaiserver.smtaiserver.javaai;


import java.util.Set;

import com.smtaiserver.smtaiserver.core.SMTAIServerRequest;
import com.smtaiserver.smtaiserver.javaai.qwen.SMTJavaAISessionQwen;
import com.smtservlet.util.Json;

public class SMTJavaAIChat 
{
	public static Json questionChat(String llmName, Set<String> setAgentGroup, String groupType, String question, String prevQuestion, boolean rawMode, SMTAIServerRequest tranReq) throws Exception
	{
		SMTJavaAISession session = new SMTJavaAISessionQwen();
		try
		{
			session.setPrevQuestion(prevQuestion);
			session.setQuestion(question);
			Json jsonResult = session.executeSupervisor(rawMode, setAgentGroup, groupType, tranReq, true);
			
			return jsonResult;
		}
		finally
		{
			session.close();
		}
	}
	
	public static Json callExtChat(String llmName, String question, String callExtId, Json callExtArgs, SMTAIServerRequest tranReq) throws Exception
	{
		SMTJavaAISession session = new SMTJavaAISessionQwen();
		try
		{
			session.setQuestion(question);
			session.executeExtChat(callExtId, callExtArgs, tranReq);
			
			return tranReq.getResultJsonWr().getRootJson();
		}
		finally
		{
			session.close();
		}
	}
	
}