fix(proxy): 同步当前 Notion 推理协议

internal/proxy/notion.go

@@ -1275,6 +1275,7 @@ func CallInference(acc *Account, messages []ChatMessage, model string, disableBu
 	session := opt.Session
 
 	var reqBody NotionInferenceRequest
+	createdSource := "ai_module"
 
 	if session != nil && session.TurnCount > 0 {
 		// ── Subsequent turn: partial transcript ──
@@ -1293,31 +1294,30 @@ func CallInference(acc *Account, messages []ChatMessage, model string, disableBu
 			IsPartialTranscript:     true,
 			GenerateTitle:           false,
 			SaveAllThreadOperations: true,
-			SetUnreadState:          false,
 			ThreadType:              "workflow",
-			AsPatchResponse:         false,
-			DebugOverrides: DebugOverrides{
-				Model:                           notionModel,
-				EmitAgentSearchExtractedResults: true,
-			},
 		}
 		log.Printf("[session] subsequent turn %d on thread %s (updated-configs=%d)",
 			session.TurnCount+1, session.ThreadID, len(session.UpdatedConfigIDs))
 	} else {
 		// ── First turn (or legacy single-turn): full transcript ──
-		var configID, contextID, now string
+		if len(attachments) > 0 {
+			createdSource = "workflows"
+		}
+		var configID, contextID, contextPageID, now string
 		if session != nil {
 			// Pre-created session from HandleAnthropicMessages
 			configID = session.ConfigID
 			contextID = session.ContextID
+			contextPageID = ensureSessionContextPageID(session)
 			now = session.OriginalDatetime
 		} else {
 			// Legacy single-turn fallback (e.g. OpenAI-compatible handler)
 			configID = generateUUIDv4()
 			contextID = generateUUIDv4()
+			contextPageID = generateUUIDv4()
 			now = time.Now().Format(time.RFC3339Nano)
 		}
-		transcript := buildFullTranscript(acc, messages, notionModel, disableBuiltinTools, enableWebSearch, opt.EnableWorkspaceSearch, opt.UseReadOnlyMode, attachments, configID, contextID, now)
+		transcript := buildFullTranscript(acc, messages, notionModel, disableBuiltinTools, enableWebSearch, opt.EnableWorkspaceSearch, opt.UseReadOnlyMode, attachments, configID, contextID, contextPageID, now)
 
 		// When attachments are present, reuse the upload thread instead of creating a new one.
 		createThread := true
@@ -1341,13 +1341,7 @@ func CallInference(acc *Account, messages []ChatMessage, model string, disableBu
 			IsPartialTranscript:     false,
 			GenerateTitle:           true,
 			SaveAllThreadOperations: true,
-			SetUnreadState:          false,
 			ThreadType:              "workflow",
-			AsPatchResponse:         false,
-			DebugOverrides: DebugOverrides{
-				Model:                           notionModel,
-				EmitAgentSearchExtractedResults: true,
-			},
 		}
 
 		if createThread {
@@ -1360,6 +1354,7 @@ func CallInference(acc *Account, messages []ChatMessage, model string, disableBu
 
 		log.Printf("[session] first turn, thread %s (session=%v)", threadID, session != nil)
 	}
+	applyWorkflowRequestProtocol(&reqBody, createdSource)
 
 	bodyBytes, err := json.Marshal(reqBody)
 	if err != nil {
@@ -1405,6 +1400,22 @@ func CallInference(acc *Account, messages []ChatMessage, model string, disableBu
 	return parseNDJSONStream(reader, requestID, cb, opt.NativeToolUses, opt.ThinkingBlocks, opt.ThinkingCallback, opt.KnownCitationURLs, opt.KnownCitationDocs, opt.KnownToolCallURLs)
 }
 
+func applyWorkflowRequestProtocol(reqBody *NotionInferenceRequest, createdSource string) {
+	reqBody.SetUnreadState = true
+	reqBody.CreatedSource = createdSource
+	reqBody.AsPatchResponse = true
+	reqBody.PatchResponseVersion = 2
+	reqBody.IsUserInAnySalesAssistedSpace = boolPtr(false)
+	reqBody.IsSpaceSalesAssisted = boolPtr(false)
+	reqBody.SupportsCustomAgentNudgeTranscriptStep = boolPtr(true)
+	reqBody.DebugOverrides = DebugOverrides{
+		EmitAgentSearchExtractedResults: true,
+		CachedInferences:                &struct{}{},
+		AnnotationInferences:            &struct{}{},
+		EmitInferences:                  boolPtr(false),
+	}
+}
+
 // buildConfigValue constructs the Notion config value map used in transcript config entries.
 // enableWorkspaceSearch: nil = use AppConfig default, non-nil = per-request override
 // useReadOnlyMode: when true, sets Notion's ASK-mode flag — model answers
@@ -1423,21 +1434,63 @@ func buildConfigValue(notionModel string, disableBuiltinTools bool, enableWebSea
 	agentEnabled := !effectiveDisable || wsSearch
 
 	configValue := map[string]interface{}{
-		"type":                       "workflow",
-		"model":                      notionModel,
-		"modelFromUser":              !isSubsequentTurn,
-		"enableAgentAutomations":     agentEnabled,
-		"enableAgentIntegrations":    agentEnabled,
-		"enableCustomAgents":         !effectiveDisable,
-		"enableAgentDiffs":           !effectiveDisable,
-		"enableCsvAttachmentSupport": true,
-		"enableScriptAgent":          !effectiveDisable,
-		"enableCreateAndRunThread":   true,
-		"useWebSearch":               enableWebSearch,
-		"useReadOnlyMode":            useReadOnlyMode,
-		"writerMode":                 false,
-		"isCustomAgent":              false,
-		"isCustomAgentBuilder":       false,
+		"type":                                           "workflow",
+		"enableAgentAutomations":                         agentEnabled,
+		"enableAgentIntegrations":                        agentEnabled,
+		"enableCustomAgents":                             !effectiveDisable,
+		"enableExperimentalIntegrations":                 false,
+		"enableScriptAgent":                              !effectiveDisable,
+		"enableScriptAgentAdvanced":                      false,
+		"enableScriptAgentSearchConnectorsInCustomAgent": false,
+		"enableScriptAgentGoogleDriveInCustomAgent":      false,
+		"enableScriptAgentGoogleDriveOAuthInCustomAgent": false,
+		"enableScriptAgentSlack":                         !effectiveDisable,
+		"enableScriptAgentMcpServers":                    !effectiveDisable,
+		"enableAgentDiffs":                               !effectiveDisable,
+		"enableCsvAttachmentSupport":                     true,
+		"showDatabaseAgentsDiscoverability":              true,
+		"enableAgentThreadTools":                         false,
+		"enableCrdtOperations":                           false,
+		"enableAgentCardCustomization":                   true,
+		"enableSystemPromptAsPage":                       false,
+		"enableUserSessionContext":                       false,
+		"enableLargeToolResultComputerOffload":           false,
+		"enableScriptAgentGtm":                           false,
+		"enablePitCrewTableViewTool":                     false,
+		"enableComputer":                                 !effectiveDisable,
+		"enableCustomAgentCreateGuidanceV2":              true,
+		"enableSoftwareFactoryPage":                      false,
+		"enableAgentGenerateImage":                       !effectiveDisable,
+		"enableQueryCalendar":                            false,
+		"enableQueryMail":                                false,
+		"enableMailExplicitToolCalls":                    true,
+		"enableMailNotificationPreferences":              false,
+		"enableMailAgentMultiProviderSupport":            true,
+		"enableNotionMailDeprecated":                     false,
+		"enableWebResearch":                              false,
+		"useRulePrioritization":                          true,
+		"useWebSearch":                                   enableWebSearch,
+		"isHipaa":                                        false,
+		"internetAccess":                                 false,
+		"manageWorkers":                                  false,
+		"useReadOnlyMode":                                useReadOnlyMode,
+		"writerMode":                                     false,
+		"model":                                          notionModel,
+		"modelFromUser":                                  true,
+		"isCustomAgent":                                  false,
+		"isCustomAgentBuilder":                           false,
+		"isCustomAgentCreate":                            false,
+		"isAgentResearchRequest":                         false,
+		"useCustomAgentDraft":                            false,
+		"enableMarkdownVNext":                            false,
+		"enableAgentSkillsV2":                            false,
+		"updatePageStaleViewGuardEnabled":                false,
+		"enableUpdatePageOrderUpdates":                   true,
+		"enableAgentSupportPropertyReorder":              true,
+		"enableAgentAskSurvey":                           true,
+		"databaseAgentConfigMode":                        false,
+		"isOnboardingAgent":                              false,
+		"isMobile":                                       false,
 	}
 
 	// searchScopes controls what the built-in search tool can access
@@ -1450,14 +1503,17 @@ func buildConfigValue(notionModel string, disableBuiltinTools bool, enableWebSea
 	}
 
 	if isSubsequentTurn {
-		configValue["isThreadStartedByAdmin"] = true
+		configValue["useContextualCoreDocsAutoLoad"] = false
+		configValue["useDocPreviewsForCoreAutoLoad"] = true
+	} else {
+		configValue["availableConnectors"] = []interface{}{}
 	}
 
 	return configValue
 }
 
 // buildContextValue constructs the Notion context value map used in transcript context entries.
-func buildContextValue(acc *Account, datetime string) map[string]interface{} {
+func buildContextValue(acc *Account, datetime, contextPageID string) map[string]interface{} {
 	return map[string]interface{}{
 		"timezone":        acc.Timezone,
 		"userName":        acc.UserName,
@@ -1468,15 +1524,17 @@ func buildContextValue(acc *Account, datetime string) map[string]interface{} {
 		"spaceViewId":     acc.SpaceViewID,
 		"currentDatetime": datetime,
 		"surface":         "ai_module",
+		"agentAccessory":  "paprika",
+		"context_page_id": contextPageID,
 	}
 }
 
 // buildFullTranscript builds a complete transcript for the first turn of a conversation.
 // Uses ResearcherTranscriptMsg (with id field) to match Notion's real client format.
-func buildFullTranscript(acc *Account, messages []ChatMessage, notionModel string, disableBuiltinTools bool, enableWebSearch bool, enableWorkspaceSearch *bool, useReadOnlyMode bool, attachments []UploadedAttachment, configID, contextID, now string) []interface{} {
+func buildFullTranscript(acc *Account, messages []ChatMessage, notionModel string, disableBuiltinTools bool, enableWebSearch bool, enableWorkspaceSearch *bool, useReadOnlyMode bool, attachments []UploadedAttachment, configID, contextID, contextPageID, now string) []interface{} {
 	hasAttachments := len(attachments) > 0
 	configValue := buildConfigValue(notionModel, disableBuiltinTools, enableWebSearch, enableWorkspaceSearch, useReadOnlyMode, hasAttachments, false)
-	contextValue := buildContextValue(acc, now)
+	contextValue := buildContextValue(acc, now, contextPageID)
 
 	if hasAttachments {
 		contextValue["surface"] = "workflows"
@@ -1555,7 +1613,11 @@ func buildFullTranscript(acc *Account, messages []ChatMessage, notionModel strin
 // It includes: config + context (reused IDs) + N updated-config placeholders + new user message.
 func buildPartialTranscript(acc *Account, newUserContent string, notionModel string, disableBuiltinTools bool, enableWebSearch bool, enableWorkspaceSearch *bool, useReadOnlyMode bool, session *Session) []interface{} {
 	configValue := buildConfigValue(notionModel, disableBuiltinTools, enableWebSearch, enableWorkspaceSearch, useReadOnlyMode, false, true)
-	contextValue := buildContextValue(acc, session.OriginalDatetime)
+	contextPageID := ensureSessionContextPageID(session)
+	contextValue := buildContextValue(acc, session.OriginalDatetime, contextPageID)
+	currentDatetime := time.Now().Format(time.RFC3339Nano)
+	currentContextValue := buildContextValue(acc, currentDatetime, contextPageID)
+	currentContextValue["surface"] = "full_page_chat"
 
 	transcript := []interface{}{
 		ResearcherTranscriptMsg{
@@ -1568,6 +1630,11 @@ func buildPartialTranscript(acc *Account, newUserContent string, notionModel str
 			Type:  "context",
 			Value: contextValue,
 		},
+		ResearcherTranscriptMsg{
+			ID:    generateUUIDv4(),
+			Type:  "context",
+			Value: currentContextValue,
+		},
 	}
 
 	// Add updated-config placeholders for each previous turn
@@ -1584,12 +1651,19 @@ func buildPartialTranscript(acc *Account, newUserContent string, notionModel str
 		Type:      "user",
 		Value:     [][]string{{newUserContent}},
 		UserID:    acc.UserID,
-		CreatedAt: time.Now().Format(time.RFC3339Nano),
+		CreatedAt: currentDatetime,
 	})
 
 	return transcript
 }
 
+func ensureSessionContextPageID(session *Session) string {
+	if session.ContextPageID == "" {
+		session.ContextPageID = generateUUIDv4()
+	}
+	return session.ContextPageID
+}
+
 func setNotionHeaders(req *http.Request, acc *Account) {
 	// Content negotiation
 	req.Header.Set("Content-Type", "application/json")
@@ -2309,6 +2383,7 @@ func parseNDJSONStream(reader io.Reader, requestID string, cb StreamCallback, na
 	patchValueTypes := make(map[string]string)
 	// Counter: "/s/N" → how many value entries added so far
 	patchValueCounts := make(map[string]int)
+	patchNextStepIndex := 0
 	// Accumulated thinking content from patch operations
 	var patchThinkingContent string
 	var patchThinkingSignature string
@@ -2392,6 +2467,37 @@ func parseNDJSONStream(reader io.Reader, requestID string, cb StreamCallback, na
 		sentClean = cleaned
 	}
 
+	handlePatchValueEntry := func(statePrefix string, index int, entry AgentValueEntry) {
+		patchValueTypes[fmt.Sprintf("%s/value/%d", statePrefix, index)] = entry.Type
+		if patchValueCounts[statePrefix] <= index {
+			patchValueCounts[statePrefix] = index + 1
+		}
+
+		switch entry.Type {
+		case "thinking":
+			prev := patchThinkingContent
+			patchThinkingContent += entry.Content
+			if entry.Signature != "" {
+				patchThinkingSignature = entry.Signature
+				lastThinkingSignature = entry.Signature
+			}
+			emitThinking(incrementalSuffix(prev, patchThinkingContent))
+		case "text":
+			if entry.Content != "" {
+				rawText += entry.Content
+				emitDelta()
+			}
+		case "tool_use":
+			// Skip — tool state must not pollute text output.
+			if entry.Name != "" && entry.ID != "" && !seenNativeToolUseIDs[entry.ID] {
+				seenNativeToolUseIDs[entry.ID] = true
+				if nativeToolUses != nil {
+					*nativeToolUses = append(*nativeToolUses, entry)
+				}
+			}
+		}
+	}
+
 	for scanner.Scan() {
 		line := strings.TrimSpace(scanner.Text())
 		if line == "" {
@@ -2635,23 +2741,42 @@ func parseNDJSONStream(reader io.Reader, requestID string, cb StreamCallback, na
 				cb("", true, &totalUsage)
 			}
 
+		case "patch-start":
+			var patchStart struct {
+				Data struct {
+					Steps []json.RawMessage `json:"s"`
+				} `json:"data"`
+			}
+			if json.Unmarshal([]byte(line), &patchStart) == nil {
+				patchNextStepIndex = len(patchStart.Data.Steps)
+			}
+
 		case "patch":
 			var patch PatchEvent
 			if err := json.Unmarshal([]byte(line), &patch); err != nil {
 				continue
 			}
 			for _, op := range patch.V {
+				if op.O == "a" && op.P == "/s/-" {
+					stepIndex := patchNextStepIndex
+					patchNextStepIndex++
+
+					var step AgentInferenceEvent
+					if json.Unmarshal(op.V, &step) == nil && step.Type == "agent-inference" {
+						statePrefix := fmt.Sprintf("/s/%d", stepIndex)
+						for idx, entry := range step.Value {
+							handlePatchValueEntry(statePrefix, idx, entry)
+						}
+					}
+				}
+
 				// Track value entry types when new entries are added
 				if op.O == "a" && strings.Contains(op.P, "/value/-") {
-					var entry struct {
-						Type string `json:"type"`
-					}
-					if json.Unmarshal(op.V, &entry) == nil && entry.Type != "" {
+					var entry AgentValueEntry
+					if json.Unmarshal(op.V, &entry) == nil {
 						statePrefix := op.P[:strings.Index(op.P, "/value/")]
 						idx := patchValueCounts[statePrefix]
-						path := fmt.Sprintf("%s/value/%d", statePrefix, idx)
-						patchValueTypes[path] = entry.Type
-						patchValueCounts[statePrefix] = idx + 1
+						handlePatchValueEntry(statePrefix, idx, entry)
 					}
 				}
 
@@ -2666,12 +2791,14 @@ func parseNDJSONStream(reader io.Reader, requestID string, cb StreamCallback, na
 				}
 
 				// Handle finishedAt for thinking → flush thinking block
-				if op.O == "a" && strings.Contains(op.P, "/finishedAt") && strings.Contains(op.P, "/value/") {
-					if thinkingBlocks != nil && patchThinkingContent != "" {
-						*thinkingBlocks = append(*thinkingBlocks, ThinkingBlock{
-							Content:   patchThinkingContent,
-							Signature: patchThinkingSignature,
-						})
+				if op.O == "a" && strings.HasSuffix(op.P, "/finishedAt") {
+					if patchThinkingContent != "" {
+						if thinkingBlocks != nil {
+							*thinkingBlocks = append(*thinkingBlocks, ThinkingBlock{
+								Content:   patchThinkingContent,
+								Signature: patchThinkingSignature,
+							})
+						}
 						patchThinkingContent = ""
 						patchThinkingSignature = ""
 					}

internal/proxy/session.go

@@ -23,6 +23,9 @@ type Session struct {
 	ConfigID  string
 	ContextID string
 
+	// ContextPageID is generated independently from the transcript IDs and reused across turns.
+	ContextPageID string
+
 	// Each completed turn produces one updated-config placeholder ID
 	UpdatedConfigIDs []string
 

internal/proxy/types.go

@@ -150,19 +150,24 @@ type UsageInfo struct {
 // ========== Notion API Types ==========
 
 type NotionInferenceRequest struct {
-	TraceID                 string               `json:"traceId"`
-	SpaceID                 string               `json:"spaceId"`
-	ThreadID                string               `json:"threadId,omitempty"`
-	Transcript              []interface{}        `json:"transcript"`
-	CreateThread            bool                 `json:"createThread"`
-	GenerateTitle           bool                 `json:"generateTitle"`
-	SaveAllThreadOperations bool                 `json:"saveAllThreadOperations"`
-	SetUnreadState          bool                 `json:"setUnreadState"`
-	ThreadType              string               `json:"threadType"`
-	AsPatchResponse         bool                 `json:"asPatchResponse"`
-	IsPartialTranscript     bool                 `json:"isPartialTranscript"`
-	ThreadParentPointer     *ThreadParentPointer `json:"threadParentPointer,omitempty"`
-	DebugOverrides          DebugOverrides       `json:"debugOverrides"`
+	TraceID                                string               `json:"traceId"`
+	SpaceID                                string               `json:"spaceId"`
+	ThreadID                               string               `json:"threadId,omitempty"`
+	Transcript                             []interface{}        `json:"transcript"`
+	CreateThread                           bool                 `json:"createThread"`
+	GenerateTitle                          bool                 `json:"generateTitle"`
+	SaveAllThreadOperations                bool                 `json:"saveAllThreadOperations"`
+	SetUnreadState                         bool                 `json:"setUnreadState"`
+	ThreadType                             string               `json:"threadType"`
+	CreatedSource                          string               `json:"createdSource,omitempty"`
+	AsPatchResponse                        bool                 `json:"asPatchResponse"`
+	PatchResponseVersion                   int                  `json:"patchResponseVersion,omitempty"`
+	IsPartialTranscript                    bool                 `json:"isPartialTranscript"`
+	IsUserInAnySalesAssistedSpace          *bool                `json:"isUserInAnySalesAssistedSpace,omitempty"`
+	IsSpaceSalesAssisted                   *bool                `json:"isSpaceSalesAssisted,omitempty"`
+	SupportsCustomAgentNudgeTranscriptStep *bool                `json:"supportsCustomAgentNudgeTranscriptStep,omitempty"`
+	ThreadParentPointer                    *ThreadParentPointer `json:"threadParentPointer,omitempty"`
+	DebugOverrides                         DebugOverrides       `json:"debugOverrides"`
 }
 
 // ThreadParentPointer identifies the parent of a thread (used only on first turn)
@@ -194,8 +199,10 @@ type ResearcherTranscriptMsg struct {
 }
 
 type DebugOverrides struct {
-	Model                           string `json:"model,omitempty"`
-	EmitAgentSearchExtractedResults bool   `json:"emitAgentSearchExtractedResults,omitempty"`
+	EmitAgentSearchExtractedResults bool      `json:"emitAgentSearchExtractedResults,omitempty"`
+	CachedInferences                *struct{} `json:"cachedInferences,omitempty"`
+	AnnotationInferences            *struct{} `json:"annotationInferences,omitempty"`
+	EmitInferences                  *bool     `json:"emitInferences,omitempty"`
 }
 
 type NDJSONEvent struct {