zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-08-04 23:00:30 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
288be805f150a6193167febf3dc12fc3045d960c
ragflow/internal/entity/models
History
jay77721 288be805f1 feat(go-models): migrate nvidia/siliconflow/groq/longcat to unified h… (#17643) 
## Summary

Relate to #17284

Migrate four OpenAI-compatible drivers (`nvidia`, `siliconflow`, `groq`,
`longcat`) to use the unified response handlers
(`HandleNonStreamingResponse` / `HandleStreamingResponse`), following
the same pattern established by `deepseek` in #17634.

- Cut ~100 lines per driver (541 lines removed, 19 added across 4
files).
- `nvidia` now extracts token usage (previously had none).
- All four drivers produce the unified `StreamUsage` log.
2026-08-03 10:05:25 +08:00
..
302ai_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
302ai.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
aliyun_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
aliyun.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
anthropic_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
anthropic.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
astraflow_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
astraflow.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
avian_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
avian.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
azure_openai.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
baichuan.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
baidu.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
base_model.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
bedrock_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
bedrock.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
builtin.go
Go: add context to lots of interface (#17253)
2026-07-22 22:30:57 +08:00
chat_tools.go
Go: add context to lots of interface (#17253)
2026-07-22 22:30:57 +08:00
chat_usage_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
cohere_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
cohere.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
cometapi_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
cometapi.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
common_test.go
feat(go-models): unify OpenAI-compatible token usage extraction (#17634)
2026-07-31 18:32:07 +08:00
common.go
fix(go-models): normalize Qwen reasoning families (#15735)
2026-06-08 19:32:19 +08:00
deepinfra_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
deepinfra.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
deepseek_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
deepseek.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
dummy.go
Go: add context to lots of interface (#17253)
2026-07-22 22:30:57 +08:00
factory.go
feat: add GreenPT model provider (#17447)
2026-07-28 19:19:00 +08:00
fishaudio_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
fishaudio.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
funasr_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
funasr.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
futurmix.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
get_driver_test.go
fix: docx/email parsing, extractor LLM driver, and chunker alignment (#17144)
2026-07-21 13:51:17 +08:00
get_driver.go
fix: docx/email parsing, extractor LLM driver, and chunker alignment (#17144)
2026-07-21 13:51:17 +08:00
gitee_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
gitee.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
google_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
google.go
fix(go): send Gemini system prompts via system_instruction instead of user turn (#17449)
2026-07-31 18:40:11 +08:00
gpustack_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
gpustack.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
greenpt_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
greenpt.go
feat: add GreenPT model provider (#17447)
2026-07-28 19:19:00 +08:00
groq_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
groq.go
feat(go-models): migrate nvidia/siliconflow/groq/longcat to unified h… (#17643)
2026-08-03 10:05:25 +08:00
huaweicloud_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
huaweicloud.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
huggingface.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
hunyuan_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
hunyuan.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
jiekouai_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
jiekouai.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
jina_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
jina.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
llm_test.go
fix(go-agent): message double (#17353)
2026-07-24 19:23:51 +08:00
llm.go
fix(go-agent): message double (#17353)
2026-07-24 19:23:51 +08:00
lmstudio.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
localai_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
localai.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
longcat_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
longcat.go
feat(go-models): migrate nvidia/siliconflow/groq/longcat to unified h… (#17643)
2026-08-03 10:05:25 +08:00
mineru_local.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
mineru.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
minimax_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
minimax.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
mistral_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
mistral.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
model_test_helpers_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
model_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
model.go
Fix: align go provider apis with python apis (#16867)
2026-07-14 21:06:11 +08:00
modelscope_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
modelscope.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
moonshot_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
moonshot.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
n1n_test.go
fix(go-models): harden N1N default transport handling (#15351)
2026-06-02 13:40:10 +08:00
n1n.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
non_chat_usage_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
novita_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
novita.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
nvidia_rerank_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
nvidia_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
nvidia.go
feat(go-models): migrate nvidia/siliconflow/groq/longcat to unified h… (#17643)
2026-08-03 10:05:25 +08:00
ollama_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
ollama.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
openai_api_compatible.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
openai_compatible_usage_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
openai_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
openai.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
openrouter_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
openrouter.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
orcarouter.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
paddleocr_local.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
paddleocr.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
parser_config.go
feat(go-models): unify OpenAI-compatible token usage extraction (#17634)
2026-07-31 18:32:07 +08:00
perplexity_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
perplexity.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
ppio_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
ppio.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
provider_response_usage_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
qiniu_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
qiniu.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
ragcon_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
ragcon.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
reasoning_family_provider_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
replicate_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
replicate.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
response_handler.go
feat(go-models): unify OpenAI-compatible token usage extraction (#17634)
2026-07-31 18:32:07 +08:00
siliconflow_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
siliconflow.go
feat(go-models): migrate nvidia/siliconflow/groq/longcat to unified h… (#17643)
2026-08-03 10:05:25 +08:00
sse_scanner_buffer_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
stepfun_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
stepfun.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
timeout_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
togetherai_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
togetherai.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
tokenhub_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
tokenhub.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
tokenpony_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
tokenpony.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
tool_call_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
types.go
Go: implement embed, rerank for PPIO provider (#17486)
2026-07-28 19:16:31 +08:00
upstage_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
upstage.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
usage_parser_test.go
feat(go-models): unify OpenAI-compatible token usage extraction (#17634)
2026-07-31 18:32:07 +08:00
usage_parser.go
feat(go-models): unify OpenAI-compatible token usage extraction (#17634)
2026-07-31 18:32:07 +08:00
vllm_rerank_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
vllm.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
volcengine_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
volcengine.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
voyage_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
voyage.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
xai_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
xai.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
xiaomi_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
xiaomi.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
xinference_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
xinference.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
xunfei_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
xunfei.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
zhipu_ai_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
zhipu-ai_test.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00
zhipu-ai.go
Go: fix SSRF (#17641)
2026-07-31 19:15:38 +08:00