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

Use proper model config for research agent.

This commit is contained in:
user 2024-12-21 12:53:31 -05:00
parent ae6052ed15
commit 85f56cb75c
2 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ra_aid/__main__.py
View File

@ -227,7 +227,11 @@ def main():
# Store config in global memory for access by is_informational_query # Store config in global memory for access by is_informational_query
_global_memory['config'] = config _global_memory['config'] = config
# Store model configuration
_global_memory['config']['provider'] = args.provider
_global_memory['config']['model'] = args.model
# Store expert provider and model in config # Store expert provider and model in config
_global_memory['config']['expert_provider'] = args.expert_provider _global_memory['config']['expert_provider'] = args.expert_provider
_global_memory['config']['expert_model'] = args.expert_model _global_memory['config']['expert_model'] = args.expert_model

5
ra_aid/tools/agent.py
View File

@ -24,8 +24,9 @@ def request_research(query: str) -> Dict[str, Any]:
- success: Whether completed or interrupted - success: Whether completed or interrupted
- reason: Reason for failure, if any - reason: Reason for failure, if any
""" """
# Initialize model # Initialize model from config
model = initialize_llm("anthropic", "claude-3-sonnet-20240229") config = _global_memory.get('config', {})
model = initialize_llm(config.get('provider', 'anthropic'), config.get('model', 'claude-3-5-sonnet-20241022'))
try: try:
# Run research agent # Run research agent