Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Dify.AI
Dify
提交
983834cd
Dify
项目概览
Dify.AI
/
Dify
8 个月 前同步成功
通知
12
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
Dify
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
未验证
提交
983834cd
编写于
9月 11, 2023
作者:
T
takatost
提交者:
GitHub
9月 11, 2023
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
feat: spark check (#1134)
上级
96d10c8b
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
54 addition
and
1 deletion
+54
-1
api/controllers/console/workspace/model_providers.py
api/controllers/console/workspace/model_providers.py
+16
-0
api/services/provider_service.py
api/services/provider_service.py
+38
-1
未找到文件。
api/controllers/console/workspace/model_providers.py
浏览文件 @
983834cd
...
@@ -285,6 +285,20 @@ class ModelProviderFreeQuotaSubmitApi(Resource):
...
@@ -285,6 +285,20 @@ class ModelProviderFreeQuotaSubmitApi(Resource):
return
result
return
result
class
ModelProviderFreeQuotaQualificationVerifyApi
(
Resource
):
@
setup_required
@
login_required
@
account_initialization_required
def
get
(
self
,
provider_name
:
str
):
provider_service
=
ProviderService
()
result
=
provider_service
.
free_quota_qualification_verify
(
tenant_id
=
current_user
.
current_tenant_id
,
provider_name
=
provider_name
)
return
result
api
.
add_resource
(
ModelProviderListApi
,
'/workspaces/current/model-providers'
)
api
.
add_resource
(
ModelProviderListApi
,
'/workspaces/current/model-providers'
)
api
.
add_resource
(
ModelProviderValidateApi
,
'/workspaces/current/model-providers/<string:provider_name>/validate'
)
api
.
add_resource
(
ModelProviderValidateApi
,
'/workspaces/current/model-providers/<string:provider_name>/validate'
)
api
.
add_resource
(
ModelProviderUpdateApi
,
'/workspaces/current/model-providers/<string:provider_name>'
)
api
.
add_resource
(
ModelProviderUpdateApi
,
'/workspaces/current/model-providers/<string:provider_name>'
)
...
@@ -300,3 +314,5 @@ api.add_resource(ModelProviderPaymentCheckoutUrlApi,
...
@@ -300,3 +314,5 @@ api.add_resource(ModelProviderPaymentCheckoutUrlApi,
'/workspaces/current/model-providers/<string:provider_name>/checkout-url'
)
'/workspaces/current/model-providers/<string:provider_name>/checkout-url'
)
api
.
add_resource
(
ModelProviderFreeQuotaSubmitApi
,
api
.
add_resource
(
ModelProviderFreeQuotaSubmitApi
,
'/workspaces/current/model-providers/<string:provider_name>/free-quota-submit'
)
'/workspaces/current/model-providers/<string:provider_name>/free-quota-submit'
)
api
.
add_resource
(
ModelProviderFreeQuotaQualificationVerifyApi
,
'/workspaces/current/model-providers/<string:provider_name>/free-quota-qualification-verify'
)
api/services/provider_service.py
浏览文件 @
983834cd
...
@@ -518,7 +518,8 @@ class ProviderService:
...
@@ -518,7 +518,8 @@ class ProviderService:
def
free_quota_submit
(
self
,
tenant_id
:
str
,
provider_name
:
str
):
def
free_quota_submit
(
self
,
tenant_id
:
str
,
provider_name
:
str
):
api_key
=
os
.
environ
.
get
(
"FREE_QUOTA_APPLY_API_KEY"
)
api_key
=
os
.
environ
.
get
(
"FREE_QUOTA_APPLY_API_KEY"
)
api_url
=
os
.
environ
.
get
(
"FREE_QUOTA_APPLY_URL"
)
api_base_url
=
os
.
environ
.
get
(
"FREE_QUOTA_APPLY_BASE_URL"
)
api_url
=
api_base_url
+
'/api/v1/providers/apply'
headers
=
{
headers
=
{
'Content-Type'
:
'application/json'
,
'Content-Type'
:
'application/json'
,
...
@@ -546,3 +547,39 @@ class ProviderService:
...
@@ -546,3 +547,39 @@ class ProviderService:
'type'
:
rst
[
'type'
],
'type'
:
rst
[
'type'
],
'result'
:
'success'
'result'
:
'success'
}
}
def
free_quota_qualification_verify
(
self
,
tenant_id
:
str
,
provider_name
:
str
):
api_key
=
os
.
environ
.
get
(
"FREE_QUOTA_APPLY_API_KEY"
)
api_base_url
=
os
.
environ
.
get
(
"FREE_QUOTA_APPLY_BASE_URL"
)
api_url
=
api_base_url
+
'/api/v1/providers/qualification-verify'
headers
=
{
'Content-Type'
:
'application/json'
,
'Authorization'
:
f
"Bearer
{
api_key
}
"
}
response
=
requests
.
post
(
api_url
,
headers
=
headers
,
json
=
{
'workspace_id'
:
tenant_id
,
'provider_name'
:
provider_name
})
if
not
response
.
ok
:
logging
.
error
(
f
"Request FREE QUOTA APPLY SERVER Error:
{
response
.
status_code
}
"
)
raise
ValueError
(
f
"Error:
{
response
.
status_code
}
"
)
rst
=
response
.
json
()
if
rst
[
"code"
]
!=
'success'
:
raise
ValueError
(
f
"error:
{
rst
[
'message'
]
}
"
)
data
=
rst
[
'data'
]
if
data
[
'qualified'
]
is
True
:
return
{
'result'
:
'success'
,
'provider_name'
:
provider_name
,
'flag'
:
True
}
else
:
return
{
'result'
:
'success'
,
'provider_name'
:
provider_name
,
'flag'
:
False
,
'reason'
:
data
[
'reason'
]
}
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录