Make agent and compaction model IDs configurable via SSM
This commit is contained in:
@@ -10,13 +10,14 @@ import boto3
|
||||
|
||||
from bedrock_agentcore.memory.client import MemoryClient
|
||||
|
||||
import config
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
MEMORY_ID = 'agentclaw_AgentClawMemory-i7Csf776AH'
|
||||
SESSION_WINDOW_SIZE = 100
|
||||
USERS_TABLE_NAME = os.environ.get('USERS_TABLE_NAME', 'agent-claw-users')
|
||||
LTM_SESSION_ID = 'ltm-extractions'
|
||||
HAIKU_MODEL_ID = 'us.anthropic.claude-3-5-haiku-20241022-v1:0'
|
||||
|
||||
_memory_client: MemoryClient | None = None
|
||||
|
||||
@@ -110,7 +111,7 @@ def _call_claude_extraction(conversation_text: str) -> dict:
|
||||
'Conversation:\n' + conversation_text
|
||||
)
|
||||
resp = bedrock.converse(
|
||||
modelId=HAIKU_MODEL_ID,
|
||||
modelId=config.COMPACTION_MODEL_ID,
|
||||
messages=[{'role': 'user', 'content': [{'text': prompt}]}],
|
||||
inferenceConfig={'maxTokens': 1024},
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user