fix: Fix job ID

julep-ai · Sep 13, 2024 · 38421e3 · 38421e3
1 parent 9118d86
commit 38421e3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/agents-api/agents_api/routers/sessions/chat.py b/agents-api/agents_api/routers/sessions/chat.py
@@ -99,11 +99,12 @@ async def chat(
     if isinstance(model_response, ModelResponse):
         total_tokens = model_response.usage.total_tokens
 
+    job_id = uuid4()
+
     if (
         chat_context.session.token_budget is not None
         and total_tokens >= chat_context.session.token_budget
     ):
-        job_id = uuid4()
         if chat_context.session.context_overflow == "adaptive":
             await run_summarization_task(session_id=session_id, job_id=job_id)
         elif chat_context.session.context_overflow == "truncate":