Merge pull request #348 from UniversityOfHelsinkiCS/langchain-rag

Langchain rag

Author: Veikkosuhonen

src/server/services/langchain/fileSearchResultsStore.ts

@@ -0,0 +1,21 @@
+import type { ChatToolOutput } from '../../../shared/tools'
+import type { User } from '../../../shared/user'
+import { redisClient } from '../../util/redis'
+
+export const ToolResultStore = {
+  getKey(user: User, fileSearchId: string): string {
+    return `user-${user.id}:fileSearch-${fileSearchId}`
+  },
+
+  async saveResults(fileSearchId: string, results: ChatToolOutput, user: User) {
+    const key = this.getKey(user, fileSearchId)
+    await redisClient.set(key, JSON.stringify(results), {
+      EX: 60 * 60 * 24 * 365, // 365 day expiration time
+    })
+  },
+  async getResults(fileSearchId: string, user: User): Promise<ChatToolOutput | null> {
+    const key = this.getKey(user, fileSearchId)
+    const results = (await redisClient.get(key)) as string
+    return results ? JSON.parse(results) : null
+  },
+}

src/shared/chat.ts

@@ -0,0 +1,34 @@
+import type { ChatToolDef } from './tools'
+
+/**
+ * Event emitted when text is added to a chat message
+ */
+export type WritingEvent = {
+  type: 'writing'
+  text: string
+}
+
+export type CompleteEvent = {
+  type: 'complete'
+  prevResponseId: string // Not yet sure what to put here in v3
+}
+
+export type ToolCallStatusEvent = {
+  type: 'toolCallStatus'
+  callId: string
+  toolName: ChatToolDef['name']
+  text: string
+  input?: ChatToolDef['input']
+}
+
+export type ToolCallResultEvent = ToolCallStatusEvent & {
+  input: ChatToolDef['input']
+  result: ChatToolDef['result']
+}
+
+export type ErrorEvent = {
+  type: 'error'
+  error: string
+}
+
+export type ChatEvent = WritingEvent | CompleteEvent | ToolCallStatusEvent | ToolCallResultEvent | ErrorEvent