Skip to content

Commit 18543a0

Browse files
committed
feat: add OpenAI embedding model parameters form
--bug=1062976 --user=刘瑞斌 【模型】openai的向量模型,设置dimensions参数后,向量化失败 https://www.tapd.cn/62980211/s/1790130
1 parent 58f29ee commit 18543a0

File tree

1 file changed

+22
-1
lines changed
  • apps/models_provider/impl/openai_model_provider/credential

1 file changed

+22
-1
lines changed

apps/models_provider/impl/openai_model_provider/credential/embedding.py

Lines changed: 22 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,9 +13,27 @@
1313

1414
from common import forms
1515
from common.exception.app_exception import AppApiException
16-
from common.forms import BaseForm
16+
from common.forms import BaseForm, TooltipLabel
1717
from models_provider.base_model_provider import BaseModelCredential, ValidCode
1818

19+
class OpenAIEmbeddingModelParams(BaseForm):
20+
dimensions = forms.SingleSelect(
21+
TooltipLabel(
22+
_('Dimensions'),
23+
_('')
24+
),
25+
required=True,
26+
default_value=1024,
27+
value_field='value',
28+
text_field='label',
29+
option_list=[
30+
{'label': '1536', 'value': '1536'},
31+
{'label': '1024', 'value': '1024'},
32+
{'label': '768', 'value': '768'},
33+
{'label': '512', 'value': '512'},
34+
]
35+
)
36+
1937

2038
class OpenAIEmbeddingCredential(BaseForm, BaseModelCredential):
2139
def is_valid(self, model_type: str, model_name, model_credential: Dict[str, object], model_params, provider,
@@ -49,5 +67,8 @@ def is_valid(self, model_type: str, model_name, model_credential: Dict[str, obje
4967
def encryption_dict(self, model: Dict[str, object]):
5068
return {**model, 'api_key': super().encryption(model.get('api_key', ''))}
5169

70+
def get_model_params_setting_form(self, model_name):
71+
return OpenAIEmbeddingModelParams()
72+
5273
api_base = forms.TextInputField('API URL', required=True)
5374
api_key = forms.PasswordInputField('API Key', required=True)

0 commit comments

Comments
 (0)