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

feat(anthropic_token_limiter.py): add dataclass import for future use and improve code readability by restructuring import statements

This commit is contained in:
Ariel Frischer 2025-03-11 23:37:20 -07:00
parent ee73c85b02
commit a3284c9d7e
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
ra_aid/anthropic_token_limiter.py
View File

@ -2,9 +2,16 @@
from functools import partial from functools import partial
from typing import Any, Dict, List, Optional, Sequence from typing import Any, Dict, List, Optional, Sequence
from dataclasses import dataclass
from langchain_anthropic import ChatAnthropic from langchain_anthropic import ChatAnthropic
from langchain_core.messages import AIMessage, BaseMessage, ToolMessage, trim_messages from langchain_core.messages import (
AIMessage,
BaseMessage,
RemoveMessage,
ToolMessage,
trim_messages,
)
from langchain_core.messages.base import message_to_dict from langchain_core.messages.base import message_to_dict
from ra_aid.anthropic_message_utils import ( from ra_aid.anthropic_message_utils import (
@ -143,6 +150,7 @@ def state_modifier(
print_messages_compact(messages) print_messages_compact(messages)
print("AFTER TRIMMING") print("AFTER TRIMMING")
print_messages_compact(result) print_messages_compact(result)
return result return result