model_template.py 2.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586
  1. import json
  2. from models.model import AppMode
  3. default_app_templates = {
  4. # workflow default mode
  5. AppMode.WORKFLOW: {
  6. 'app': {
  7. 'mode': AppMode.WORKFLOW.value,
  8. 'enable_site': True,
  9. 'enable_api': True
  10. }
  11. },
  12. # completion default mode
  13. AppMode.COMPLETION: {
  14. 'app': {
  15. 'mode': AppMode.COMPLETION.value,
  16. 'enable_site': True,
  17. 'enable_api': True
  18. },
  19. 'model_config': {
  20. 'model': {
  21. "provider": "openai",
  22. "name": "gpt-4",
  23. "mode": "chat",
  24. "completion_params": {}
  25. },
  26. 'user_input_form': json.dumps([
  27. {
  28. "paragraph": {
  29. "label": "Query",
  30. "variable": "query",
  31. "required": True,
  32. "default": ""
  33. }
  34. }
  35. ]),
  36. 'pre_prompt': '{{query}}'
  37. },
  38. },
  39. # chat default mode
  40. AppMode.CHAT: {
  41. 'app': {
  42. 'mode': AppMode.CHAT.value,
  43. 'enable_site': True,
  44. 'enable_api': True
  45. },
  46. 'model_config': {
  47. 'model': {
  48. "provider": "openai",
  49. "name": "gpt-4",
  50. "mode": "chat",
  51. "completion_params": {}
  52. }
  53. }
  54. },
  55. # advanced-chat default mode
  56. AppMode.ADVANCED_CHAT: {
  57. 'app': {
  58. 'mode': AppMode.ADVANCED_CHAT.value,
  59. 'enable_site': True,
  60. 'enable_api': True
  61. }
  62. },
  63. # agent-chat default mode
  64. AppMode.AGENT_CHAT: {
  65. 'app': {
  66. 'mode': AppMode.AGENT_CHAT.value,
  67. 'enable_site': True,
  68. 'enable_api': True
  69. },
  70. 'model_config': {
  71. 'model': {
  72. "provider": "openai",
  73. "name": "gpt-4",
  74. "mode": "chat",
  75. "completion_params": {}
  76. }
  77. }
  78. }
  79. }