| 
									
										
										
										
											2023-05-15 08:51:32 +08:00
										 |  |  | import json | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2024-04-08 18:51:46 +08:00
										 |  |  | from models.model import AppMode | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | default_app_templates = { | 
					
						
							|  |  |  |     # workflow default mode | 
					
						
							|  |  |  |     AppMode.WORKFLOW: { | 
					
						
							|  |  |  |         'app': { | 
					
						
							|  |  |  |             'mode': AppMode.WORKFLOW.value, | 
					
						
							|  |  |  |             'enable_site': True, | 
					
						
							|  |  |  |             'enable_api': True | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |     }, | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2023-05-15 08:51:32 +08:00
										 |  |  |     # completion default mode | 
					
						
							| 
									
										
										
										
											2024-04-08 18:51:46 +08:00
										 |  |  |     AppMode.COMPLETION: { | 
					
						
							| 
									
										
										
										
											2023-05-15 08:51:32 +08:00
										 |  |  |         'app': { | 
					
						
							| 
									
										
										
										
											2024-04-08 18:51:46 +08:00
										 |  |  |             'mode': AppMode.COMPLETION.value, | 
					
						
							| 
									
										
										
										
											2023-05-15 08:51:32 +08:00
										 |  |  |             'enable_site': True, | 
					
						
							| 
									
										
										
										
											2024-04-08 18:51:46 +08:00
										 |  |  |             'enable_api': True | 
					
						
							| 
									
										
										
										
											2023-05-15 08:51:32 +08:00
										 |  |  |         }, | 
					
						
							|  |  |  |         'model_config': { | 
					
						
							| 
									
										
										
										
											2024-04-08 18:51:46 +08:00
										 |  |  |             'model': { | 
					
						
							| 
									
										
										
										
											2023-05-15 08:51:32 +08:00
										 |  |  |                 "provider": "openai", | 
					
						
							| 
									
										
										
										
											2024-04-08 18:51:46 +08:00
										 |  |  |                 "name": "gpt-4", | 
					
						
							|  |  |  |                 "mode": "chat", | 
					
						
							| 
									
										
										
										
											2024-02-29 12:43:07 +08:00
										 |  |  |                 "completion_params": {} | 
					
						
							| 
									
										
										
										
											2024-04-08 18:51:46 +08:00
										 |  |  |             }, | 
					
						
							| 
									
										
										
										
											2023-09-10 00:12:34 +08:00
										 |  |  |             'user_input_form': json.dumps([ | 
					
						
							|  |  |  |                 { | 
					
						
							|  |  |  |                     "paragraph": { | 
					
						
							|  |  |  |                         "label": "Query", | 
					
						
							|  |  |  |                         "variable": "query", | 
					
						
							|  |  |  |                         "required": True, | 
					
						
							|  |  |  |                         "default": "" | 
					
						
							|  |  |  |                     } | 
					
						
							|  |  |  |                 } | 
					
						
							|  |  |  |             ]), | 
					
						
							|  |  |  |             'pre_prompt': '{{query}}' | 
					
						
							| 
									
										
										
										
											2024-04-08 18:51:46 +08:00
										 |  |  |         }, | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2023-05-15 08:51:32 +08:00
										 |  |  |     }, | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     # chat default mode | 
					
						
							| 
									
										
										
										
											2024-04-08 18:51:46 +08:00
										 |  |  |     AppMode.CHAT: { | 
					
						
							| 
									
										
										
										
											2023-05-15 08:51:32 +08:00
										 |  |  |         'app': { | 
					
						
							| 
									
										
										
										
											2024-04-08 18:51:46 +08:00
										 |  |  |             'mode': AppMode.CHAT.value, | 
					
						
							| 
									
										
										
										
											2023-05-15 08:51:32 +08:00
										 |  |  |             'enable_site': True, | 
					
						
							| 
									
										
										
										
											2024-04-08 18:51:46 +08:00
										 |  |  |             'enable_api': True | 
					
						
							| 
									
										
										
										
											2023-05-15 08:51:32 +08:00
										 |  |  |         }, | 
					
						
							|  |  |  |         'model_config': { | 
					
						
							| 
									
										
										
										
											2024-04-08 18:51:46 +08:00
										 |  |  |             'model': { | 
					
						
							| 
									
										
										
										
											2023-05-15 08:51:32 +08:00
										 |  |  |                 "provider": "openai", | 
					
						
							| 
									
										
										
										
											2024-04-08 18:51:46 +08:00
										 |  |  |                 "name": "gpt-4", | 
					
						
							| 
									
										
										
										
											2023-10-12 23:13:10 +08:00
										 |  |  |                 "mode": "chat", | 
					
						
							| 
									
										
										
										
											2024-02-29 12:43:07 +08:00
										 |  |  |                 "completion_params": {} | 
					
						
							| 
									
										
										
										
											2024-04-08 18:51:46 +08:00
										 |  |  |             } | 
					
						
							| 
									
										
										
										
											2023-05-15 08:51:32 +08:00
										 |  |  |         } | 
					
						
							|  |  |  |     }, | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2024-04-08 18:51:46 +08:00
										 |  |  |     # advanced-chat default mode | 
					
						
							|  |  |  |     AppMode.ADVANCED_CHAT: { | 
					
						
							|  |  |  |         'app': { | 
					
						
							|  |  |  |             'mode': AppMode.ADVANCED_CHAT.value, | 
					
						
							|  |  |  |             'enable_site': True, | 
					
						
							|  |  |  |             'enable_api': True | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |     }, | 
					
						
							| 
									
										
										
										
											2023-05-15 08:51:32 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2024-04-08 18:51:46 +08:00
										 |  |  |     # agent-chat default mode | 
					
						
							|  |  |  |     AppMode.AGENT_CHAT: { | 
					
						
							|  |  |  |         'app': { | 
					
						
							|  |  |  |             'mode': AppMode.AGENT_CHAT.value, | 
					
						
							|  |  |  |             'enable_site': True, | 
					
						
							|  |  |  |             'enable_api': True | 
					
						
							|  |  |  |         }, | 
					
						
							|  |  |  |         'model_config': { | 
					
						
							|  |  |  |             'model': { | 
					
						
							|  |  |  |                 "provider": "openai", | 
					
						
							|  |  |  |                 "name": "gpt-4", | 
					
						
							|  |  |  |                 "mode": "chat", | 
					
						
							|  |  |  |                 "completion_params": {} | 
					
						
							|  |  |  |             } | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | } |