作者 lixiang

删除无用代码

... ... @@ -31,6 +31,4 @@ public interface QuestionEmbeddingMapper {
@Param("limit") int limit,
@Param("minSimilarity") Double minSimilarity);
List<QuestionEmbedding> similaritySearch(@Param("vector") float[] vector,
@Param("limit") int limit);
}
\ No newline at end of file
... ...
... ... @@ -82,10 +82,5 @@
]]>
</select>
<select id="similaritySearch" resultMap="questionEmbeddingResultMap">
<!-- SELECT *, embedding <-> #{vector} AS similarity-->
<!-- FROM question_embedding-->
<!-- ORDER BY similarity ASC-->
<!-- LIMIT #{limit}-->
</select>
</mapper>
\ No newline at end of file
... ...
... ... @@ -9,15 +9,22 @@ import org.springframework.web.multipart.MultipartFile;
import java.util.List;
public interface IQuestionEmbeddingService {
Page<QuestionEmbedding> findAll(QuestionEmbedding questionEmbedding, Integer pageNo, Integer pageSize);
Integer findQuestionCount(QuestionEmbedding questionEmbedding);
QuestionEmbedding findById(String id);
int insert(QuestionEmbedding record);
int update(QuestionEmbedding record);
int removeByIds(List<String> ids);
int deleteById(String id);
List<QuestionEmbedding> similaritySearchByQuestion(String question, int limit, Double minSimilarity, AiragChatsetting chatSetting);
List<QuestionEmbedding> similaritySearch(float[] vector, int limit);
Result<?> processZipUpload(MultipartFile file, String knowledgeId);
}
\ No newline at end of file
... ...
... ... @@ -118,13 +118,10 @@ public class QuestionEmbeddingServiceImpl implements IQuestionEmbeddingService {
Response<Embedding> embedding = aiModelUtils.getEmbedding(chatSetting.getEmbeddingId(), question);
return questionEmbeddingMapper.similaritySearchByQuestion(embedding.content().vector(), limit, 1-minSimilarity);
return questionEmbeddingMapper.similaritySearchByQuestion(embedding.content().vector(), limit, 1 - minSimilarity);
}
@Override
public List<QuestionEmbedding> similaritySearch(float[] vector, int limit) {
return questionEmbeddingMapper.similaritySearch(vector, limit);
}
@Override
public Result<?> processZipUpload(MultipartFile zipFile, String knowledgeId) {
... ...