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

fix

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