avraham
/
context-efficient-ai
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

biome format

Browse Source
master
Avraham Sakal 2 months ago
parent 5e5549e85f
commit a8eba580d0
14 changed files with 143 additions and 102 deletions
Show Stats Download Patch File Download Diff File

2
database/generated/Database.ts
Unescape Escape View File

@ -1,7 +1,7 @@
// @generated
// This file is automatically generated by Kanel. Do not modify manually.
import type { default as PublicSchema } from './public/PublicSchema';
import type { default as PublicSchema } from "./public/PublicSchema";
type Database = PublicSchema;

6
database/generated/public/Conversations.ts
Unescape Escape View File

@ -1,11 +1,11 @@
// @generated
// This file is automatically generated by Kanel. Do not modify manually.
import type { UsersId } from './Users';
import type { ColumnType, Selectable, Insertable, Updateable } from 'kysely';
import type { UsersId } from "./Users";
import type { ColumnType, Selectable, Insertable, Updateable } from "kysely";
/** Identifier type for public.conversations */
export type ConversationsId = number & { __brand: 'public.conversations' };
export type ConversationsId = number & { __brand: "public.conversations" };
/** Represents the table public.conversations */
export default interface ConversationsTable {

22
database/generated/public/FactTriggers.ts
Unescape Escape View File

@ -1,12 +1,12 @@
// @generated
// This file is automatically generated by Kanel. Do not modify manually.
import type { FactsId } from './Facts';
import type { ConversationsId } from './Conversations';
import type { ColumnType, Selectable, Insertable, Updateable } from 'kysely';
import type { FactsId } from "./Facts";
import type { ConversationsId } from "./Conversations";
import type { ColumnType, Selectable, Insertable, Updateable } from "kysely";
/** Identifier type for public.fact_triggers */
export type FactTriggersId = number & { __brand: 'public.fact_triggers' };
export type FactTriggersId = number & { __brand: "public.fact_triggers" };
/** Represents the table public.fact_triggers */
export default interface FactTriggersTable {
@ -18,9 +18,17 @@ export default interface FactTriggersTable {
priority_multiplier: ColumnType<number | null, number | null, number | null>;
priority_multiplier_reason: ColumnType<string | null, string | null, string | null>;
scope_conversation_id: ColumnType<ConversationsId | null, ConversationsId | null, ConversationsId | null>;
priority_multiplier_reason: ColumnType<
string | null,
string | null,
string | null
>;
scope_conversation_id: ColumnType<
ConversationsId | null,
ConversationsId | null,
ConversationsId | null
>;
created_at: ColumnType<string | null, string | null, string | null>;
}

14
database/generated/public/Facts.ts
Unescape Escape View File

@ -1,12 +1,12 @@
// @generated
// This file is automatically generated by Kanel. Do not modify manually.
import type { UsersId } from './Users';
import type { MessagesId } from './Messages';
import type { ColumnType, Selectable, Insertable, Updateable } from 'kysely';
import type { UsersId } from "./Users";
import type { MessagesId } from "./Messages";
import type { ColumnType, Selectable, Insertable, Updateable } from "kysely";
/** Identifier type for public.facts */
export type FactsId = number & { __brand: 'public.facts' };
export type FactsId = number & { __brand: "public.facts" };
/** Represents the table public.facts */
export default interface FactsTable {
@ -14,7 +14,11 @@ export default interface FactsTable {
user_id: ColumnType<UsersId | null, UsersId | null, UsersId | null>;
source_message_id: ColumnType<MessagesId | null, MessagesId | null, MessagesId | null>;
source_message_id: ColumnType<
MessagesId | null,
MessagesId | null,
MessagesId | null
>;
content: ColumnType<string | null, string | null, string | null>;

14
database/generated/public/Messages.ts
Unescape Escape View File

@ -1,18 +1,22 @@
// @generated
// This file is automatically generated by Kanel. Do not modify manually.
import type { ConversationsId } from './Conversations';
import type { default as Role } from './Role';
import type { ColumnType, Selectable, Insertable, Updateable } from 'kysely';
import type { ConversationsId } from "./Conversations";
import type { default as Role } from "./Role";
import type { ColumnType, Selectable, Insertable, Updateable } from "kysely";
/** Identifier type for public.messages */
export type MessagesId = number & { __brand: 'public.messages' };
export type MessagesId = number & { __brand: "public.messages" };
/** Represents the table public.messages */
export default interface MessagesTable {
id: ColumnType<MessagesId, never, never>;
conversation_id: ColumnType<ConversationsId | null, ConversationsId | null, ConversationsId | null>;
conversation_id: ColumnType<
ConversationsId | null,
ConversationsId | null,
ConversationsId | null
>;
index: ColumnType<number | null, number | null, number | null>;

12
database/generated/public/PublicSchema.ts
Unescape Escape View File

@ -1,12 +1,12 @@
// @generated
// This file is automatically generated by Kanel. Do not modify manually.
import type { default as UsersTable } from './Users';
import type { default as MessagesTable } from './Messages';
import type { default as ToolsTable } from './Tools';
import type { default as FactTriggersTable } from './FactTriggers';
import type { default as FactsTable } from './Facts';
import type { default as ConversationsTable } from './Conversations';
import type { default as UsersTable } from "./Users";
import type { default as MessagesTable } from "./Messages";
import type { default as ToolsTable } from "./Tools";
import type { default as FactTriggersTable } from "./FactTriggers";
import type { default as FactsTable } from "./Facts";
import type { default as ConversationsTable } from "./Conversations";
export default interface PublicSchema {
users: UsersTable;

5
database/generated/public/Role.ts
Unescape Escape View File

@ -2,9 +2,6 @@
// This file is automatically generated by Kanel. Do not modify manually.
/** Represents the enum public.role */
type Role =
| 'user'
| 'assistant'
| 'system';
type Role = "user" | "assistant" | "system";
export default Role;

20
database/generated/public/Tools.ts
Unescape Escape View File

@ -1,12 +1,12 @@
// @generated
// This file is automatically generated by Kanel. Do not modify manually.
import type { UsersId } from './Users';
import type { MessagesId } from './Messages';
import type { ColumnType, Selectable, Insertable, Updateable } from 'kysely';
import type { UsersId } from "./Users";
import type { MessagesId } from "./Messages";
import type { ColumnType, Selectable, Insertable, Updateable } from "kysely";
/** Identifier type for public.tools */
export type ToolsId = number & { __brand: 'public.tools' };
export type ToolsId = number & { __brand: "public.tools" };
/** Represents the table public.tools */
export default interface ToolsTable {
@ -14,7 +14,11 @@ export default interface ToolsTable {
user_id: ColumnType<UsersId | null, UsersId | null, UsersId | null>;
source_message_id: ColumnType<MessagesId | null, MessagesId | null, MessagesId | null>;
source_message_id: ColumnType<
MessagesId | null,
MessagesId | null,
MessagesId | null
>;
name: ColumnType<string | null, string | null, string | null>;
@ -22,7 +26,11 @@ export default interface ToolsTable {
parameter_schema: ColumnType<unknown | null, unknown | null, unknown | null>;
implementation_language: ColumnType<string | null, string | null, string | null>;
implementation_language: ColumnType<
string | null,
string | null,
string | null
>;
implementation_code: ColumnType<string | null, string | null, string | null>;

4
database/generated/public/Users.ts
Unescape Escape View File

@ -1,10 +1,10 @@
// @generated
// This file is automatically generated by Kanel. Do not modify manually.
import type { ColumnType, Selectable, Insertable, Updateable } from 'kysely';
import type { ColumnType, Selectable, Insertable, Updateable } from "kysely";
/** Identifier type for public.users */
export type UsersId = number & { __brand: 'public.users' };
export type UsersId = number & { __brand: "public.users" };
/** Represents the table public.users */
export default interface UsersTable {

2
opencode.json
Unescape Escape View File

@ -37,4 +37,4 @@
},
"share": "disabled",
"autoupdate": true
}
}

2
package.json
Unescape Escape View File

@ -63,4 +63,4 @@
"wrangler": "^4.20.5"
},
"type": "module"
}
}

94
pages/chat/fact-triggers.ts
Unescape Escape View File

@ -79,49 +79,61 @@ export const factTriggers = router({
return { ok: true };
}),
generateFromFact: publicProcedure
.input((x) => x as {
previousRunningSummary: string;
messagesSincePreviousRunningSummary: Array<DraftMessage>;
mainResponseContent: string;
fact: Fact;
})
.mutation(async ({ input: { previousRunningSummary, messagesSincePreviousRunningSummary, mainResponseContent, fact } }) => {
const factTriggers = await generateObject<{
factTriggers: Array<string>;
}>({
model: openrouter("mistralai/mistral-nemo"),
messages: [
{
role: "system" as const,
content: factTriggersSystemPrompt({
previousRunningSummary,
messagesSincePreviousRunningSummary,
mainResponseContent,
}),
},
{
role: "user" as const,
content: factTriggersUserPrompt({
factContent: fact.content,
}),
},
],
schema: jsonSchema({
type: "object",
properties: {
factTriggers: {
type: "array",
items: {
type: "string",
},
.input(
(x) =>
x as {
previousRunningSummary: string;
messagesSincePreviousRunningSummary: Array<DraftMessage>;
mainResponseContent: string;
fact: Fact;
},
)
.mutation(
async ({
input: {
previousRunningSummary,
messagesSincePreviousRunningSummary,
mainResponseContent,
fact,
},
}) => {
const factTriggers = await generateObject<{
factTriggers: Array<string>;
}>({
model: openrouter("mistralai/mistral-nemo"),
messages: [
{
role: "system" as const,
content: factTriggersSystemPrompt({
previousRunningSummary,
messagesSincePreviousRunningSummary,
mainResponseContent,
}),
},
{
role: "user" as const,
content: factTriggersUserPrompt({
factContent: fact.content,
}),
},
],
schema: jsonSchema({
type: "object",
properties: {
factTriggers: {
type: "array",
items: {
type: "string",
},
},
}),
// maxSteps: 3,
// tools: undefined,
});
return factTriggers;
})
},
}),
// maxSteps: 3,
// tools: undefined,
});
return factTriggers;
},
),
});
export const createCaller = createCallerFactory(factTriggers);

34
pages/chat/messages.ts
Unescape Escape View File

@ -6,9 +6,7 @@ import {
import { createCaller as createConversationsCaller } from "./conversations.js";
import { openrouter } from "./provider.js";
import { generateObject, generateText, jsonSchema } from "ai";
import type {
DraftMessage,
} from "../../types.js";
import type { DraftMessage } from "../../types.js";
const runningSummarySystemPrompt = ({
previousRunningSummary,
@ -47,7 +45,6 @@ ${mainResponseContent}
Generate a new running summary of the conversation.`;
export const messages = router({
fetchByConversationId: publicProcedure
.input((x) => x as { conversationId: string })
@ -58,13 +55,23 @@ export const messages = router({
});
}),
generateRunningSummary: publicProcedure
.input((x) => x as {
previousRunningSummary: string;
messagesSincePreviousRunningSummary: Array<DraftMessage>;
mainResponseContent: string;
})
.mutation(async ({ input: { previousRunningSummary, messagesSincePreviousRunningSummary, mainResponseContent } }) => {
const runningSummaryResponse = await generateText({
.input(
(x) =>
x as {
previousRunningSummary: string;
messagesSincePreviousRunningSummary: Array<DraftMessage>;
mainResponseContent: string;
},
)
.mutation(
async ({
input: {
previousRunningSummary,
messagesSincePreviousRunningSummary,
mainResponseContent,
},
}) => {
const runningSummaryResponse = await generateText({
model: openrouter("mistralai/mistral-nemo"),
messages: [
{
@ -84,8 +91,9 @@ export const messages = router({
maxSteps: 3,
tools: undefined,
});
return runningSummaryResponse;
}),
return runningSummaryResponse;
},
),
});
export const createCaller = createCallerFactory(messages);

14
pages/chat/trpc.ts
Unescape Escape View File

@ -38,7 +38,6 @@ ${previousRunningSummary}
</running_summary>
`;
export const chat = router({
conversations,
messages,
@ -139,11 +138,12 @@ export const chat = router({
* with the model's response to the database. The new running summary is
* based on the previous running summary combined with the all messages
* since that summary was produced. */
const runningSummaryResponse = await messagesCaller.generateRunningSummary({
messagesSincePreviousRunningSummary,
mainResponseContent: mainResponse.text,
previousRunningSummary,
});
const runningSummaryResponse =
await messagesCaller.generateRunningSummary({
messagesSincePreviousRunningSummary,
mainResponseContent: mainResponse.text,
previousRunningSummary,
});
const insertedAssistantMessage: CommittedMessage = {
id: nanoid(),
conversationId,
@ -193,7 +193,7 @@ export const chat = router({
mainResponseContent: mainResponse.text,
previousRunningSummary,
messagesSincePreviousRunningSummary,
fact
fact,
});
const insertedFactTriggers: Array<FactTrigger> =
factTriggers.object.factTriggers.map((factTrigger) => ({

Write Preview
Loading…
Cancel
Save