Method fetchWorkspaceContext

coderd/x/chatd/chatd.go:8940–9013

fetchWorkspaceContext retrieves fresh instruction files and skills from the workspace agent without persisting. It handles agent connection, context configuration fetching, content sanitization, and metadata stamping. Returns the workspace agent, the stamped parts, discovered skills, and whether the

(
	ctx context.Context,
	chat database.Chat,
	getWorkspaceAgent func(context.Context) (database.WorkspaceAgent, error),
	getWorkspaceConn func(context.Context) (workspacesdk.AgentConn, error),
)

// no valid workspace or the agent lookup failed;
// workspaceConnOK is false in that case.
func (p *Server) fetchWorkspaceContext(
	ctx context.Context,
	chat database.Chat,
	getWorkspaceAgent func(context.Context) (database.WorkspaceAgent, error),
	getWorkspaceConn func(context.Context) (workspacesdk.AgentConn, error),
) (agent *database.WorkspaceAgent, agentParts []codersdk.ChatMessagePart, discoveredSkills []chattool.SkillMeta, workspaceConnOK bool) {
	if !chat.WorkspaceID.Valid || getWorkspaceAgent == nil {
		return nil, nil, nil, false
	}

	loadedAgent, agentErr := getWorkspaceAgent(ctx)
	if agentErr != nil {
		return nil, nil, nil, false
	}

	directory := loadedAgent.ExpandedDirectory
	if directory == "" {
		directory = loadedAgent.Directory
	}

	// Fetch context configuration from the agent. Parts
	// arrive pre-populated with context-file and skill entries
	// so we don't need additional round-trips.
	if getWorkspaceConn != nil {
		instructionCtx, cancel := context.WithTimeout(ctx, p.instructionLookupTimeout)
		defer cancel()

		conn, connErr := getWorkspaceConn(instructionCtx)
		if connErr != nil {
			p.logger.Debug(ctx, "failed to resolve workspace connection for instruction files",
				slog.F("chat_id", chat.ID),
				slog.Error(connErr),
			)
		} else {
			workspaceConnOK = true

			agentCfg, cfgErr := conn.ContextConfig(instructionCtx)
			if cfgErr != nil {
				p.logger.Debug(ctx, "failed to fetch context config from agent",
					slog.F("chat_id", chat.ID), slog.Error(cfgErr))
				// Treat a transient ContextConfig failure the
				// same as a failed connection so no sentinel is
				// persisted. The next turn will retry.
				workspaceConnOK = false
			} else {
				agentParts = agentCfg.Parts
			}
		}
	}

	// Stamp server-side fields and sanitize content. The
	// agent cannot know its own UUID, OS metadata, or
	// directory — those are added here at the trust boundary.
	agentID := uuid.NullUUID{UUID: loadedAgent.ID, Valid: true}

	for i := range agentParts {
		agentParts[i].ContextFileAgentID = agentID
		switch agentParts[i].Type {