| | |
| | | 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; |
| | |
| | | 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){ |
| | |
| | | } |
| | | |
| | | 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())); |
| | | } |