Commit 3d187137 authored by alex yao's avatar alex yao

fix

parent bfc5ed6c
...@@ -71,6 +71,7 @@ public class WebSearchFunction extends AbstractLargeModelFunction { ...@@ -71,6 +71,7 @@ public class WebSearchFunction extends AbstractLargeModelFunction {
.cse().list().setCx(CX) .cse().list().setCx(CX)
.setKey(KEY) .setKey(KEY)
.setQ(query) .setQ(query)
.setStart(1L)
.setNum(3) .setNum(3)
.execute(); .execute();
List<Result> items = execute.getItems(); List<Result> items = execute.getItems();
...@@ -86,12 +87,13 @@ public class WebSearchFunction extends AbstractLargeModelFunction { ...@@ -86,12 +87,13 @@ public class WebSearchFunction extends AbstractLargeModelFunction {
String title = item.getTitle(); String title = item.getTitle();
String snippet = item.getSnippet(); String snippet = item.getSnippet();
WebSearchFunctionResult webSearchResult = new WebSearchFunctionResult(); WebSearchFunctionResult webSearchResult = new WebSearchFunctionResult();
webSearchResult.setTitle(title); // webSearchResult.setTitle(title);
webSearchResult.setUrl(link); // webSearchResult.setUrl(link);
webSearchResult.setSnippet(snippet); // webSearchResult.setSnippet(snippet);
webSearchResult.setContent(htmlContent); webSearchResult.setContent(htmlContent);
results.add(webSearchResult); results.add(webSearchResult);
} }
logger.info("web search result:{}", results.toString());
return JsonUtils.serialize(results); return JsonUtils.serialize(results);
} catch (Exception e) { } catch (Exception e) {
logger.error("web search error:" + e.getMessage()); logger.error("web search error:" + e.getMessage());
......
...@@ -6,37 +6,37 @@ package cn.com.poc.thirdparty.resource.demand.ai.function.web_seach; ...@@ -6,37 +6,37 @@ package cn.com.poc.thirdparty.resource.demand.ai.function.web_seach;
*/ */
public class WebSearchFunctionResult { public class WebSearchFunctionResult {
private String title; // private String title;
//
private String url; // private String url;
//
private String snippet; // private String snippet;
private String content; private String content;
public String getTitle() { // public String getTitle() {
return title; // return title;
} // }
//
public void setTitle(String title) { // public void setTitle(String title) {
this.title = title; // this.title = title;
} // }
//
public String getUrl() { // public String getUrl() {
return url; // return url;
} // }
//
public void setUrl(String url) { // public void setUrl(String url) {
this.url = url; // this.url = url;
} // }
//
public String getSnippet() { // public String getSnippet() {
return snippet; // return snippet;
} // }
//
public void setSnippet(String snippet) { // public void setSnippet(String snippet) {
this.snippet = snippet; // this.snippet = snippet;
} // }
public String getContent() { public String getContent() {
return content; return content;
...@@ -45,4 +45,14 @@ public class WebSearchFunctionResult { ...@@ -45,4 +45,14 @@ public class WebSearchFunctionResult {
public void setContent(String content) { public void setContent(String content) {
this.content = content; this.content = content;
} }
@Override
public String toString() {
return "{" +
// "title='" + title + '\'' +
// ", url='" + url + '\'' +
// ", snippet='" + snippet + '\'' +
"content='" + content + '\'' +
'}';
}
} }
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