topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-12 12:37:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

(fix):fix python service interact (#494)

Browse Source
This commit is contained in:
Scott
2023-12-11 19:31:36 +08:00
committed by GitHub
parent 3cdfcae01c
commit f3f60af231
5 changed files with 12 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
chat/core/src/main/java/com/tencent/supersonic/chat/parser/sql/llm/OneStepsSqlGeneration.java
View File

@@ -19,6 +19,6 @@ public class OneStepsSqlGeneration implements SqlGeneration, InitializingBean {
@Override @Override
public void afterPropertiesSet() { public void afterPropertiesSet() {
SqlGenerationFactory.addSqlGenerationForFactory(SqlGenerationMode.ONE_STEP, this); SqlGenerationFactory.addSqlGenerationForFactory(SqlGenerationMode.ONE_STEP_AUTO_COT, this);
} }
} }

2
chat/core/src/main/java/com/tencent/supersonic/chat/parser/sql/llm/TwoStepCSSqlGeneration.java
View File

@@ -19,6 +19,6 @@ public class TwoStepCSSqlGeneration implements SqlGeneration, InitializingBean {
@Override @Override
public void afterPropertiesSet() { public void afterPropertiesSet() {
SqlGenerationFactory.addSqlGenerationForFactory(SqlGenerationMode.TWO_STEP_CS, this); SqlGenerationFactory.addSqlGenerationForFactory(SqlGenerationMode.TWO_STEP_AUTO_COT_SELF_CONSISTENCY, this);
} }
} }

2
chat/core/src/main/java/com/tencent/supersonic/chat/parser/sql/llm/TwoStepSqlGeneration.java
View File

@@ -65,6 +65,6 @@ public class TwoStepSqlGeneration implements SqlGeneration, InitializingBean {
@Override @Override
public void afterPropertiesSet() { public void afterPropertiesSet() {
SqlGenerationFactory.addSqlGenerationForFactory(SqlGenerationMode.TWO_STEP, this); SqlGenerationFactory.addSqlGenerationForFactory(SqlGenerationMode.TWO_STEP_AUTO_COT, this);
} }
} }

12
chat/core/src/main/java/com/tencent/supersonic/chat/query/llm/s2sql/LLMReq.java
View File

@@ -1,5 +1,6 @@
package com.tencent.supersonic.chat.query.llm.s2sql; package com.tencent.supersonic.chat.query.llm.s2sql;
import com.fasterxml.jackson.annotation.JsonValue;
import java.util.List; import java.util.List;
import lombok.Data; import lombok.Data;
@@ -18,7 +19,7 @@ public class LLMReq {
private String priorExts; private String priorExts;
private SqlGenerationMode sqlGenerationMode = SqlGenerationMode.TWO_STEP; private SqlGenerationMode sqlGenerationMode = SqlGenerationMode.TWO_STEP_AUTO_COT_SELF_CONSISTENCY;
@Data @Data
public static class ElementValue { public static class ElementValue {
@@ -48,11 +49,13 @@ public class LLMReq {
public enum SqlGenerationMode { public enum SqlGenerationMode {
ONE_STEP("ONE_STEP"), ONE_STEP_AUTO_COT("1_pass_auto_cot"),
TWO_STEP("TWO_STEP"), ONE_STEP_AUTO_COT_SELF_CONSISTENCY("1_pass_auto_cot_self_consistency"),
TWO_STEP_CS("TWO_STEP_CS"); TWO_STEP_AUTO_COT("2_pass_auto_cot"),
TWO_STEP_AUTO_COT_SELF_CONSISTENCY("2_pass_auto_cot_self_consistency");
private String name; private String name;
@@ -61,6 +64,7 @@ public class LLMReq {
this.name = name; this.name = name;
} }
@JsonValue
public String getName() { public String getName() {
return name; return name;
} }

2
chat/python/services_router/query2sql_service.py
View File

@@ -45,7 +45,7 @@ async def query2sql(query_body: Mapping[str, Any]):
else: else:
filter_condition = query_body['filterCondition'] filter_condition = query_body['filterCondition']
if 'sql_generation_mode' not in query_body: if 'sqlGenerationMode' not in query_body:
raise HTTPException(status_code=400, detail="sql_generation_mode is not in query_body") raise HTTPException(status_code=400, detail="sql_generation_mode is not in query_body")
else: else:
sql_generation_mode = query_body['sql_generation_mode'] sql_generation_mode = query_body['sql_generation_mode']