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

Merge pull request #137 from ariel-frischer/use-correct-37-sonnet-state-modifier 

Use correct state_modifier when using openrouter claude 3.7
This commit is contained in:
Ariel Frischer 2025-03-15 09:54:27 -07:00 committed by GitHub
parent cde8eee4fa 6c159d39d4
commit 5d07a7f7b8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ra_aid/agent_utils.py
View File

@ -52,6 +52,7 @@ from ra_aid.database.repositories.human_input_repository import (
from ra_aid.database.repositories.trajectory_repository import get_trajectory_repository
from ra_aid.database.repositories.config_repository import get_config_repository
from ra_aid.anthropic_token_limiter import (
get_model_name_from_chat_model,
sonnet_35_state_modifier,
state_modifier,
get_model_token_limit,
@ -102,11 +103,10 @@ def build_agent_kwargs(
):
def wrapped_state_modifier(state: AgentState) -> list[BaseMessage]:
if not hasattr(model, 'model'):
return state_modifier(state, model, max_input_tokens=max_input_tokens)
model_name = get_model_name_from_chat_model(model)
if any(
pattern in model.model
pattern in model_name
for pattern in ["claude-3.5", "claude3.5", "claude-3-5"]
):
return sonnet_35_state_modifier(