Remove sqlglot

julep-ai · Dec 27, 2024 · 9e82981 · 9e82981
1 parent 40c1a82
commit 9e82981
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/agents-api/agents_api/queries/chat/prepare_chat_context.py b/agents-api/agents_api/queries/chat/prepare_chat_context.py
@@ -2,7 +2,6 @@
 from uuid import UUID
 
 from beartype import beartype
-from sqlglot import parse_one
 
 from ...common.protocol.sessions import ChatContext, make_session
 from ..utils import (
@@ -14,7 +13,7 @@
 T = TypeVar("T")
 
 
-sql_query = parse_one("""
+sql_query = """
 SELECT * FROM 
 (
     SELECT jsonb_agg(u) AS users FROM (
@@ -101,7 +100,7 @@
             session_id = $2 AND 
             session_lookup.participant_type = 'agent'
     ) r
-) AS toolsets""").sql(pretty=True)
+) AS toolsets"""
 
 
 def _transform(d):

diff --git a/drafts/cozo b/drafts/cozo