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

get model from config

This commit is contained in:
AI Christianson 2025-03-15 22:02:05 -04:00
parent c848c04ee3
commit 1dc9326154
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
ra_aid/server/api_v1_spawn_agent.py
View File

@ -17,10 +17,10 @@ from ra_aid.database.repositories.research_note_repository import ResearchNoteRe
from ra_aid.database.repositories.related_files_repository import RelatedFilesRepositoryManager from ra_aid.database.repositories.related_files_repository import RelatedFilesRepositoryManager
from ra_aid.database.repositories.trajectory_repository import TrajectoryRepositoryManager from ra_aid.database.repositories.trajectory_repository import TrajectoryRepositoryManager
from ra_aid.database.repositories.work_log_repository import WorkLogRepositoryManager from ra_aid.database.repositories.work_log_repository import WorkLogRepositoryManager
from ra_aid.database.repositories.config_repository import ConfigRepositoryManager from ra_aid.database.repositories.config_repository import ConfigRepositoryManager, get_config_repository
from ra_aid.env_inv_context import EnvInvManager from ra_aid.env_inv_context import EnvInvManager
from ra_aid.env_inv import EnvDiscovery from ra_aid.env_inv import EnvDiscovery
from ra_aid.database import ensure_migrations_applied from ra_aid.llm import initialize_llm
# Create logger # Create logger
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -133,10 +133,18 @@ def run_agent_thread(
# Import here to avoid circular imports # Import here to avoid circular imports
from ra_aid.__main__ import run_research_agent from ra_aid.__main__ import run_research_agent
# Get the provider and model from config repository
provider = get_config_repository().get("provider", "anthropic")
model_name = get_config_repository().get("model", "claude-3-7-sonnet-20250219")
temperature = get_config_repository().get("temperature")
# Initialize model with provider and model name from config
model = initialize_llm(provider, model_name, temperature=temperature)
# Run the research agent # Run the research agent
run_research_agent( run_research_agent(
base_task_or_query=message, base_task_or_query=message,
model=None, # Use default model model=model, # Use the initialized model from config
expert_enabled=expert_enabled, expert_enabled=expert_enabled,
research_only=research_only, research_only=research_only,
hil=False, # No human-in-the-loop for API hil=False, # No human-in-the-loop for API