WIP

julep-ai · Dec 18, 2024 · 987fbdc · 987fbdc
1 parent f30569b
commit 987fbdc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/agents-api/agents_api/queries/chat/prepare_chat_context.py b/agents-api/agents_api/queries/chat/prepare_chat_context.py
@@ -7,6 +7,7 @@
 from ..session.prepare_session_data import prepare_session_data
 from ..utils import (
     wrap_in_class,
+    pg_query,
 )
 
 ModelT = TypeVar("ModelT", bound=Any)
@@ -64,7 +65,7 @@
         ],
     },
 )
-@cozo_query
+@pg_query
 @beartype
 def prepare_chat_context(
     *,