test(coderd/x/chatd): cover project memory extraction and turn prepar…

…ation Exercises extractProjectMemories end to end against a mocked store and an intercepted model call: cursor fencing, transcript windowing by revision, normalization of upserts and deletes, the per-project cap, and that model failures leave the cursor untouched. Covers prepareGeneration placing the project memory index and tools only on root chats in a project with the experiment enabled.
coder · f0ssel · Sep 11, 2026 · Sep 11, 2026 · Sep 11, 2026 · Sep 11, 2026
commit 61ccfbf22b29547b2cea75fdeba0c4da83c8fb51
diff --git a/coderd/x/chatd/generation_preparer_internal_test.go b/coderd/x/chatd/generation_preparer_internal_test.go
@@ -290,6 +290,165 @@ func TestPrepareGenerationComputerUseIgnoresChatTransportOverride(t *testing.T)
 	require.True(t, sawInlinedText, "attachment was not inlined as text")
 }
 
+func TestPrepareGenerationProjectMemory(t *testing.T) {
+	t.Parallel()
+
+	tests := []struct {
+		name            string
+		project         bool
+		subagent        bool
+		experiments     codersdk.Experiments
+		wantMemoryBlock bool
+		wantMemoryTools bool
+	}{
+		{
+			name:            "RootProjectExperimentEnabled",
+			project:         true,
+			experiments:     codersdk.Experiments{codersdk.ExperimentChatProjects},
+			wantMemoryBlock: true,
+			wantMemoryTools: true,
+		},
+		{
+			name:        "SubagentProject",
+			project:     true,
+			subagent:    true,
+			experiments: codersdk.Experiments{codersdk.ExperimentChatProjects},
+		},
+		{
+			name:        "RootWithoutProject",
+			experiments: codersdk.Experiments{codersdk.ExperimentChatProjects},
+		},
+		{
+			name:        "ExperimentDisabled",
+			project:     true,
+			experiments: codersdk.Experiments{},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			t.Parallel()
+
+			db, ps := dbtestutil.NewDB(t)
+			ctx := chatdTestContext(t)
+			user := dbgen.User(t, db, database.User{})
+			org := dbgen.Organization(t, db, database.Organization{})
+			dbgen.OrganizationMember(t, db, database.OrganizationMember{
+				UserID:         user.ID,
+				OrganizationID: org.ID,
+			})
+			provider := dbgen.AIProviderWithOptionalKey(t, db, database.AIProvider{
+				Type: database.AIProviderTypeOpenai,
+			}, "test-key")
+			modelConfig := dbgen.ChatModelConfig(t, db, database.ChatModelConfig{
+				Model:          "gpt-4o-mini",
+				AIProviderID:   uuid.NullUUID{UUID: provider.ID, Valid: true},
+				OrganizationID: org.ID,
+			}, func(p *database.InsertChatModelConfigParams) {
+				p.Enabled = true
+			})
+
+			projectID := uuid.NullUUID{}
+			if tt.project {
+				project := dbgen.ChatProject(t, db, database.ChatProject{
+					OrganizationID: org.ID,
+					CreatedBy:      user.ID,
+				})
+				projectID = uuid.NullUUID{UUID: project.ID, Valid: true}
+				dbgen.ChatProjectMemory(t, db, database.ChatProjectMemory{
+					ProjectID:      project.ID,
+					OrganizationID: org.ID,
+					CreatedBy:      user.ID,
+					Name:           "release_notes",
+					Type:           database.ChatProjectMemoryTypeProject,
+					Description:    "Durable release process",
+					Body:           "Run the release checklist.",
+				})
+			}
+
+			parentChatID := uuid.NullUUID{}
+			rootChatID := uuid.NullUUID{}
+			if tt.subagent {
+				parent := dbgen.Chat(t, db, database.Chat{
+					OrganizationID:    org.ID,
+					OwnerID:           user.ID,
+					LastModelConfigID: modelConfig.ID,
+				})
+				parentChatID = uuid.NullUUID{UUID: parent.ID, Valid: true}
+				rootChatID = parentChatID
+			}
+			created, err := chatstate.CreateChat(ctx, db, ps, chatstate.CreateChatInput{
+				OrganizationID:    org.ID,
+				OwnerID:           user.ID,
+				ProjectID:         projectID,
+				ParentChatID:      parentChatID,
+				RootChatID:        rootChatID,
+				LastModelConfigID: modelConfig.ID,
+				Title:             "project memory preparation",
+				ClientType:        database.ChatClientTypeApi,
+				InitialMessages: []chatstate.Message{
+					{
+						Role:           database.ChatMessageRoleUser,
+						Content:        mustMarshalText(t, "inspect the release process"),
+						Visibility:     database.ChatMessageVisibilityBoth,
+						ModelConfigID:  uuid.NullUUID{UUID: modelConfig.ID, Valid: true},
+						CreatedBy:      uuid.NullUUID{UUID: user.ID, Valid: true},
+						ContentVersion: chatprompt.CurrentContentVersion,
+					},
+				},
+			})
+			require.NoError(t, err)
+			_, err = db.UpdateUserChatCustomPrompt(ctx, database.UpdateUserChatCustomPromptParams{
+				UserID:           user.ID,
+				ChatCustomPrompt: "Keep responses concise.",
+			})
+			require.NoError(t, err)
+
+			server := newInternalTestServer(
+				t,
+				db,
+				ps,
+				chatprovider.ProviderAPIKeys{},
+				withInternalTestServerExperiments(tt.experiments),
+				withInternalTestServerTransportFactory(&aibridgeTestFactory{}),
+			)
+			prepared, err := server.prepareGeneration(ctx, generationPrepareInput{
+				Chat:     created.Chat,
+				Messages: created.InitialMessages,
+			})
+			require.NoError(t, err)
+			t.Cleanup(prepared.Cleanup)
+
+			var systemPrompt strings.Builder
+			for _, message := range prepared.Prompt {
+				if message.Role != fantasy.MessageRoleSystem {
+					continue
+				}
+				for _, part := range message.Content {
+					if text, ok := part.(fantasy.TextPart); ok {
+						systemPrompt.WriteString(text.Text)
+						systemPrompt.WriteString("\n")
+					}
+				}
+			}
+			gotSystemPrompt := systemPrompt.String()
+			require.Equal(t, tt.wantMemoryBlock, strings.Contains(gotSystemPrompt, "<project-memory>"))
+			if tt.wantMemoryBlock {
+				require.Contains(t, gotSystemPrompt, "- release_notes [project]: Durable release process")
+				require.Less(t, strings.Index(gotSystemPrompt, "<project-memory>"), strings.Index(gotSystemPrompt, "<user-instructions>"))
+			}
+
+			toolNames := make(map[string]bool, len(prepared.Tools))
+			for _, tool := range prepared.Tools {
+				toolNames[tool.Info().Name] = true
+			}
+			for _, name := range []string{"read_project_memory", "save_project_memory", "delete_project_memory"} {
+				require.Equal(t, tt.wantMemoryTools, toolNames[name], name)
+			}
+		})
+	}
+}
+
 func TestPrepareGenerationSubagentUsesOwnerSyntheticAPIKey(t *testing.T) {
 	t.Parallel()