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
3b55dc08
Unverified
Commit
3b55dc08
authored
Feb 25, 2024
by
Yeuoly
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
fix: linter
parent
ad84b996
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
19 additions
and
17 deletions
+19
-17
model_runtime.py
api/controllers/inner_api/model_runtime.py
+3
-3
model_runner.py
api/core/app_runner/model_runner.py
+6
-5
completion_service.py
api/services/completion_service.py
+10
-9
No files found.
api/controllers/inner_api/model_runtime.py
View file @
3b55dc08
import
json
import
json
from
collections.abc
import
Generator
from
typing
import
Union
from
flask_restful
import
Resource
,
reqparse
from
flask
import
Response
from
flask
import
Response
from
flask.helpers
import
stream_with_context
from
flask.helpers
import
stream_with_context
from
flask_restful
import
Resource
,
reqparse
from
controllers.console.setup
import
setup_required
from
controllers.console.setup
import
setup_required
from
controllers.inner_api
import
api
from
controllers.inner_api
import
api
from
controllers.inner_api.wraps
import
inner_api_only
from
controllers.inner_api.wraps
import
inner_api_only
from
services.completion_service
import
CompletionService
from
services.completion_service
import
CompletionService
from
typing
import
Generator
,
Union
class
EnterpriseModelInvokeLLMApi
(
Resource
):
class
EnterpriseModelInvokeLLMApi
(
Resource
):
"""Model invoke API for enterprise edition"""
"""Model invoke API for enterprise edition"""
...
...
api/core/app_runner/model_runner.py
View file @
3b55dc08
from
core.provider_manager
import
ProviderManager
,
ProviderModelBundle
from
collections.abc
import
Generator
from
typing
import
Optional
,
Union
,
cast
from
core.model_runtime.entities.llm_entities
import
LLMResult
,
LLMResultChunk
,
LLMUsage
from
core.model_runtime.entities.llm_entities
import
LLMResult
,
LLMResultChunk
,
LLMUsage
from
core.model_runtime.entities.message_entities
import
PromptMessage
,
PromptMessageTool
from
core.model_runtime.entities.message_entities
import
PromptMessage
,
PromptMessageTool
from
core.model_runtime.model_providers.__base.large_language_model
import
LargeLanguageModel
from
core.model_runtime.entities.model_entities
import
ModelType
from
core.model_runtime.entities.model_entities
import
ModelType
from
core.model_runtime.model_providers.__base.large_language_model
import
LargeLanguageModel
from
core.model_runtime.utils.encoders
import
jsonable_encoder
from
core.model_runtime.utils.encoders
import
jsonable_encoder
from
core.provider_manager
import
ProviderModelBundle
from
events.inner_event
import
model_was_invoked
from
events.inner_event
import
model_was_invoked
from
typing
import
Generator
,
Union
,
cast
,
Optional
class
ModelRunner
:
class
ModelRunner
:
"""
"""
...
...
api/services/completion_service.py
View file @
3b55dc08
...
@@ -4,23 +4,23 @@ from typing import Any, Union
...
@@ -4,23 +4,23 @@ from typing import Any, Union
from
sqlalchemy
import
and_
from
sqlalchemy
import
and_
from
core.app_runner.model_runner
import
ModelRunner
from
core.application_manager
import
ApplicationManager
from
core.application_manager
import
ApplicationManager
from
core.provider_manager
import
ProviderManager
from
core.errors.error
import
ModelCurrentlyNotSupportError
,
ProviderTokenNotInitError
,
QuotaExceededError
from
core.entities.application_entities
import
InvokeFrom
from
core.entities.application_entities
import
InvokeFrom
from
core.entities.model_entities
import
ModelStatus
from
core.entities.model_entities
import
ModelStatus
from
core.
model_runtime.entities.model_entities
import
ModelType
from
core.
errors.error
import
ModelCurrentlyNotSupportError
,
ProviderTokenNotInitError
,
QuotaExceededError
from
core.file.message_file_parser
import
MessageFileParser
from
core.file.message_file_parser
import
MessageFileParser
from
core.model_runtime.entities.message_entities
import
(
from
core.model_runtime.entities.message_entities
import
(
PromptMessage
,
UserPromptMessage
,
SystemPromptMessage
,
AssistantPromptMessage
,
AssistantPromptMessage
,
Tool
PromptMessage
,
PromptMessage
,
PromptMessageRole
,
PromptMessageRole
,
PromptMessageTool
PromptMessageTool
,
SystemPromptMessage
,
ToolPromptMessage
,
UserPromptMessage
,
)
)
from
core.app_runner.model_runner
import
ModelRunner
from
core.model_runtime.entities.model_entities
import
ModelType
from
core.provider_manager
import
ProviderManager
from
extensions.ext_database
import
db
from
extensions.ext_database
import
db
from
models.model
import
Account
,
App
,
AppModelConfig
,
Conversation
,
EndUser
,
Message
from
models.model
import
Account
,
App
,
AppModelConfig
,
Conversation
,
EndUser
,
Message
from
services.app_model_config_service
import
AppModelConfigService
from
services.app_model_config_service
import
AppModelConfigService
...
@@ -29,6 +29,7 @@ from services.errors.app_model_config import AppModelConfigBrokenError
...
@@ -29,6 +29,7 @@ from services.errors.app_model_config import AppModelConfigBrokenError
from
services.errors.conversation
import
ConversationCompletedError
,
ConversationNotExistsError
from
services.errors.conversation
import
ConversationCompletedError
,
ConversationNotExistsError
from
services.errors.message
import
MessageNotExistsError
from
services.errors.message
import
MessageNotExistsError
class
CompletionService
:
class
CompletionService
:
@
classmethod
@
classmethod
...
...
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