feat: add flavor for session

toeverything · Apr 2, 2024 · c11d872 · c11d872
1 parent 499df4c
commit c11d872
Show file tree

Hide file tree

Showing 5 changed files with 44 additions and 9 deletions.
diff --git a/packages/backend/server/migrations/20240325125057_ai_sessions/migration.sql b/packages/backend/server/migrations/20240325125057_ai_sessions/migration.sql
@@ -1,11 +1,12 @@
 -- CreateTable
 CREATE TABLE "ai_sessions" (
-    "id" VARCHAR NOT NULL,
+    "id" VARCHAR(36) NOT NULL,
     "user_id" VARCHAR NOT NULL,
     "workspace_id" VARCHAR NOT NULL,
     "doc_id" VARCHAR NOT NULL,
     "prompt_name" VARCHAR NOT NULL,
     "action" BOOLEAN NOT NULL,
+    "flavor" VARCHAR NOT NULL,
     "model" VARCHAR NOT NULL,
     "messages" JSON NOT NULL,
     "created_at" TIMESTAMPTZ(6) NOT NULL DEFAULT CURRENT_TIMESTAMP,

diff --git a/packages/backend/server/schema.prisma b/packages/backend/server/schema.prisma
@@ -455,6 +455,9 @@ model AiSession {
   docId       String   @map("doc_id") @db.VarChar
   promptName  String   @map("prompt_name") @db.VarChar
   action      Boolean  @db.Boolean
+  // an mark identifying which view to use to display the session
+  // it is only used in the frontend and does not affect the backend
+  flavor      String   @db.VarChar
   model       String   @db.VarChar
   messages    Json     @db.Json
   createdAt   DateTime @default(now()) @map("created_at") @db.Timestamptz(6)

diff --git a/packages/backend/server/src/plugins/copilot/resolver.ts b/packages/backend/server/src/plugins/copilot/resolver.ts
@@ -33,11 +33,7 @@ import {
 
 registerEnumType(AvailableModels, { name: 'CopilotModel' });
 
-@ObjectType('Copilot')
-export class CopilotType {
-  @Field(() => ID)
-  workspaceId!: string;
-}
+// ================== Input Types ==================
 
 @InputType()
 class CreateChatSessionInput {
@@ -47,13 +43,24 @@ class CreateChatSessionInput {
   @Field(() => String)
   docId!: string;
 
-  @Field(() => Boolean)
+  @Field(() => Boolean, {
+    description: 'Whether the session is for action',
+  })
   action!: boolean;
 
-  @Field(() => String)
+  @Field(() => String, {
+    description: 'An mark identifying which view to use to display the session',
+  })
+  flavor!: string;
+
+  @Field(() => String, {
+    description: 'The model to use for the session',
+  })
   model!: AvailableModel;
 
-  @Field(() => String)
+  @Field(() => String, {
+    description: 'The prompt name to use for the session',
+  })
   promptName!: string;
 }
 
@@ -72,6 +79,8 @@ class QueryChatHistoriesInput implements Partial<ListHistoriesOptions> {
   sessionId: string | undefined;
 }
 
+// ================== Return Types ==================
+
 @ObjectType('ChatMessage')
 class ChatMessageType implements Partial<ChatMessage> {
   @Field(() => String)
@@ -108,6 +117,14 @@ class CopilotQuotaType {
   used!: number;
 }
 
+// ================== Resolver ==================
+
+@ObjectType('Copilot')
+export class CopilotType {
+  @Field(() => ID)
+  workspaceId!: string;
+}
+
 @Resolver(() => CopilotType)
 export class CopilotResolver {
   constructor(

diff --git a/packages/backend/server/src/plugins/copilot/session.ts b/packages/backend/server/src/plugins/copilot/session.ts
@@ -27,6 +27,7 @@ export interface ChatSessionOptions {
   promptName: string;
   // options
   action: boolean;
+  flavor: string;
   model: AvailableModel;
 }
 
@@ -148,6 +149,7 @@ export class ChatSessionService {
       create: {
         id: state.sessionId,
         action: state.action,
+        flavor: state.flavor,
         model: state.model,
         messages: state.messages,
         // connect
@@ -189,6 +191,7 @@ export class ChatSessionService {
           docId: session.docId,
           promptName: session.promptName,
           action: session.action,
+          flavor: session.flavor,
           model: session.model as AvailableModel,
           prompt: await this.prompt.get(session.promptName),
           messages: messages.success ? messages.data : [],

diff --git a/packages/backend/server/src/schema.gql b/packages/backend/server/src/schema.gql
@@ -34,9 +34,17 @@ type CopilotQuota {
 }
 
 input CreateChatSessionInput {
+  """Whether the session is for action"""
   action: Boolean!
   docId: String!
+
+  """An mark identifying which view to use to display the session"""
+  flavor: String!
+
+  """The model to use for the session"""
   model: String!
+
+  """The prompt name to use for the session"""
   promptName: String!
   workspaceId: String!
 }
@@ -300,6 +308,9 @@ type ServerConfigType {
   """credentials requirement"""
   credentialsRequirement: CredentialsRequirementType!
 
+  """enable telemetry"""
+  enableTelemetry: Boolean!
+
   """enabled server features"""
   features: [ServerFeature!]!