package com.smtaiserver.smtaiserver.javaai.llm.qwen;

import com.alibaba.dashscope.aigc.generation.Generation;
import com.alibaba.dashscope.embeddings.TextEmbedding;
import com.alibaba.dashscope.utils.Constants;
import com.smtaiserver.smtaiserver.database.SMTDatabase.DBRecord;
import com.smtaiserver.smtaiserver.javaai.llm.core.SMTLLMConnect;
import com.smtaiserver.smtaiserver.javaai.llm.core.SMTLLMFactory;
import com.smtservlet.util.Json;

public class SMTLLMFactoryQwen extends SMTLLMFactory 
{
	private Json								_jsonFactory;
	private Json								_jsonConnect;
	
	@Override
	public void initInstance(DBRecord rec) throws Exception
	{
		super.initInstance(rec);
		
		_jsonFactory = Json.read(rec.getString("factory_args"));
		_jsonConnect = Json.read(rec.getString("connect_args"));
		
		Constants.apiKey= _jsonFactory.getJson("key").asString(); //"sk-57653d19816c44ed81c731ebadeaa3d9";

	}
	
	public Json getConnectJson()
	{
		return _jsonConnect;
	}
	
	public SMTLLMConnect allocLLM() throws Exception
	{
		Generation gen = new Generation(); //_qwennLLMPool.borrowObject();
		return new SMTLLMConnectQwen(gen, this);
	}
	
	public TextEmbedding allocLLMEmbedding() throws Exception
	{
		//return _qwenEmbedPool.borrowObject();
		return new TextEmbedding();
	}
	
	public void freeLLMEmbedding(TextEmbedding embed)
	{
		//_qwenEmbedPool.returnObject(embed);
	}
	
	public void returnGenToPool(Generation gen)
	{
		//_qwennLLMPool.returnObject(gen);
	}
}