fix: Customize summarization model via environment variable

julep-ai · Jun 6, 2024 · ddbf7d0 · ddbf7d0
1 parent 2c4ceb6
commit ddbf7d0
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/agents-api/agents_api/activities/summarization.py b/agents-api/agents_api/activities/summarization.py
@@ -178,7 +178,9 @@ async def summarization(session_id: str) -> None:
         trim_messages(entries, model=summarization_model_name),
         get_entities(entries, model=summarization_model_name),
     )
-    summarized = await summarize_messages(trimmed_messages)
+    summarized = await summarize_messages(
+        trimmed_messages, model=summarization_model_name
+    )
     ts_delta = (entries[1]["timestamp"] - entries[0]["timestamp"]) / 2
     new_entities_entry = Entry(
         session_id=session_id,