zach
/
RA.Aid
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add default support for OpenAI models (#29) 

* add support for default OpenAI provider

* clarify CLI options
This commit is contained in:
Guillermo Creus Botella 2025-01-01 15:34:51 +01:00 committed by GitHub
parent 8e805dba6d
commit 07c71408b7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
ra_aid/__main__.py
View File

@ -34,6 +34,7 @@ logger = get_logger(__name__)
def parse_arguments(args=None): def parse_arguments(args=None):
VALID_PROVIDERS = ['anthropic', 'openai', 'openrouter', 'openai-compatible'] VALID_PROVIDERS = ['anthropic', 'openai', 'openrouter', 'openai-compatible']
ANTHROPIC_DEFAULT_MODEL = 'claude-3-5-sonnet-20241022' ANTHROPIC_DEFAULT_MODEL = 'claude-3-5-sonnet-20241022'
OPENAI_DEFAULT_MODEL = 'gpt-4o'
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(
description='RA.Aid - AI Agent for executing programming and research tasks', description='RA.Aid - AI Agent for executing programming and research tasks',
@ -63,14 +64,14 @@ Examples:
parser.add_argument( parser.add_argument(
'--provider', '--provider',
type=str, type=str,
default='anthropic', default='openai' if (os.getenv('OPENAI_API_KEY') and not os.getenv('ANTHROPIC_API_KEY')) else 'anthropic',
choices=VALID_PROVIDERS, choices=VALID_PROVIDERS,
help='The LLM provider to use' help='The LLM provider to use'
) )
parser.add_argument( parser.add_argument(
'--model', '--model',
type=str, type=str,
help='The model name to use (required for non-Anthropic providers)' help='The model name to use'
) )
parser.add_argument( parser.add_argument(
'--cowboy-mode', '--cowboy-mode',
@ -122,8 +123,10 @@ Examples:
# Validate provider # Validate provider
if parsed_args.provider not in VALID_PROVIDERS: if parsed_args.provider not in VALID_PROVIDERS:
parser.error(f"Invalid provider: {parsed_args.provider}") parser.error(f"Invalid provider: {parsed_args.provider}")
# Handle model defaults and requirements # Handle model defaults and requirements
if parsed_args.provider == "openai":
parsed_args.model = parsed_args.model or OPENAI_DEFAULT_MODEL
if parsed_args.provider == 'anthropic': if parsed_args.provider == 'anthropic':
# Always use default model for Anthropic # Always use default model for Anthropic
parsed_args.model = ANTHROPIC_DEFAULT_MODEL parsed_args.model = ANTHROPIC_DEFAULT_MODEL