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