add completion app creation back

This commit is contained in:
takatost 2024-03-17 16:29:54 +08:00
parent b99eadecf6
commit 73c2b35dfe
3 changed files with 34 additions and 3 deletions

View File

@ -1,3 +1,5 @@
import json
from models.model import AppMode from models.model import AppMode
default_app_templates = { default_app_templates = {
@ -10,6 +12,35 @@ default_app_templates = {
} }
}, },
# completion default mode
AppMode.COMPLETION: {
'app': {
'mode': AppMode.COMPLETION.value,
'enable_site': True,
'enable_api': True
},
'model_config': {
'model': {
"provider": "openai",
"name": "gpt-4",
"mode": "chat",
"completion_params": {}
},
'user_input_form': json.dumps([
{
"paragraph": {
"label": "Query",
"variable": "query",
"required": True,
"default": ""
}
}
]),
'pre_prompt': '{{query}}'
},
},
# chat default mode # chat default mode
AppMode.CHAT: { AppMode.CHAT: {
'app': { 'app': {

View File

@ -22,7 +22,7 @@ from core.tools.utils.configuration import ToolParameterConfigurationManager
from core.tools.tool_manager import ToolManager from core.tools.tool_manager import ToolManager
ALLOW_CREATE_APP_MODES = ['chat', 'agent-chat', 'advanced-chat', 'workflow'] ALLOW_CREATE_APP_MODES = ['chat', 'agent-chat', 'advanced-chat', 'workflow', 'completion']
class AppListApi(Resource): class AppListApi(Resource):

View File

@ -150,10 +150,10 @@ class AppService:
if not workflow: if not workflow:
raise ValueError("Missing workflow in data argument " raise ValueError("Missing workflow in data argument "
"when app mode is advanced-chat or workflow") "when app mode is advanced-chat or workflow")
elif app_mode in [AppMode.CHAT, AppMode.AGENT_CHAT]: elif app_mode in [AppMode.CHAT, AppMode.AGENT_CHAT, AppMode.COMPLETION]:
if not model_config_data: if not model_config_data:
raise ValueError("Missing model_config in data argument " raise ValueError("Missing model_config in data argument "
"when app mode is chat or agent-chat") "when app mode is chat, agent-chat or completion")
else: else:
raise ValueError("Invalid app mode") raise ValueError("Invalid app mode")