Unverified Commit 4aed1fe8 authored by takatost's avatar takatost Committed by GitHub

fix: Azure text-davinci-003 missing (#1867)

parent 2381264a
...@@ -448,6 +448,46 @@ LLM_BASE_MODELS = [ ...@@ -448,6 +448,46 @@ LLM_BASE_MODELS = [
currency='USD', currency='USD',
) )
) )
),
AzureBaseModel(
base_model_name='text-davinci-003',
entity=AIModelEntity(
model='fake-deployment-name',
label=I18nObject(
en_US='fake-deployment-name-label',
),
model_type=ModelType.LLM,
fetch_from=FetchFrom.CUSTOMIZABLE_MODEL,
model_properties={
'mode': LLMMode.COMPLETION.value,
'context_size': 4096,
},
parameter_rules=[
ParameterRule(
name='temperature',
**PARAMETER_RULE_TEMPLATE[DefaultParameterName.TEMPERATURE],
),
ParameterRule(
name='top_p',
**PARAMETER_RULE_TEMPLATE[DefaultParameterName.TOP_P],
),
ParameterRule(
name='presence_penalty',
**PARAMETER_RULE_TEMPLATE[DefaultParameterName.PRESENCE_PENALTY],
),
ParameterRule(
name='frequency_penalty',
**PARAMETER_RULE_TEMPLATE[DefaultParameterName.FREQUENCY_PENALTY],
),
_get_max_tokens(default=512, min_val=1, max_val=4096),
],
pricing=PriceConfig(
input=0.02,
output=0.02,
unit=0.001,
currency='USD',
)
)
) )
] ]
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment