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
af7be9bd
Unverified
Commit
af7be9bd
authored
Jan 05, 2024
by
takatost
Committed by
GitHub
Jan 05, 2024
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
Feat/optimize entity construct (#1935)
parent
2cfd5568
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
9 additions
and
2 deletions
+9
-2
model_provider_cache.py
api/core/helper/model_provider_cache.py
+1
-1
provider_manager.py
api/core/provider_manager.py
+8
-1
No files found.
api/core/helper/model_provider_cache.py
View file @
af7be9bd
...
@@ -40,7 +40,7 @@ class ProviderCredentialsCache:
...
@@ -40,7 +40,7 @@ class ProviderCredentialsCache:
:param credentials: provider credentials
:param credentials: provider credentials
:return:
:return:
"""
"""
redis_client
.
setex
(
self
.
cache_key
,
36
00
,
json
.
dumps
(
credentials
))
redis_client
.
setex
(
self
.
cache_key
,
864
00
,
json
.
dumps
(
credentials
))
def
delete
(
self
)
->
None
:
def
delete
(
self
)
->
None
:
"""
"""
...
...
api/core/provider_manager.py
View file @
af7be9bd
...
@@ -229,11 +229,18 @@ class ProviderManager:
...
@@ -229,11 +229,18 @@ class ProviderManager:
return
None
return
None
provider_instance
=
model_provider_factory
.
get_provider_instance
(
default_model
.
provider_name
)
provider_instance
=
model_provider_factory
.
get_provider_instance
(
default_model
.
provider_name
)
provider_schema
=
provider_instance
.
get_provider_schema
()
return
DefaultModelEntity
(
return
DefaultModelEntity
(
model
=
default_model
.
model_name
,
model
=
default_model
.
model_name
,
model_type
=
model_type
,
model_type
=
model_type
,
provider
=
DefaultModelProviderEntity
(
**
provider_instance
.
get_provider_schema
()
.
to_simple_provider
()
.
dict
())
provider
=
DefaultModelProviderEntity
(
provider
=
provider_schema
.
provider
,
label
=
provider_schema
.
label
,
icon_small
=
provider_schema
.
icon_small
,
icon_large
=
provider_schema
.
icon_large
,
supported_model_types
=
provider_schema
.
supported_model_types
)
)
)
def
update_default_model_record
(
self
,
tenant_id
:
str
,
model_type
:
ModelType
,
provider
:
str
,
model
:
str
)
\
def
update_default_model_record
(
self
,
tenant_id
:
str
,
model_type
:
ModelType
,
provider
:
str
,
model
:
str
)
\
...
...
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