76 lines
2.3 KiB
TypeScript
76 lines
2.3 KiB
TypeScript
import type { Entity } from "@triplit/client";
|
|
import type schema from "#triplit/schema";
|
|
import { nanoid } from "nanoid";
|
|
import { assert } from "~~/utils/assert";
|
|
|
|
export type Agent = Readonly<Entity<typeof schema, 'agents'> & { topics: Readonly<Entity<typeof schema, 'topics'>>[] }>;
|
|
|
|
export const useAgents = () => {
|
|
const nuxtApp = useNuxtApp();
|
|
const triplit = useTriplitClient();
|
|
|
|
// dont leaking between different users/requests
|
|
if (!nuxtApp._agentsState) {
|
|
nuxtApp._agentsState = {
|
|
list: ref<Agent[]>([]),
|
|
initPromise: null as Promise<void> | null,
|
|
};
|
|
}
|
|
|
|
const state = nuxtApp._agentsState as {
|
|
list: Ref<Agent[]>;
|
|
initPromise: Promise<void> | null;
|
|
};
|
|
|
|
const init = (): Promise<void> => {
|
|
if (state.initPromise) return state.initPromise;
|
|
|
|
state.initPromise = (async () => {
|
|
const query = triplit.query('agents')
|
|
.Include('topics', (rel) => rel('topics').Order('createdAt', 'DESC'))
|
|
.Order('createdAt', 'ASC');
|
|
|
|
const { results } = await useQuery('agents', triplit, query)
|
|
watch(results, (newAgents) => {
|
|
if (newAgents && newAgents.length > 0) {
|
|
state.list.value = newAgents as unknown as Agent[];
|
|
}
|
|
}, { immediate: true, flush: 'sync' });
|
|
})();
|
|
|
|
return state.initPromise;
|
|
};
|
|
|
|
const getAgent = (id: MaybeRef<string>) => {
|
|
return computed(() => state.list.value.find((agent) => agent.id === toRef(id).value) || null);
|
|
}
|
|
|
|
const createAgent = async () => {
|
|
const triplit = useTriplitClient();
|
|
const { user } = useAuth();
|
|
if (!user.value) throw new Error('No user');
|
|
|
|
const id = nanoid();
|
|
await triplit.insert('agents', {
|
|
id,
|
|
name: 'New Agent',
|
|
userId: user.value.id,
|
|
systemPrompt: 'You are a helpful assistant.',
|
|
defaultModelId: null,
|
|
imageUrl: null,
|
|
createdAt: new Date().toISOString(),
|
|
});
|
|
|
|
assert('flush' in triplit);
|
|
await triplit.flush();
|
|
|
|
return state.list.value.find((agent) => agent.id === id)!;
|
|
};
|
|
return {
|
|
init,
|
|
agents: state.list,
|
|
getAgent,
|
|
createAgent
|
|
};
|
|
};
|