config.py 1.7 KB

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