youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-23 20:36:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add LLM quota deduction functionality and enhance model configuration handling in llm_utils.py; update test cases for LLM node context handling

Browse Source
This commit is contained in:
Novice
2026-03-24 08:45:11 +08:00
parent ed1bd338f1
commit dcd614ca77
4 changed files with 57 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
api/tests/unit_tests/core/workflow/nodes/test_llm_node_streaming.py
View File

@ -35,7 +35,14 @@ def patch_deduct_llm_quota(monkeypatch):
def _make_llm_node(reasoning_format: str) -> LLMNode:
node = LLMNode.__new__(LLMNode)
object.__setattr__(node, "_node_data", types.SimpleNamespace(reasoning_format=reasoning_format, tools=[]))
object.__setattr__(node, "tenant_id", "tenant")
object.__setattr__(
node,
"_run_context",
{"_dify": types.SimpleNamespace(
tenant_id="tenant", app_id="app", user_id="user",
user_from="account", invoke_from="debugger",
)},
)
return node