Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
A
ai-chat-agent
Project
Project
Details
Activity
Releases
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
李超鹏
ai-chat-agent
Commits
0f1badb5
Commit
0f1badb5
authored
Dec 26, 2023
by
李超鹏
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
历史记录
parent
33502dd9
Pipeline
#7456
canceled with stages
Changes
2
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
26 additions
and
2 deletions
+26
-2
KnowledgeService.java
src/main/java/cc/shenjiying/ai/manager/KnowledgeService.java
+9
-0
KnowLedgeResource.java
...ain/java/cc/shenjiying/ai/resource/KnowLedgeResource.java
+17
-2
No files found.
src/main/java/cc/shenjiying/ai/manager/KnowledgeService.java
View file @
0f1badb5
...
@@ -10,6 +10,9 @@ import cc.shenjiying.ai.repository.entity.Question;
...
@@ -10,6 +10,9 @@ import cc.shenjiying.ai.repository.entity.Question;
import
cc.shenjiying.ai.repository.entity.Questioner
;
import
cc.shenjiying.ai.repository.entity.Questioner
;
import
cc.shenjiying.ai.repository.mapper.FeedbackMapper
;
import
cc.shenjiying.ai.repository.mapper.FeedbackMapper
;
import
cc.shenjiying.ai.repository.mapper.QuestionMapper
;
import
cc.shenjiying.ai.repository.mapper.QuestionMapper
;
import
com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper
;
import
com.baomidou.mybatisplus.core.conditions.query.QueryWrapper
;
import
com.baomidou.mybatisplus.extension.plugins.pagination.Page
;
import
io.quarkus.logging.Log
;
import
io.quarkus.logging.Log
;
import
io.quarkus.runtime.util.StringUtil
;
import
io.quarkus.runtime.util.StringUtil
;
import
io.smallrye.common.vertx.ContextLocals
;
import
io.smallrye.common.vertx.ContextLocals
;
...
@@ -21,6 +24,7 @@ import jakarta.ws.rs.PathParam;
...
@@ -21,6 +24,7 @@ import jakarta.ws.rs.PathParam;
import
jakarta.ws.rs.QueryParam
;
import
jakarta.ws.rs.QueryParam
;
import
org.eclipse.microprofile.rest.client.inject.RestClient
;
import
org.eclipse.microprofile.rest.client.inject.RestClient
;
import
java.util.List
;
import
java.util.Objects
;
import
java.util.Objects
;
import
java.util.stream.Collectors
;
import
java.util.stream.Collectors
;
...
@@ -90,5 +94,10 @@ public class KnowledgeService {
...
@@ -90,5 +94,10 @@ public class KnowledgeService {
return
entity
.
getId
();
return
entity
.
getId
();
}
}
public
List
<
Question
>
history
(
Long
pageNum
,
Long
pageSize
)
{
return
questionMapper
.
selectList
(
new
Page
<>(
pageNum
,
pageSize
),
new
LambdaQueryWrapper
<
Question
>().
eq
(
Question:
:
getUserId
,
QuestionerHolder
.
getId
())
.
orderByDesc
(
Question:
:
getCreateTime
));
}
}
}
src/main/java/cc/shenjiying/ai/resource/KnowLedgeResource.java
View file @
0f1badb5
...
@@ -4,6 +4,7 @@ import cc.shenjiying.ai.config.QuestionerHolder;
...
@@ -4,6 +4,7 @@ import cc.shenjiying.ai.config.QuestionerHolder;
import
cc.shenjiying.ai.dto.ChatAIResponse
;
import
cc.shenjiying.ai.dto.ChatAIResponse
;
import
cc.shenjiying.ai.dto.FeedbackSaveDTO
;
import
cc.shenjiying.ai.dto.FeedbackSaveDTO
;
import
cc.shenjiying.ai.manager.KnowledgeService
;
import
cc.shenjiying.ai.manager.KnowledgeService
;
import
cc.shenjiying.ai.repository.entity.Question
;
import
io.quarkiverse.bucket4j.runtime.RateLimited
;
import
io.quarkiverse.bucket4j.runtime.RateLimited
;
import
io.smallrye.mutiny.Multi
;
import
io.smallrye.mutiny.Multi
;
import
io.smallrye.mutiny.Uni
;
import
io.smallrye.mutiny.Uni
;
...
@@ -14,8 +15,11 @@ import org.eclipse.microprofile.openapi.annotations.Operation;
...
@@ -14,8 +15,11 @@ import org.eclipse.microprofile.openapi.annotations.Operation;
import
org.eclipse.microprofile.openapi.annotations.parameters.Parameter
;
import
org.eclipse.microprofile.openapi.annotations.parameters.Parameter
;
import
org.eclipse.microprofile.openapi.annotations.tags.Tag
;
import
org.eclipse.microprofile.openapi.annotations.tags.Tag
;
import
org.jboss.resteasy.reactive.RestPath
;
import
org.jboss.resteasy.reactive.RestPath
;
import
org.jboss.resteasy.reactive.RestQuery
;
import
org.jboss.resteasy.reactive.RestStreamElementType
;
import
org.jboss.resteasy.reactive.RestStreamElementType
;
import
java.util.List
;
@Path
(
"/chat"
)
@Path
(
"/chat"
)
@Tag
(
name
=
"问答"
)
@Tag
(
name
=
"问答"
)
public
class
KnowLedgeResource
{
public
class
KnowLedgeResource
{
...
@@ -23,13 +27,13 @@ public class KnowLedgeResource {
...
@@ -23,13 +27,13 @@ public class KnowLedgeResource {
KnowledgeService
knowledgeService
;
KnowledgeService
knowledgeService
;
@HeaderParam
(
QuestionerHolder
.
TOKEN_KEY
)
@HeaderParam
(
QuestionerHolder
.
TOKEN_KEY
)
@Parameter
(
name
=
QuestionerHolder
.
TOKEN_KEY
,
description
=
"登录返回的token"
)
@Parameter
(
name
=
QuestionerHolder
.
TOKEN_KEY
,
description
=
"登录返回的token"
)
String
myHeaderId
;
String
myHeaderId
;
@GET
@GET
@Path
(
"/stream"
)
@Path
(
"/stream"
)
@Operation
(
summary
=
"流式请求"
)
@Operation
(
summary
=
"流式请求"
)
@Parameter
(
name
=
QuestionerHolder
.
TOKEN_KEY
,
description
=
"登录返回的token"
)
@Parameter
(
name
=
QuestionerHolder
.
TOKEN_KEY
,
description
=
"登录返回的token"
)
@Produces
(
MediaType
.
SERVER_SENT_EVENTS
)
@Produces
(
MediaType
.
SERVER_SENT_EVENTS
)
@RestStreamElementType
(
MediaType
.
APPLICATION_JSON
)
@RestStreamElementType
(
MediaType
.
APPLICATION_JSON
)
@RateLimited
(
bucket
=
"chat"
)
@RateLimited
(
bucket
=
"chat"
)
...
@@ -61,4 +65,15 @@ public class KnowLedgeResource {
...
@@ -61,4 +65,15 @@ public class KnowLedgeResource {
return
Uni
.
createFrom
().
item
(()
->
knowledgeService
.
newFeedback
(
feedback
));
return
Uni
.
createFrom
().
item
(()
->
knowledgeService
.
newFeedback
(
feedback
));
}
}
@GET
@Path
(
"/history"
)
@Operation
(
summary
=
"历史记录"
)
@Produces
(
MediaType
.
APPLICATION_JSON
)
public
List
<
Question
>
history
(
@RestQuery
(
"pageNum"
)
Long
pageNum
,
@RestQuery
(
"pageSize"
)
Long
pageSize
)
{
return
knowledgeService
.
history
(
pageNum
,
pageSize
);
}
}
}
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