feat: add conversational RollCalc assistant
This commit is contained in:
@@ -0,0 +1,75 @@
|
||||
"""Local terminal demo for the constrained Ollama RollCalc adapter."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import argparse
|
||||
from dataclasses import replace
|
||||
import json
|
||||
|
||||
from conversation_service import ConversationService
|
||||
from ollama_nlu import OllamaConfig, OllamaNLUClient
|
||||
|
||||
|
||||
def _arguments() -> argparse.Namespace:
|
||||
parser = argparse.ArgumentParser(
|
||||
description="Constrained local natural-language demo for RollCalc",
|
||||
)
|
||||
parser.add_argument("--model", help="Override ROLLCALC_OLLAMA_MODEL")
|
||||
parser.add_argument("--url", help="Override ROLLCALC_OLLAMA_URL")
|
||||
parser.add_argument("--timeout", type=float, help="Ollama timeout in seconds")
|
||||
parser.add_argument(
|
||||
"--debug",
|
||||
action="store_true",
|
||||
help="Show raw model JSON, validated interpretation, outcome, and latency",
|
||||
)
|
||||
return parser.parse_args()
|
||||
|
||||
|
||||
def main() -> int:
|
||||
args = _arguments()
|
||||
config = OllamaConfig.from_env()
|
||||
overrides = {}
|
||||
if args.model:
|
||||
overrides["model"] = args.model
|
||||
if args.url:
|
||||
overrides["base_url"] = args.url
|
||||
if args.timeout is not None:
|
||||
overrides["timeout_seconds"] = args.timeout
|
||||
if overrides:
|
||||
config = replace(config, **overrides)
|
||||
|
||||
service = ConversationService(OllamaNLUClient(config))
|
||||
conversation_id = service.create_conversation()
|
||||
print(
|
||||
f"RollCalc NLU demo ({config.model}). "
|
||||
"Enter 'exit' or 'quit' to stop."
|
||||
)
|
||||
while True:
|
||||
try:
|
||||
message = input("> ").strip()
|
||||
except (EOFError, KeyboardInterrupt):
|
||||
print()
|
||||
return 0
|
||||
if message.casefold() in {"exit", "quit"}:
|
||||
return 0
|
||||
if not message:
|
||||
continue
|
||||
|
||||
response = service.handle_message(
|
||||
conversation_id,
|
||||
message,
|
||||
include_diagnostics=args.debug,
|
||||
)
|
||||
print(response["message"])
|
||||
if args.debug:
|
||||
debug = {
|
||||
"raw_model_json": response.get("nlu", {}).get("raw_model_json"),
|
||||
"validated_interpretation": response.get("interpretation"),
|
||||
"deterministic_outcome": response.get("result", {}).get("status"),
|
||||
"latency_ms": response.get("nlu", {}).get("latency_ms"),
|
||||
}
|
||||
print(json.dumps(debug, ensure_ascii=False, indent=2))
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
raise SystemExit(main())
|
||||
Reference in New Issue
Block a user