Merge pull request #4 from palico-ai/context

FEAT: Allowing users to add context to their prompt builder

src/app/handlers/index.ts

@@ -1,16 +1,21 @@
 import { type ChatCompletionTool } from 'openai/resources/chat/completions'
 import { type RequestHandler } from '../types'
 import zodToJsonSchema from 'zod-to-json-schema'
+import { type MessageContext } from '../../types'
+
+// TODO: Add request validation
 
 export interface GetSystemPromptResponseBody {
   prompt: string
 }
 
 export const GetSystemPromptRequestHandler: RequestHandler<
-unknown,
+MessageContext,
 GetSystemPromptResponseBody
-> = async (_, app) => {
-  const prompt = await app.promptBuilder.getSystemPrompt()
+> = async (payload, app) => {
+  const prompt = await app.promptBuilder.getSystemPrompt({
+    context: payload
+  })
   return {
     statusCode: 200,
     body: {
@@ -21,6 +26,7 @@ GetSystemPromptResponseBody
 
 export interface GetPromptRequestBody {
   query: string
+  context: Record<string, unknown>
 }
 
 export interface GetPromptResponseBody {
@@ -31,7 +37,9 @@ export const GetPromptRequestHandler: RequestHandler<
 GetPromptRequestBody,
 GetPromptResponseBody
 > = async (payload, app) => {
-  const prompt = await app.promptBuilder.getPromptForQuery(payload.query)
+  const prompt = await app.promptBuilder.getPromptForQuery(payload.query, {
+    context: payload.context
+  })
   return {
     statusCode: 200,
     body: {

src/app/lambda.ts

@@ -1,10 +1,10 @@
-import { ApplicationConfig } from "../types";
-import CurrentProject from "../utils/current_project";
-import EventHandler from "./request_handler";
+import { type ApplicationConfig } from '../types'
+import CurrentProject from '../utils/current_project'
+import EventHandler from './request_handler'
 
 export const handler = async (event: any): Promise<any> => {
-  const appConfig : ApplicationConfig = await CurrentProject.getApplicationConfig(false);
-  const app = new EventHandler(appConfig);
-  const response = await app.handle(event);
-  return response;
-};
+  const appConfig: ApplicationConfig = await CurrentProject.getApplicationConfig(false)
+  const app = new EventHandler(appConfig)
+  const response = await app.handle(event)
+  return response
+}

src/types.ts

@@ -42,10 +42,16 @@ export interface Toolset {
 }
 
 export interface PromptBuilder {
-  getSystemPrompt: () => Promise<string>
-  getPromptForQuery: (query: string) => Promise<string>
+  getSystemPrompt: (params: {
+    context: MessageContext
+  }) => Promise<string>
+  getPromptForQuery: (query: string, params: {
+    context: MessageContext
+  }) => Promise<string>
 }
 
+export type MessageContext = Record<string, unknown>
+
 export interface ProjectConfig {
   orgId: number
   projectId: number