61 lines
1.4 KiB
Python
61 lines
1.4 KiB
Python
import asyncio
|
|
import os
|
|
from dotenv import load_dotenv
|
|
|
|
from agent import Agent
|
|
from util.logger import get_logger
|
|
|
|
load_dotenv()
|
|
|
|
# Konfiguration
|
|
AWS_REGION = os.getenv("AWS_REGION")
|
|
AWS_ACCESS_KEY_ID = os.getenv("AWS_ACCESS_KEY_ID")
|
|
AWS_SECRET_ACCESS_KEY = os.getenv("AWS_SECRET_ACCESS_KEY")
|
|
BEDROCK_MODEL = os.getenv("BEDROCK_MODEL")
|
|
MCP_URL = os.getenv("MCP_URL")
|
|
|
|
|
|
async def main():
|
|
log = get_logger("NORA")
|
|
|
|
print("\n" + "="*40)
|
|
print(" N.O.R.A START")
|
|
print("="*40 + "\n")
|
|
|
|
agent = Agent()
|
|
|
|
try:
|
|
log.info("Connecting to MCP...")
|
|
await agent.mcp.connect()
|
|
|
|
log.info("Loading tools...")
|
|
await agent.load_tools()
|
|
|
|
print("N.O.R.A ist bereit. (Schreibe 'exit' oder 'quit' zum Beenden)\n")
|
|
|
|
while True:
|
|
try:
|
|
user = input("Du: ").strip()
|
|
|
|
if user.lower() in ["exit", "quit", "bye"]:
|
|
break
|
|
|
|
if not user:
|
|
continue
|
|
|
|
response = await agent.run(user)
|
|
print(f"\nN.O.R.A: {response}\n")
|
|
|
|
except KeyboardInterrupt:
|
|
break
|
|
except Exception as e:
|
|
log.error(f"Fehler in Main-Loop: {e}")
|
|
print("Ein Fehler ist aufgetreten. Bitte versuche es erneut.")
|
|
|
|
finally:
|
|
await agent.mcp.close()
|
|
log.info("Shutdown complete")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main()) |