mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-26 05:48:36 +08:00
342302ceef
* simplified get * resolved merge conflicts * removed default param for get * add dood setup * add readme * better build process * multi-stage build * revert makefile * rm entrypoint.sh * adjust ssh box for docker * update readme * update readme * fix hostname * change workspace setting * add workspace_mount_base * fixes for workspace dir * clean up frontend * refactor dockerfile * try download.py * change docker order a bit * remove workspace_dir from frontend settings * fix merge issues * Update opendevin/config.py * remove relpath logic from server * rename workspace_mount_base to workspace_base * remove workspace dir plumbing for now * delint * delint * move workspace base dir * remove refs to workspace_dir * factor out constant * fix local directory usage * dont require dir * fix docs * fix arg parsing for task * implement WORKSPACE_MOUNT_PATH * fix workspace dir * fix ports * fix merge issues * add makefile * revert settingsService * fix string * Add address * Update Dockerfile * Update local_box.py * fix lint * move to port 3000 --------- Co-authored-by: மனோஜ்குமார் பழனிச்சாமி <smartmanoj42857@gmail.com> Co-authored-by: enyst <engel.nyst@gmail.com>
97 lines
2.9 KiB
Python
97 lines
2.9 KiB
Python
import copy
|
|
import os
|
|
|
|
import argparse
|
|
import toml
|
|
from dotenv import load_dotenv
|
|
|
|
from opendevin.schema import ConfigType
|
|
|
|
load_dotenv()
|
|
|
|
DEFAULT_CONFIG: dict = {
|
|
ConfigType.LLM_API_KEY: None,
|
|
ConfigType.LLM_BASE_URL: None,
|
|
ConfigType.WORKSPACE_BASE: os.getcwd(),
|
|
ConfigType.WORKSPACE_MOUNT_PATH: None,
|
|
ConfigType.WORKSPACE_MOUNT_REWRITE: None,
|
|
ConfigType.LLM_MODEL: 'gpt-3.5-turbo-1106',
|
|
ConfigType.SANDBOX_CONTAINER_IMAGE: 'ghcr.io/opendevin/sandbox',
|
|
ConfigType.RUN_AS_DEVIN: 'true',
|
|
ConfigType.LLM_EMBEDDING_MODEL: 'local',
|
|
ConfigType.LLM_DEPLOYMENT_NAME: None,
|
|
ConfigType.LLM_API_VERSION: None,
|
|
ConfigType.LLM_NUM_RETRIES: 6,
|
|
ConfigType.LLM_COOLDOWN_TIME: 1,
|
|
ConfigType.MAX_ITERATIONS: 100,
|
|
# GPT-4 pricing is $10 per 1M input tokens. Since tokenization happens on LLM side,
|
|
# we cannot easily count number of tokens, but we can count characters.
|
|
# Assuming 5 characters per token, 5 million is a reasonable default limit.
|
|
ConfigType.MAX_CHARS: 5_000_000,
|
|
ConfigType.AGENT: 'MonologueAgent',
|
|
ConfigType.SANDBOX_TYPE: 'ssh',
|
|
ConfigType.USE_HOST_NETWORK: 'false',
|
|
ConfigType.SSH_HOSTNAME: 'localhost',
|
|
ConfigType.DISABLE_COLOR: 'false',
|
|
}
|
|
|
|
config_str = ''
|
|
if os.path.exists('config.toml'):
|
|
with open('config.toml', 'rb') as f:
|
|
config_str = f.read().decode('utf-8')
|
|
|
|
tomlConfig = toml.loads(config_str)
|
|
config = DEFAULT_CONFIG.copy()
|
|
for k, v in config.items():
|
|
if k in os.environ:
|
|
config[k] = os.environ[k]
|
|
elif k in tomlConfig:
|
|
config[k] = tomlConfig[k]
|
|
|
|
|
|
def parse_arguments():
|
|
parser = argparse.ArgumentParser(
|
|
description='Run an agent with a specific task')
|
|
parser.add_argument(
|
|
'-d',
|
|
'--directory',
|
|
type=str,
|
|
help='The working directory for the agent',
|
|
)
|
|
args, _ = parser.parse_known_args()
|
|
if args.directory:
|
|
config[ConfigType.WORKSPACE_BASE] = os.path.abspath(args.directory)
|
|
print(f"Setting workspace base to {config[ConfigType.WORKSPACE_BASE]}")
|
|
|
|
|
|
parse_arguments()
|
|
|
|
|
|
def finalize_config():
|
|
if config.get(ConfigType.WORKSPACE_MOUNT_REWRITE) and not config.get(ConfigType.WORKSPACE_MOUNT_PATH):
|
|
base = config.get(ConfigType.WORKSPACE_BASE) or os.getcwd()
|
|
parts = config[ConfigType.WORKSPACE_MOUNT_REWRITE].split(':')
|
|
config[ConfigType.WORKSPACE_MOUNT_PATH] = base.replace(parts[0], parts[1])
|
|
|
|
|
|
finalize_config()
|
|
|
|
|
|
def get(key: str, required: bool = False):
|
|
"""
|
|
Get a key from the environment variables or config.toml or default configs.
|
|
"""
|
|
value = config.get(key)
|
|
if not value and required:
|
|
raise KeyError(f"Please set '{key}' in `config.toml` or `.env`.")
|
|
return value
|
|
|
|
|
|
def get_fe_config() -> dict:
|
|
"""
|
|
Get all the frontend configuration values by performing a deep copy.
|
|
"""
|
|
fe_config = copy.deepcopy(config)
|
|
del fe_config['LLM_API_KEY']
|
|
return fe_config
|