ZuperAI-14B / toolkit /chat.py
Tyesee's picture
toolkit: chat.py
00fc882 verified
Raw History Blame Contribute Delete
1.33 kB
# -*- coding: utf-8 -*-
"""CLI-чат с ZuperAI (№6). Терминальный диалог, как `ollama run`, но с нашим стеком.
Запуск:
venv/Scripts/python.exe chat.py [deepseek|lmstudio|local]
Выход: 'exit' / 'quit' / Ctrl+C.
"""
import sys
from brain import pick, generate
SYSTEM = ("Ты — ZuperAI, опытный программист. Отвечай по делу, на русском. "
"Пиши рабочий код, когда просят. Комментарии и докстринги — на русском.")
def main():
backend = pick(sys.argv[1] if len(sys.argv) > 1 else None)
print(f"[*] мозг: {backend}. Пиши вопрос (или exit/выход).")
history = [{"role": "system", "content": SYSTEM}]
while True:
try:
line = input(">>> ")
except (EOFError, KeyboardInterrupt):
print()
break
line = line.strip()
if not line:
continue
if line.lower() in ("exit", "quit", "выход"):
break
history.append({"role": "user", "content": line})
ans = generate(backend, history, 1024)
history.append({"role": "assistant", "content": ans})
print(ans)
print()
if __name__ == "__main__":
main()