🌡v0.2.1,新增2个配置项

This commit is contained in:
2024-10-04 22:00:02 +08:00
parent e4facb65f7
commit a0bfa7fa7f
5 changed files with 23 additions and 7 deletions

View File

@ -121,7 +121,9 @@ async def marsho(
usermsg += str(clean_text)
response = await client.complete(
messages=context.build(target.id, target.private)+[UserMessage(content=usermsg)],
model=model_name
model=model_name,
temperature=config.marshoai_temperature,
max_tokens=config.marshoai_max_tokens
)
#await UniMessage(str(response)).send()
choice = response.choices[0]

View File

@ -11,4 +11,6 @@ class ConfigModel(BaseModel):
marshoai_enable_praises: bool = True
marshoai_enable_time_prompt: bool = True
marshoai_azure_endpoint: str = "https://models.inference.ai.azure.com"
marshoai_temperature: float = None
marshoai_max_tokens: int = None
config: ConfigModel = get_plugin_config(ConfigModel)