File size: 796 Bytes
8663f54 29cdc9d 8663f54 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | import numpy as np
import os
import sys
sys.path.insert(0, os.path.expanduser("~/vitalis_devcore"))
from vitalis_ide.math_core.kernel import VitalisKernel
class InferenceEngine:
def __init__(self):
self.kernel = VitalisKernel()
def reason(self, prompt: str) -> str:
tokens = prompt.strip().split()
vec = self.kernel.vectorize_tokens(tokens)
confidence = float(np.mean(np.abs(vec)))
if "scaffold" in prompt.lower():
return "scaffold"
elif "write" in prompt.lower() or "fix" in prompt.lower():
return "write"
else:
return f"[INFER] Confidence={confidence:.3f} | Input={prompt[:80]}"
def embed(self, text: str) -> np.ndarray:
return self.kernel.vectorize_tokens(text.strip().split())
|