types: small tweaks

Author: zAlweNy26

compose.dev.yml

@@ -34,4 +34,4 @@ services:
     container_name: ccat_ts_cache_dev
     expose:
       - 6379
-    restart: unless-stopped
+    restart: unless-stopped

compose.yml

@@ -36,4 +36,4 @@ services:
     container_name: ccat_ts_cache
     expose:
       - 6379
-    restart: unless-stopped
+    restart: unless-stopped

src/context.ts

@@ -40,12 +40,19 @@ export const swaggerTags = {
 
 const jsonLiterals = t.Union([t.String(), t.Number(), t.Boolean(), t.Null()])
 
-// TODO: Wait until https://github.com/elysiajs/elysia/issues/848 is fixed, to use t.Intersect()
-export const messageInput = t.Record(t.String(), t.Any(), {
+export const messageInput = t.Intersect([
+	t.Object({
+		text: t.String(),
+	}),
+	t.Record(t.String(), t.Any()),
+], {
 	$id: 'messageInput',
 	title: 'Message Input',
 	description: 'Message to pass to the cat',
 	default: { text: 'Hello, world' },
+	examples: [{
+		text: 'Hello, world',
+	}],
 })
 
 export const memoryMessage = t.Object({
@@ -60,15 +67,15 @@ export const memoryMessage = t.Object({
 			input: t.Union([t.String(), t.Null()]),
 			observation: t.String(),
 		})),
-		memory: t.Optional(t.Intersect([
+		memory: t.Intersect([
 			t.Object({
 				episodic: t.Array(t.Record(t.String(), t.Any())),
 				declarative: t.Array(t.Record(t.String(), t.Any())),
 				procedural: t.Array(t.Record(t.String(), t.Any())),
 			}),
 			t.Record(t.String(), t.Array(t.Record(t.String(), t.Any()))),
-		])),
-		interactions: t.Optional(t.Array(t.Record(t.String(), t.Any()))),
+		]),
+		interactions: t.Array(t.Record(t.String(), t.Any())),
 	})),
 }, {
 	$id: 'memoryMessage',

src/dtos/message.ts

@@ -65,18 +65,22 @@ export type ModelInteraction = LLMInteraction | EmbedderInteraction
 /**
  * The content of a memory message.
  */
-export interface MemoryMessage {
+export type MemoryMessage = {
 	role: 'AI' | 'User'
 	what: string
 	who: string
 	when: number
+} & ({
+	role: 'AI'
 	why?: {
 		input: string
 		intermediateSteps: IntermediateStep[]
-		memory?: WorkingMemory
-		interactions?: ModelInteraction[]
+		memory: WorkingMemory
+		interactions: ModelInteraction[]
 	}
-}
+} | {
+	role: 'User'
+})
 
 /**
  * A message object sent by the user.

src/looking_glass/stray-cat.ts

@@ -148,6 +148,14 @@ export class StrayCat {
 
 		const response = this.userMessage = await madHatter.executeHook('beforeReadMessage', msg, this)
 
+		if (!('text' in response)) {
+			log.warn('The message does not contain any text. Ignoring it...')
+			return {
+				type: 'notification',
+				content: 'The message does not contain any text. Ignoring it...',
+			}
+		}
+
 		// FEATURE: Find another way to handle this
 		if (response.text.length > cheshireCat.embedderSize) {
 			log.warn(`The input is too long. Storing it as document...`)
@@ -206,7 +214,7 @@ export class StrayCat {
 
 		if (save) this.chatHistory.push(structuredClone(finalOutput))
 
-		if (!returnWhy) delete finalOutput.why
+		if (!returnWhy && finalOutput.role === 'AI') delete finalOutput.why
 
 		this.modelsInteractions = []
 

src/routes/general.ts

@@ -76,7 +76,7 @@ export const generalRoutes = new Elysia({
 	},
 }).post('/chat', async ({ stray, body, query }) => {
 	const { save, why } = query
-	const res = await stray.run(body as any, save, why)
+	const res = await stray.run(body, save, why)
 	return res
 }, {
 	body: 'messageInput',