Skip to content

pref(provider):移除 Provider 透明重连机制#174

Merged
minorcell merged 1 commit into
1024XEngineer:mainfrom
phantom5099:main
Apr 7, 2026
Merged

pref(provider):移除 Provider 透明重连机制#174
minorcell merged 1 commit into
1024XEngineer:mainfrom
phantom5099:main

Conversation

@phantom5099

Copy link
Copy Markdown
Collaborator

变更说明

OpenAI Provider 的 Chat() 方法此前在 SSE 流中途断连时,会自动将已累积的部分 assistant 消息(文本片段 + 未完成的 tool call)注入请求上下文并重新发起请求,实现"透明续传"。该机制在实际使用中导致输出不稳定:注入的半成品消息改变模型注意力分布、provider 与 runtime 双层累积状态不一致、上下文随重连次数膨胀。本次改动将 Chat() 退化为语义清晰的"单次请求-单次响应"模式,流中断时直接返回错误,重试策略完全由上层 runtime 层统一管理。

主要改动

internal/provider/openai/openai.go

  • 简化 Chat():移除重连循环、指数退避、originalMessages 拷贝逻辑,直接内联原 chatOnce() 的 HTTP 请求与流消费流程。失败即返回,不做隐式重试。
  • 删除 chatOnce():合并进 Chat()
  • 简化 consumeStream():移除 accumText/accumCalls 跨周期共享参数,tool call 累积改为方法内部局部变量。
  • 删除 buildAssistantMsg():不再需要构造 assistant 快照用于重连上下文注入。
  • 删除 mergeToolCallDeltaWithAccum()consumeStream 内部直接调用 mergeToolCallDelta()

internal/provider/errors.go

  • 删除 MarkNonRetryable():仅服务于透明重连耗尽后的"防叠加标记",移除后无业务调用方。
  • 删除 IsRecoverableStreamError():仅服务于透明重连循环中的可恢复性判断,移除后无业务调用方。

internal/provider/openai/openai_test.go

  • 删除透明重连相关测试共 11 个函数(TestProviderChatReconnect_*TestBuildAssistantMsg_*TestMergeToolCallDeltaWithAccum_*TestReconnect_*)及辅助函数(countMessagesByRoleroundTripperFunc)。
  • 修复 consumeStream() 签名变更涉及的 3 个测试。

internal/provider/errors_test.go

  • 删除 TestIsRecoverableStreamError_* 共 8 个测试函数。
  • 删除 TestMarkNonRetryable_* 共 4 个测试函数。

docs/provider-api-reference.md

  • 移除已删除的 IsRecoverableStreamError API 条目。

未改动项

  • transport/retry.go (RetryTransport) — HTTP 层 RoundTripper 中间件,仅在首次请求未收到响应时重试 429/5xx,与流中途续传无关,保留不变。
  • ProviderError.Retryable 字段 — 被 RetryTransport 和 runtime 广泛使用,保留不变。
  • runtime.go 重试机制 — 整轮 Chat 调用失败后的完整重试,保留不变。

预期收益

  1. 输出稳定性:消除因注入半成品 assistant 消息导致的模型输出不确定性和重复/矛盾问题。
  2. 语义清晰Chat() 从"可能多次请求的复合操作"回归为"单次请求-单次响应"的基本语义,错误处理路径可预测。
  3. 职责归位:重试策略由 runtime 层统一管理(带完整上下文重建),符合项目模块职责边界要求(Provider 不越权做部分语义重试)。
  4. 代码精简:累计减少约 755 行代码(openai.go ~80 行、openai_test.go ~400 行、errors.go ~60 行、errors_test.go ~215 行)。

@codecov

codecov Bot commented Apr 6, 2026

Copy link
Copy Markdown

Codecov Report

❌ Patch coverage is 66.66667% with 1 line in your changes missing coverage. Please review.

Files with missing lines Patch % Lines
internal/provider/openai/openai.go 66.66% 0 Missing and 1 partial ⚠️

📢 Thoughts on this report? Let us know!

@minorcell minorcell merged commit 4cf078d into 1024XEngineer:main Apr 7, 2026
1 of 2 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants