fa71f973a6
Some checks failed
- New fusionagi/gpu/ module with TensorBackend protocol abstraction - TensorFlowBackend: GPU-accelerated ops with TensorCore mixed-precision - NumPyBackend: CPU fallback (always available, no extra deps) - Auto-selects best available backend at runtime - GPU-accelerated operations: - Cosine similarity matrix (batched, XLA-compiled) - Multi-head attention for consensus scoring - Batch hypothesis scoring on GPU - Semantic similarity search (pairwise, nearest-neighbor, deduplication) - New TensorFlowAdapter (fusionagi/adapters/): - LLMAdapter for local TF/Keras model inference - TensorCore mixed-precision support - GPU-accelerated embedding synthesis fallback - Reasoning pipeline integration: - gpu_scoring.py: drop-in GPU replacement for multi_path scoring - Super Big Brain: use_gpu config flag, GPU scoring when available - Memory integration: - gpu_search.py: GPU-accelerated semantic search for SemanticGraphMemory - Self-improvement integration: - gpu_training.py: gradient-based heuristic weight optimization - Reflective memory training loop with loss tracking - Dependencies: gpu extra (tensorflow>=2.16, numpy>=1.26) - 64 new tests (276 total), all passing - Architecture spec: docs/gpu_tensorcore_integration.md Co-Authored-By: Nakamoto, S <defi@defi-oracle.io>
65 lines
1.6 KiB
Python
65 lines
1.6 KiB
Python
"""Reasoning engine: chain-of-thought, tree-of-thought, and native symbolic reasoning."""
|
|
|
|
from fusionagi.reasoning.cot import (
|
|
build_cot_messages,
|
|
run_chain_of_thought,
|
|
)
|
|
from fusionagi.reasoning.tot import (
|
|
run_tree_of_thought,
|
|
run_tree_of_thought_detailed,
|
|
ThoughtBranch,
|
|
ThoughtNode,
|
|
ToTResult,
|
|
expand_node,
|
|
prune_subtree,
|
|
merge_subtrees,
|
|
)
|
|
from fusionagi.reasoning.native import (
|
|
NativeReasoningProvider,
|
|
analyze_prompt,
|
|
produce_head_output,
|
|
PromptAnalysis,
|
|
)
|
|
from fusionagi.reasoning.decomposition import decompose_recursive
|
|
from fusionagi.reasoning.multi_path import generate_and_score_parallel
|
|
from fusionagi.reasoning.recomposition import recompose, RecomposedResponse
|
|
from fusionagi.reasoning.meta_reasoning import (
|
|
challenge_assumptions,
|
|
detect_contradictions,
|
|
revisit_node,
|
|
)
|
|
from fusionagi.reasoning.gpu_scoring import (
|
|
generate_and_score_gpu,
|
|
score_claims_gpu,
|
|
deduplicate_claims_gpu,
|
|
)
|
|
|
|
__all__ = [
|
|
"build_cot_messages",
|
|
"run_chain_of_thought",
|
|
"run_tree_of_thought",
|
|
"run_tree_of_thought_detailed",
|
|
"ThoughtBranch",
|
|
"ThoughtNode",
|
|
"ToTResult",
|
|
"expand_node",
|
|
"prune_subtree",
|
|
"merge_subtrees",
|
|
"NativeReasoningProvider",
|
|
"analyze_prompt",
|
|
"produce_head_output",
|
|
"PromptAnalysis",
|
|
"decompose_recursive",
|
|
"load_context_for_reasoning",
|
|
"build_compact_prompt",
|
|
"generate_and_score_parallel",
|
|
"recompose",
|
|
"RecomposedResponse",
|
|
"challenge_assumptions",
|
|
"detect_contradictions",
|
|
"revisit_node",
|
|
"generate_and_score_gpu",
|
|
"score_claims_gpu",
|
|
"deduplicate_claims_gpu",
|
|
]
|