scriptoriumadmin/supermemory
1
0
Fork 0
mirror of https://github.com/supermemoryai/supermemory.git synced 2026-09-08 22:21:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

include metadata in response, add type to metadata

Browse source
This commit is contained in:
Dhravya 2024-06-16 12:25:36 -05:00
parent 2ffb0542eb
commit a9f5ebc9e4
4 changed files with 24 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
apps/cf-ai-backend/src/helper.ts
View file

@ -124,6 +124,7 @@ export async function batchCreateChunksAndEmbeddings({
space: body.space ?? "",
url: body.url,
user: body.user,
type: body.type ?? "page",
},
},
],

4
apps/cf-ai-backend/src/index.ts
View file

@ -180,7 +180,9 @@ app.post(
idsAsStrings.map(async (id) => await c.env.KV.get(id)),
);
return c.json({ ids: storedContent });
const metadata = normalizedData.map((datapoint) => datapoint.metadata);
return c.json({ ids: storedContent, metadata });
}
const preparedContext = normalizedData.map(

1
apps/cf-ai-backend/src/types.ts
View file

@ -46,4 +46,5 @@ export const vectorObj = z.object({
space: z.string().optional(),
url: z.string(),
user: z.string(),
type: z.string().optional(),
});

20
apps/web/app/(dash)/chat/chatWindow.tsx
View file

@ -20,6 +20,8 @@ import rehypeKatex from "rehype-katex";
import rehypeHighlight from "rehype-highlight";
import { code, p } from "./markdownRenderHelpers";
import { codeLanguageSubset } from "@/lib/constants";
import { z } from "zod";
import { toast } from "sonner";
function ChatWindow({
q,
@ -56,7 +58,23 @@ function ChatWindow({
// TODO: handle this properly
const sources = await sourcesFetch.json();
console.log(sources);
const sourcesZod = z.object({
ids: z.array(z.string()),
metadata: z.array(z.any()),
});
const sourcesParsed = sourcesZod.safeParse(sources);
if (!sourcesParsed.success) {
console.log(sources);
console.error(sourcesParsed.error);
toast.error("Something went wrong while getting the sources");
return;
}
console.log(sourcesParsed.data.ids);
console.log(sourcesParsed.data.metadata);
const resp = await fetch(`/api/chat?q=${query}&spaces=${spaces}`, {
method: "POST",