space / src /xblago /cli.py
RomanMaloev's picture
Deploy from GitHub Actions
b539678 verified
Raw History Blame Contribute Delete
1.44 kB
from __future__ import annotations
import argparse
from pathlib import Path
from xblago.agents.personal_assistant import PersonalAssistantAgent
from xblago.core.config import load_llm_config
from xblago.core.llm import build_chat_provider
from xblago.core.memory import JsonMemoryStore, default_memory_path
from xblago.core.messages import UserMessage
def build_parser() -> argparse.ArgumentParser:
parser = argparse.ArgumentParser(description="Run the xBLAGO personal assistant.")
parser.add_argument(
"--memory",
type=Path,
default=default_memory_path(),
help="Path to the local JSON memory file.",
)
return parser
def main() -> None:
args = build_parser().parse_args()
memory = JsonMemoryStore(args.memory)
llm_config = load_llm_config()
chat_provider = build_chat_provider(llm_config)
agent = PersonalAssistantAgent(memory=memory, chat_provider=chat_provider)
print(agent.welcome_message())
while True:
try:
raw_text = input("\nYou: ").strip()
except (EOFError, KeyboardInterrupt):
print("\nxBLAGO: До встречи. Я буду рядом, когда понадоблюсь.")
return
if not raw_text:
continue
response = agent.respond(UserMessage(text=raw_text))
print(f"xBLAGO: {response.text}")
if response.should_end_session:
return