Commit 6c4e6a66 authored by alex yao's avatar alex yao

feat:长文/写作 添加思考模式/联网模式

parent fa46d72a
...@@ -615,7 +615,7 @@ public class LongTextDialoguesServiceImpl implements LongTextDialoguesService { ...@@ -615,7 +615,7 @@ public class LongTextDialoguesServiceImpl implements LongTextDialoguesService {
FunctionCallResult functionCallResult = checkPluginUseEntity.getFunctionCallResult(); FunctionCallResult functionCallResult = checkPluginUseEntity.getFunctionCallResult();
if (functionCallResult != null) { if (functionCallResult != null) {
FunctionCall functionCall = functionCallResult.getFunctionCall(); FunctionCall functionCall = functionCallResult.getFunctionCall();
if ("web_search".equals(functionCall.getName())) { if (functionCallResult.isNeed() && functionCall != null && "web_search".equals(functionCall.getName())) {
AbstractFunctionResult<List<WebSearchFunctionResult>> functionResult = webSearchFunction.doFunction(functionCall.getArguments(), dialoguesId, null, null); AbstractFunctionResult<List<WebSearchFunctionResult>> functionResult = webSearchFunction.doFunction(functionCall.getArguments(), dialoguesId, null, null);
if (functionResult != null && CollectionUtils.isNotEmpty(functionResult.getFunctionResult())) { if (functionResult != null && CollectionUtils.isNotEmpty(functionResult.getFunctionResult())) {
List<WebSearchFunctionResult> webSearchFunctionResults = functionResult.getFunctionResult(); List<WebSearchFunctionResult> webSearchFunctionResults = functionResult.getFunctionResult();
......
...@@ -30,26 +30,26 @@ public class LongTextDialoguesCallDto { ...@@ -30,26 +30,26 @@ public class LongTextDialoguesCallDto {
/** /**
* 是否启用搜索引擎 * 是否启用搜索引擎
*/ */
private Boolean enableSearchEngine; private boolean enableSearchEngine;
/** /**
* 是否开启深度思考 * 是否开启深度思考
*/ */
private Boolean enableDeepThinking; private boolean enableDeepThinking;
public Boolean getEnableSearchEngine() { public boolean getEnableSearchEngine() {
return enableSearchEngine; return enableSearchEngine;
} }
public void setEnableSearchEngine(Boolean enableSearchEngine) { public void setEnableSearchEngine(boolean enableSearchEngine) {
this.enableSearchEngine = enableSearchEngine; this.enableSearchEngine = enableSearchEngine;
} }
public Boolean getEnableDeepThinking() { public boolean getEnableDeepThinking() {
return enableDeepThinking; return enableDeepThinking;
} }
public void setEnableDeepThinking(Boolean enableDeepThinking) { public void setEnableDeepThinking(boolean enableDeepThinking) {
this.enableDeepThinking = enableDeepThinking; this.enableDeepThinking = enableDeepThinking;
} }
......
...@@ -28,12 +28,12 @@ public class PortalDialoguesDto { ...@@ -28,12 +28,12 @@ public class PortalDialoguesDto {
/** /**
* 是否启用搜索引擎 * 是否启用搜索引擎
*/ */
private Boolean enableSearchEngine; private boolean enableSearchEngine;
/** /**
* 是否开启深度思考 * 是否开启深度思考
*/ */
private Boolean enableDeepThinking; private boolean enableDeepThinking;
/** /**
* 文件地址 * 文件地址
...@@ -69,11 +69,11 @@ public class PortalDialoguesDto { ...@@ -69,11 +69,11 @@ public class PortalDialoguesDto {
this.dialoguesId = dialoguesId; this.dialoguesId = dialoguesId;
} }
public Boolean getEnableSearchEngine() { public boolean getEnableSearchEngine() {
return enableSearchEngine; return enableSearchEngine;
} }
public void setEnableSearchEngine(Boolean enableSearchEngine) { public void setEnableSearchEngine(boolean enableSearchEngine) {
this.enableSearchEngine = enableSearchEngine; this.enableSearchEngine = enableSearchEngine;
} }
......
...@@ -238,7 +238,7 @@ public class AiWritingServiceImpl implements AiWritingService { ...@@ -238,7 +238,7 @@ public class AiWritingServiceImpl implements AiWritingService {
FunctionCallResult functionCallResult = checkPluginUseEntity.getFunctionCallResult(); FunctionCallResult functionCallResult = checkPluginUseEntity.getFunctionCallResult();
if (functionCallResult != null) { if (functionCallResult != null) {
FunctionCall functionCall = functionCallResult.getFunctionCall(); FunctionCall functionCall = functionCallResult.getFunctionCall();
if ("web_search".equals(functionCall.getName())) { if (functionCallResult.isNeed() && functionCall != null && "web_search".equals(functionCall.getName())) {
AbstractFunctionResult<List<WebSearchFunctionResult>> functionResult = webSearchFunction.doFunction(functionCall.getArguments(), dialoguesId, null, null); AbstractFunctionResult<List<WebSearchFunctionResult>> functionResult = webSearchFunction.doFunction(functionCall.getArguments(), dialoguesId, null, null);
if (functionResult != null && CollectionUtils.isNotEmpty(functionResult.getFunctionResult())) { if (functionResult != null && CollectionUtils.isNotEmpty(functionResult.getFunctionResult())) {
List<WebSearchFunctionResult> webSearchFunctionResults = functionResult.getFunctionResult(); List<WebSearchFunctionResult> webSearchFunctionResults = functionResult.getFunctionResult();
......
...@@ -30,27 +30,27 @@ public class AiWritingDto { ...@@ -30,27 +30,27 @@ public class AiWritingDto {
/** /**
* 是否启用搜索引擎 * 是否启用搜索引擎
*/ */
private Boolean enableSearchEngine; private boolean enableSearchEngine;
/** /**
* 是否开启深度思考 * 是否开启深度思考
*/ */
private Boolean enableDeepThinking; private boolean enableDeepThinking;
public Boolean getEnableSearchEngine() { public boolean getEnableSearchEngine() {
return enableSearchEngine; return enableSearchEngine;
} }
public void setEnableSearchEngine(Boolean enableSearchEngine) { public void setEnableSearchEngine(boolean enableSearchEngine) {
this.enableSearchEngine = enableSearchEngine; this.enableSearchEngine = enableSearchEngine;
} }
public Boolean getEnableDeepThinking() { public boolean getEnableDeepThinking() {
return enableDeepThinking; return enableDeepThinking;
} }
public void setEnableDeepThinking(Boolean enableDeepThinking) { public void setEnableDeepThinking(boolean enableDeepThinking) {
this.enableDeepThinking = enableDeepThinking; this.enableDeepThinking = enableDeepThinking;
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment