Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
D
dify
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-tech
dify
Commits
3ebec8fa
Unverified
Commit
3ebec8fa
authored
Jan 12, 2024
by
killpanda
Committed by
GitHub
Jan 12, 2024
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
fixup /stop api (#2012)
Co-authored-by:
mayue
<
mayue05@qiyi.com
>
parent
f877d19c
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
9 additions
and
5 deletions
+9
-5
completion.py
api/controllers/service_api/app/completion.py
+9
-5
No files found.
api/controllers/service_api/app/completion.py
View file @
3ebec8fa
...
@@ -13,7 +13,7 @@ from core.application_queue_manager import ApplicationQueueManager
...
@@ -13,7 +13,7 @@ from core.application_queue_manager import ApplicationQueueManager
from
core.entities.application_entities
import
InvokeFrom
from
core.entities.application_entities
import
InvokeFrom
from
core.errors.error
import
ModelCurrentlyNotSupportError
,
ProviderTokenNotInitError
,
QuotaExceededError
from
core.errors.error
import
ModelCurrentlyNotSupportError
,
ProviderTokenNotInitError
,
QuotaExceededError
from
core.model_runtime.errors.invoke
import
InvokeError
from
core.model_runtime.errors.invoke
import
InvokeError
from
flask
import
Response
,
stream_with_context
from
flask
import
Response
,
stream_with_context
,
request
from
flask_restful
import
reqparse
from
flask_restful
import
reqparse
from
libs.helper
import
uuid_value
from
libs.helper
import
uuid_value
from
services.completion_service
import
CompletionService
from
services.completion_service
import
CompletionService
...
@@ -75,11 +75,13 @@ class CompletionApi(AppApiResource):
...
@@ -75,11 +75,13 @@ class CompletionApi(AppApiResource):
class
CompletionStopApi
(
AppApiResource
):
class
CompletionStopApi
(
AppApiResource
):
def
post
(
self
,
app_model
,
end_user
,
task_id
):
def
post
(
self
,
app_model
,
_
,
task_id
):
if
app_model
.
mode
!=
'completion'
:
if
app_model
.
mode
!=
'completion'
:
raise
AppUnavailableError
()
raise
AppUnavailableError
()
ApplicationQueueManager
.
set_stop_flag
(
task_id
,
InvokeFrom
.
SERVICE_API
,
end_user
.
id
)
end_user_id
=
request
.
get_json
()
.
get
(
'user'
)
ApplicationQueueManager
.
set_stop_flag
(
task_id
,
InvokeFrom
.
SERVICE_API
,
end_user_id
)
return
{
'result'
:
'success'
},
200
return
{
'result'
:
'success'
},
200
...
@@ -139,11 +141,13 @@ class ChatApi(AppApiResource):
...
@@ -139,11 +141,13 @@ class ChatApi(AppApiResource):
class
ChatStopApi
(
AppApiResource
):
class
ChatStopApi
(
AppApiResource
):
def
post
(
self
,
app_model
,
end_user
,
task_id
):
def
post
(
self
,
app_model
,
_
,
task_id
):
if
app_model
.
mode
!=
'chat'
:
if
app_model
.
mode
!=
'chat'
:
raise
NotChatAppError
()
raise
NotChatAppError
()
ApplicationQueueManager
.
set_stop_flag
(
task_id
,
InvokeFrom
.
SERVICE_API
,
end_user
.
id
)
end_user_id
=
request
.
get_json
()
.
get
(
'user'
)
ApplicationQueueManager
.
set_stop_flag
(
task_id
,
InvokeFrom
.
SERVICE_API
,
end_user_id
)
return
{
'result'
:
'success'
},
200
return
{
'result'
:
'success'
},
200
...
...
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