config.py 1.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748
  1. from enum import Enum
  2. class ConfigType(str, Enum):
  3. # For frontend
  4. LLM_CUSTOM_LLM_PROVIDER = 'LLM_CUSTOM_LLM_PROVIDER'
  5. LLM_DROP_PARAMS = 'LLM_DROP_PARAMS'
  6. LLM_MAX_INPUT_TOKENS = 'LLM_MAX_INPUT_TOKENS'
  7. LLM_MAX_OUTPUT_TOKENS = 'LLM_MAX_OUTPUT_TOKENS'
  8. LLM_TOP_P = 'LLM_TOP_P'
  9. LLM_TEMPERATURE = 'LLM_TEMPERATURE'
  10. LLM_TIMEOUT = 'LLM_TIMEOUT'
  11. LLM_API_KEY = 'LLM_API_KEY'
  12. LLM_BASE_URL = 'LLM_BASE_URL'
  13. AWS_ACCESS_KEY_ID = 'AWS_ACCESS_KEY_ID'
  14. AWS_SECRET_ACCESS_KEY = 'AWS_SECRET_ACCESS_KEY'
  15. AWS_REGION_NAME = 'AWS_REGION_NAME'
  16. WORKSPACE_BASE = 'WORKSPACE_BASE'
  17. WORKSPACE_MOUNT_PATH = 'WORKSPACE_MOUNT_PATH'
  18. WORKSPACE_MOUNT_REWRITE = 'WORKSPACE_MOUNT_REWRITE'
  19. WORKSPACE_MOUNT_PATH_IN_SANDBOX = 'WORKSPACE_MOUNT_PATH_IN_SANDBOX'
  20. CACHE_DIR = 'CACHE_DIR'
  21. LLM_MODEL = 'LLM_MODEL'
  22. CONFIRMATION_MODE = 'CONFIRMATION_MODE'
  23. SANDBOX_CONTAINER_IMAGE = 'SANDBOX_CONTAINER_IMAGE'
  24. RUN_AS_DEVIN = 'RUN_AS_DEVIN'
  25. LLM_EMBEDDING_MODEL = 'LLM_EMBEDDING_MODEL'
  26. LLM_EMBEDDING_BASE_URL = 'LLM_EMBEDDING_BASE_URL'
  27. LLM_EMBEDDING_DEPLOYMENT_NAME = 'LLM_EMBEDDING_DEPLOYMENT_NAME'
  28. LLM_API_VERSION = 'LLM_API_VERSION'
  29. LLM_NUM_RETRIES = 'LLM_NUM_RETRIES'
  30. LLM_RETRY_MIN_WAIT = 'LLM_RETRY_MIN_WAIT'
  31. LLM_RETRY_MAX_WAIT = 'LLM_RETRY_MAX_WAIT'
  32. AGENT_MEMORY_MAX_THREADS = 'AGENT_MEMORY_MAX_THREADS'
  33. AGENT_MEMORY_ENABLED = 'AGENT_MEMORY_ENABLED'
  34. MAX_ITERATIONS = 'MAX_ITERATIONS'
  35. AGENT = 'AGENT'
  36. E2B_API_KEY = 'E2B_API_KEY'
  37. SANDBOX_BOX_TYPE = 'SANDBOX_BOX_TYPE'
  38. SANDBOX_USER_ID = 'SANDBOX_USER_ID'
  39. SANDBOX_TIMEOUT = 'SANDBOX_TIMEOUT'
  40. USE_HOST_NETWORK = 'USE_HOST_NETWORK'
  41. SSH_HOSTNAME = 'SSH_HOSTNAME'
  42. DISABLE_COLOR = 'DISABLE_COLOR'
  43. DEBUG = 'DEBUG'
  44. FILE_UPLOADS_MAX_FILE_SIZE_MB = 'FILE_UPLOADS_MAX_FILE_SIZE_MB'
  45. FILE_UPLOADS_RESTRICT_FILE_TYPES = 'FILE_UPLOADS_RESTRICT_FILE_TYPES'
  46. FILE_UPLOADS_ALLOWED_EXTENSIONS = 'FILE_UPLOADS_ALLOWED_EXTENSIONS'