From a4e49a0cb40aaa54acde89e3dd386b75934364cb Mon Sep 17 00:00:00 2001 From: Diwank Tomer Date: Fri, 16 Aug 2024 16:14:32 -0400 Subject: [PATCH] wip Signed-off-by: Diwank Tomer --- agents-api/agents_api/activities/truncation.py | 1 + agents-api/agents_api/env.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/agents-api/agents_api/activities/truncation.py b/agents-api/agents_api/activities/truncation.py index 06f33d8b6..642cc3c5b 100644 --- a/agents-api/agents_api/activities/truncation.py +++ b/agents-api/agents_api/activities/truncation.py @@ -31,6 +31,7 @@ def get_extra_entries(messages: list[Entry], token_count_threshold: int) -> list @activity.defn @beartype async def truncation(session_id: str, token_count_threshold: int) -> None: + print(session_id, token_count_threshold) session_id = UUID(session_id) # delete_entries( diff --git a/agents-api/agents_api/env.py b/agents-api/agents_api/env.py index 863a19b4f..38ba2101c 100644 --- a/agents-api/agents_api/env.py +++ b/agents-api/agents_api/env.py @@ -66,7 +66,7 @@ temporal_client_cert: str = env.str("TEMPORAL_CLIENT_CERT", default=None) temporal_private_key: str = env.str("TEMPORAL_PRIVATE_KEY", default=None) temporal_endpoint = env.str("TEMPORAL_ENDPOINT", default="localhost:7233") -temporal_task_queue = env.str("TEMPORAL_TASK_QUEUE", default="memory-task-queue") +temporal_task_queue = env.str("TEMPORAL_TASK_QUEUE", default="julep-task-queue") # Consolidate environment variables