fix: Fix doc listing query

julep-ai · May 23, 2024 · 3a6bbd2 · 3a6bbd2
1 parent 8a1053a
commit 3a6bbd2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/agents-api/agents_api/models/docs/list_docs.py b/agents-api/agents_api/models/docs/list_docs.py
@@ -69,14 +69,13 @@ def list_docs_snippets_by_owner_query(
                 created_at,
                 metadata,
             }},
-            {metadata_filter_str}
-
             *information_snippets {{
                 doc_id,
                 snippet_idx,
                 title,
                 snippet,
-            }}
+            }},
+            {metadata_filter_str}
     }}"""
 
     return (query, {"owner_id": owner_id})
diff --git a/agents-api/agents_api/routers/agents/routers.py b/agents-api/agents_api/routers/agents/routers.py
@@ -368,6 +368,7 @@ async def list_docs(
                 id=row["doc_id"],
                 title=row["title"],
                 content=row["snippet"],
+                metadata=row.get("metadata")
             )
             for _, row in resp.iterrows()
         ]