Package com.alibaba.cloud.ai.service.base

Class BaseNl2SqlService

java.lang.Object

com.alibaba.cloud.ai.service.base.BaseNl2SqlService

Direct Known Subclasses:

AnalyticNl2SqlService, SimpleNl2SqlService

public class BaseNl2SqlService
extends Object

Field Summary

Fields

Modifier and Type	Field	Description
final LlmService	aiService
protected final Accessor	dbAccessor
protected final DbConfig	dbConfig
protected final BaseSchemaService	schemaService
protected final BaseVectorStoreService	vectorStoreService

Constructor Summary

Constructors

Constructor	Description
BaseNl2SqlService(BaseVectorStoreService vectorStoreService, BaseSchemaService schemaService, LlmService aiService, Accessor dbAccessor, DbConfig dbConfig)

Method Summary

Modifier and Type	Method	Description
String	executeSql(String sql)
List<String>	expandQuestion(String query)	将问题扩展为多个不同表述的问题变体
List<String>	extractEvidences(String query)	抽取证据
List<String>	extractEvidences(String query, String agentId)	抽取证据 - 支持智能体隔离
List<String>	extractKeywords(String query, List<String> evidenceList)
Set<String>	fineSelect(SchemaDTO schemaDTO, String sqlGenerateSchemaMissingAdvice)
SchemaDTO	fineSelect(SchemaDTO schemaDTO, String query, List<String> evidenceList)
SchemaDTO	fineSelect(SchemaDTO schemaDTO, String query, List<String> evidenceList, String sqlGenerateSchemaMissingAdvice)
String	generateSql(List<String> evidenceList, String query, SchemaDTO schemaDTO)
String	generateSql(List<String> evidenceList, String query, SchemaDTO schemaDTO, String sql, String exceptionMessage)
String	isRecallInfoSatisfyRequirement(String query, SchemaDTO schemaDTO, List<String> evidenceList)
String	nl2sql(String query)
String	rewrite(String query)
reactor.core.publisher.Flux<org.springframework.ai.chat.model.ChatResponse>	rewriteStream(String query)
reactor.core.publisher.Flux<org.springframework.ai.chat.model.ChatResponse>	rewriteStream(String query, String agentId)
SchemaDTO	select(String query, List<String> evidenceList)
SchemaDTO	select(String query, List<String> evidenceList, String agentId)
String	semanticConsistency(String sql, String queryPrompt)	Deprecated.
reactor.core.publisher.Flux<org.springframework.ai.chat.model.ChatResponse>	semanticConsistencyStream(String sql, String queryPrompt)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Details

vectorStoreService

protected final BaseVectorStoreService vectorStoreService

schemaService

protected final BaseSchemaService schemaService

aiService

public final LlmService aiService

dbAccessor

protected final Accessor dbAccessor

dbConfig

protected final DbConfig dbConfig

Constructor Details

BaseNl2SqlService

public BaseNl2SqlService(BaseVectorStoreService vectorStoreService,
 BaseSchemaService schemaService,
 LlmService aiService,
 Accessor dbAccessor,
 DbConfig dbConfig)

Method Details

rewriteStream

public reactor.core.publisher.Flux<org.springframework.ai.chat.model.ChatResponse> rewriteStream(String query)
                                                                                          throws Exception

Throws:

Exception

rewriteStream

public reactor.core.publisher.Flux<org.springframework.ai.chat.model.ChatResponse> rewriteStream(String query,
 String agentId)
                                                                                          throws Exception

Throws:

Exception

rewrite

public String rewrite(String query)
               throws Exception

Throws:

Exception

nl2sql

public String nl2sql(String query)
              throws Exception

Throws:

Exception

executeSql

public String executeSql(String sql)
                  throws Exception

Throws:

Exception

semanticConsistency

@Deprecated
public String semanticConsistency(String sql,
 String queryPrompt)
                           throws Exception

Deprecated.

Throws:

Exception

semanticConsistencyStream

public reactor.core.publisher.Flux<org.springframework.ai.chat.model.ChatResponse> semanticConsistencyStream(String sql,
 String queryPrompt)
                                                                                                      throws Exception

Throws:

Exception

expandQuestion

public List<String> expandQuestion(String query)

将问题扩展为多个不同表述的问题变体

Parameters:

query - 原始问题

Returns:

包含原始问题和扩展问题的列表

extractEvidences

public List<String> extractEvidences(String query)

抽取证据

extractEvidences

public List<String> extractEvidences(String query,
 String agentId)

抽取证据 - 支持智能体隔离

extractKeywords

public List<String> extractKeywords(String query,
 List<String> evidenceList)

select

public SchemaDTO select(String query,
 List<String> evidenceList)
                 throws Exception

Throws:

Exception

select

public SchemaDTO select(String query,
 List<String> evidenceList,
 String agentId)
                 throws Exception

Throws:

Exception

isRecallInfoSatisfyRequirement

public String isRecallInfoSatisfyRequirement(String query,
 SchemaDTO schemaDTO,
 List<String> evidenceList)

generateSql

public String generateSql(List<String> evidenceList,
 String query,
 SchemaDTO schemaDTO,
 String sql,
 String exceptionMessage)
                   throws Exception

Throws:

Exception

generateSql

public String generateSql(List<String> evidenceList,
 String query,
 SchemaDTO schemaDTO)
                   throws Exception

Throws:

Exception

fineSelect

public Set<String> fineSelect(SchemaDTO schemaDTO,
 String sqlGenerateSchemaMissingAdvice)

fineSelect

public SchemaDTO fineSelect(SchemaDTO schemaDTO,
 String query,
 List<String> evidenceList)

fineSelect

public SchemaDTO fineSelect(SchemaDTO schemaDTO,
 String query,
 List<String> evidenceList,
 String sqlGenerateSchemaMissingAdvice)