31 lines
762 B
Python
31 lines
762 B
Python
import resource
|
|
import subprocess
|
|
import time
|
|
import llm
|
|
|
|
TRICU_PATH = "/usr/local/bin/tricu"
|
|
|
|
def tricubridge(input: str) -> str:
|
|
try:
|
|
result = subprocess.run(
|
|
[TRICU_PATH, "eval", "+RTS", "-M250M", "-RTS"],
|
|
input=input,
|
|
capture_output=True,
|
|
text=True,
|
|
check=True,
|
|
)
|
|
|
|
if len(result.stdout) > 2000:
|
|
return "Output too large; limited to 2000 chars."
|
|
else:
|
|
return result.stdout.strip()
|
|
|
|
except subprocess.CalledProcessError as e:
|
|
return f"Error executing Tricu code: {e.stderr or str(e)}"
|
|
except Exception as e:
|
|
return f"Unexpected error: {str(e)}"
|
|
|
|
@llm.hookimpl
|
|
def register_tools(register):
|
|
register(tricubridge)
|