Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Contribute to GitLab
Sign in
Toggle navigation
P
poc-api
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
poc
poc-api
Commits
59897f3e
Commit
59897f3e
authored
Jul 04, 2025
by
alex yao
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
fix: web search 插件读取pdf文件
parent
e308ea99
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
15 additions
and
3 deletions
+15
-3
WebSearchFunction.java
...ource/demand/ai/function/web_seach/WebSearchFunction.java
+4
-2
WebSearchFunctionResult.java
...demand/ai/function/web_seach/WebSearchFunctionResult.java
+11
-1
No files found.
src/main/java/cn/com/poc/thirdparty/resource/demand/ai/function/web_seach/WebSearchFunction.java
View file @
59897f3e
...
@@ -88,6 +88,7 @@ public class WebSearchFunction extends AbstractLargeModelFunction {
...
@@ -88,6 +88,7 @@ public class WebSearchFunction extends AbstractLargeModelFunction {
Search
execute
=
getCustomSearchAPI
()
Search
execute
=
getCustomSearchAPI
()
.
cse
().
list
().
setCx
(
CX
)
.
cse
().
list
().
setCx
(
CX
)
.
setKey
(
KEY
)
.
setKey
(
KEY
)
.
setFileType
(
"pdf,html"
)
.
setQ
(
query
)
.
setQ
(
query
)
.
setStart
(
1L
)
.
setStart
(
1L
)
.
setNum
(
3
)
.
setNum
(
3
)
...
@@ -97,7 +98,7 @@ public class WebSearchFunction extends AbstractLargeModelFunction {
...
@@ -97,7 +98,7 @@ public class WebSearchFunction extends AbstractLargeModelFunction {
StringBuilder
stringBuilder
=
new
StringBuilder
();
StringBuilder
stringBuilder
=
new
StringBuilder
();
int
i
=
0
;
int
i
=
0
;
for
(
WebSearchFunctionResult
webSearchFunctionResult
:
results
)
{
for
(
WebSearchFunctionResult
webSearchFunctionResult
:
results
)
{
stringBuilder
.
append
(
i
).
append
(
":"
).
append
(
webSearchFunctionResult
.
getContent
()).
append
(
"\n"
);
stringBuilder
.
append
(
i
).
append
(
":"
).
append
(
webSearchFunctionResult
.
get
Html
Content
()).
append
(
"\n"
);
}
}
result
.
setPromptContent
(
stringBuilder
.
toString
());
result
.
setPromptContent
(
stringBuilder
.
toString
());
result
.
setFunctionResult
(
results
);
result
.
setFunctionResult
(
results
);
...
@@ -137,7 +138,8 @@ public class WebSearchFunction extends AbstractLargeModelFunction {
...
@@ -137,7 +138,8 @@ public class WebSearchFunction extends AbstractLargeModelFunction {
WebSearchFunctionResult
webSearchResult
=
new
WebSearchFunctionResult
();
WebSearchFunctionResult
webSearchResult
=
new
WebSearchFunctionResult
();
webSearchResult
.
setTitle
(
item
.
getTitle
());
webSearchResult
.
setTitle
(
item
.
getTitle
());
webSearchResult
.
setUrl
(
item
.
getLink
());
webSearchResult
.
setUrl
(
item
.
getLink
());
webSearchResult
.
setContent
(
htmlContent
);
webSearchResult
.
setContent
(
item
.
getSnippet
());
webSearchResult
.
setHtmlContent
(
item
.
getSnippet
());
results
.
add
(
webSearchResult
);
results
.
add
(
webSearchResult
);
}
}
countDownLatch
.
countDown
();
countDownLatch
.
countDown
();
...
...
src/main/java/cn/com/poc/thirdparty/resource/demand/ai/function/web_seach/WebSearchFunctionResult.java
View file @
59897f3e
...
@@ -10,10 +10,20 @@ public class WebSearchFunctionResult {
...
@@ -10,10 +10,20 @@ public class WebSearchFunctionResult {
private
String
url
;
private
String
url
;
//
//
// private String snippet;
private
String
htmlContent
;
private
String
content
;
private
String
content
;
public
String
getHtmlContent
()
{
return
htmlContent
;
}
public
void
setHtmlContent
(
String
htmlContent
)
{
this
.
htmlContent
=
htmlContent
;
}
public
String
getTitle
()
{
public
String
getTitle
()
{
return
title
;
return
title
;
}
}
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment