nodes/HuangYuChuh-ComfyUI-LLMs-Toolkit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat(nodes): add dedicated reasoning output pin for deepseek models

Browse Source 
Motivation: Expose the parsed reasoning_content to downstream ComfyUI nodes so users can route or display the internal thoughts of models like DeepSeek.

Implementation: Appended 'reasoning' to RETURN_TYPES and RETURN_NAMES in openai_compatible.py. Updated _success and _error_result standard return tuple length from 3 to 4.
This commit is contained in:
科林_Cyril Pilgrim
2026-03-01 23:18:39 +08:00
parent 8d66620e59
commit 8eea73f574
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes/openai_compatible.py
+6 -6
View File
@@ -121,8 +121,8 @@ class OpenAICompatibleLoader:
}
}
RETURN_TYPES = ("STRING", "INT", "INT")
RETURN_NAMES = ("text", "input_tokens", "output_tokens")
RETURN_TYPES = ("STRING", "STRING", "INT", "INT")
RETURN_NAMES = ("text", "reasoning", "input_tokens", "output_tokens")
FUNCTION = "generate"
CATEGORY = "🚦ComfyUI_LLMs_Toolkit/Generate"
OUTPUT_NODE = True
@@ -185,12 +185,12 @@ class OpenAICompatibleLoader:
# ── Result helpers ───────────────────────────────────────────────────
@staticmethod
def _success(text: str, in_tok: int, out_tok: int) -> dict:
def _success(text: str, reasoning: str, in_tok: int, out_tok: int) -> dict:
return {
"ui": {
"text": [f"Token Usage:\nInput: {in_tok}\nOutput: {out_tok}"]
},
"result": (text, in_tok, out_tok)
"result": (text, reasoning, in_tok, out_tok)
}
@staticmethod
@@ -200,7 +200,7 @@ class OpenAICompatibleLoader:
"ui": {
"text": [f"⚠ Error:\n{error_msg}"]
},
"result": (f"[Error] {error_msg}", 0, 0)
"result": (f"[Error] {error_msg}", "", 0, 0)
}
# ── Main entry ───────────────────────────────────────────────────────
@@ -315,7 +315,7 @@ class OpenAICompatibleLoader:
if enable_memory:
self._conversation_history.append({"role": "assistant", "content": response_content})
return self._success(response_content, input_tokens, output_tokens)
return self._success(response_content, reasoning_content, input_tokens, output_tokens)
except Exception as e:
elapsed_ms = int((time.time() - start) * 1000)