From 8fa38e89682a4670d5b5e14519bf1288dc0b522b Mon Sep 17 00:00:00 2001
From: TangCheng <tangchengmail@163.com>
Date: 星期一, 14 四月 2025 12:21:45 +0800
Subject: [PATCH] 千问llm去掉连接池
---
JAVA/SMTAIServer/src/main/java/com/smtaiserver/smtaiserver/javaai/llm/qwen/SMTLLMFactoryQwen.java | 57 ++++++---------------------------------------------------
1 files changed, 6 insertions(+), 51 deletions(-)
diff --git a/JAVA/SMTAIServer/src/main/java/com/smtaiserver/smtaiserver/javaai/llm/qwen/SMTLLMFactoryQwen.java b/JAVA/SMTAIServer/src/main/java/com/smtaiserver/smtaiserver/javaai/llm/qwen/SMTLLMFactoryQwen.java
index 0b618e7..b10ec04 100644
--- a/JAVA/SMTAIServer/src/main/java/com/smtaiserver/smtaiserver/javaai/llm/qwen/SMTLLMFactoryQwen.java
+++ b/JAVA/SMTAIServer/src/main/java/com/smtaiserver/smtaiserver/javaai/llm/qwen/SMTLLMFactoryQwen.java
@@ -1,11 +1,5 @@
package com.smtaiserver.smtaiserver.javaai.llm.qwen;
-import org.apache.commons.pool2.BasePooledObjectFactory;
-import org.apache.commons.pool2.PooledObject;
-import org.apache.commons.pool2.impl.DefaultPooledObject;
-import org.apache.commons.pool2.impl.GenericObjectPool;
-import org.apache.commons.pool2.impl.GenericObjectPoolConfig;
-
import com.alibaba.dashscope.aigc.generation.Generation;
import com.alibaba.dashscope.embeddings.TextEmbedding;
import com.alibaba.dashscope.utils.Constants;
@@ -16,35 +10,6 @@
public class SMTLLMFactoryQwen extends SMTLLMFactory
{
- public class SMTQwenLLMPoolFactory extends BasePooledObjectFactory<Generation>
- {
- @Override
- public Generation create() throws Exception {
- return new Generation();
- }
-
- @Override
- public PooledObject<Generation> wrap(Generation obj) {
- return new DefaultPooledObject<>(obj);
- }
- }
-
- public class SMTQwenEmbedPoolFactory extends BasePooledObjectFactory<TextEmbedding>
- {
- @Override
- public TextEmbedding create() throws Exception {
- return new TextEmbedding();
- }
-
- @Override
- public PooledObject<TextEmbedding> wrap(TextEmbedding obj) {
- return new DefaultPooledObject<>(obj);
- }
- }
-
- ////////////////////////////////////////////////////////////////////////////////////////
- private GenericObjectPool<TextEmbedding> _qwenEmbedPool;
- private GenericObjectPool<Generation> _qwennLLMPool;
private Json _jsonFactory;
private Json _jsonConnect;
@@ -57,18 +22,7 @@
_jsonConnect = Json.read(rec.getString("connect_args"));
Constants.apiKey= _jsonFactory.getJson("key").asString(); //"sk-57653d19816c44ed81c731ebadeaa3d9";
- {
- GenericObjectPoolConfig<Generation> config = new GenericObjectPoolConfig<>();
- config.setMaxTotal(32);
- config.setMinIdle(32);
- _qwennLLMPool = new GenericObjectPool<>(new SMTQwenLLMPoolFactory(), config);
- }
- {
- GenericObjectPoolConfig<TextEmbedding> config = new GenericObjectPoolConfig<>();
- config.setMaxTotal(32);
- config.setMinIdle(32);
- _qwenEmbedPool = new GenericObjectPool<>(new SMTQwenEmbedPoolFactory(), config);
- }
+
}
public Json getConnectJson()
@@ -78,22 +32,23 @@
public SMTLLMConnect allocLLM() throws Exception
{
- Generation gen = _qwennLLMPool.borrowObject();
+ Generation gen = new Generation(); //_qwennLLMPool.borrowObject();
return new SMTLLMConnectQwen(gen, this);
}
public TextEmbedding allocLLMEmbedding() throws Exception
{
- return _qwenEmbedPool.borrowObject();
+ //return _qwenEmbedPool.borrowObject();
+ return new TextEmbedding();
}
public void freeLLMEmbedding(TextEmbedding embed)
{
- _qwenEmbedPool.returnObject(embed);
+ //_qwenEmbedPool.returnObject(embed);
}
public void returnGenToPool(Generation gen)
{
- _qwennLLMPool.returnObject(gen);
+ //_qwennLLMPool.returnObject(gen);
}
}
--
Gitblit v1.9.3