shikeying
2024-05-08 8924870a053f0b882ada86421c062cbdb9cff093
deploy-jar-template/src/main/java/com/iplatform/api/MilvusChatApi.java
@@ -7,7 +7,6 @@
import com.iplatform.milvus.ScoreText;
import com.iplatform.milvus.SearchResult;
import com.iplatform.milvus.service.EventServiceImpl;
import com.walker.infrastructure.utils.FileUtils;
import com.walker.infrastructure.utils.JsonUtils;
import com.walker.infrastructure.utils.StringUtils;
import com.walker.support.milvus.OutData;
@@ -26,21 +25,26 @@
import java.util.Map;
import java.util.concurrent.TimeUnit;
@RestController
@RequestMapping("/test/milvus")
//@RestController
//@RequestMapping("/test/milvus")
public class MilvusChatApi extends SystemController {
    private EventServiceImpl eventService;
    private RestTemplate restTemplate;
    private MilvusEngine milvusEngine;
    private Long existId = 4048L;
    private Long existId = null;
    private boolean isBreak = false;
    private static final String URL_EMBEDDING = "http://120.26.128.84:7003/ai/text/embedding";
    private static final String URL_SEARCH_SIMILAR = "http://120.26.128.84:7003/ai/text/search_similar";
    private static final String URL_MILVUS = "120.26.128.84";
    private static final String URL_EMBEDDING = "http://10.8.4.99:7003/ai/text/embedding";
    private static final String URL_SEARCH_SIMILAR = "http://10.8.4.99:7003/ai/text/search_similar";
    private static final String URL_MILVUS = "10.8.4.99";
//    private static final String URL_EMBEDDING = "http://120.26.128.84:7003/ai/text/embedding";
//    private static final String URL_SEARCH_SIMILAR = "http://120.26.128.84:7003/ai/text/search_similar";
//    private static final String URL_MILVUS = "120.26.128.84";
    private static final double BEST_MATCH_SCORE = 0.75;
//    private static final int VECTOR_DIMENSION = 768;
    private static final int VECTOR_DIMENSION = 512;
    @Autowired
    public MilvusChatApi(EventServiceImpl eventService, RestTemplate restTemplate){
@@ -199,7 +203,7 @@
    }
    private List<Float> transfer2FloatList(List<?> list){
        List<Float> vector = new ArrayList<>(768);
        List<Float> vector = new ArrayList<>(VECTOR_DIMENSION);
        for(int i=0; i<list.size(); i++){
            vector.add(Float.parseFloat(list.get(i).toString()));
        }