main.py

import time
from langchain_openai import ChatOpenAI
from src.agents.manager_agent import ManagerAgent
from src.utils import send_telegram_message, receive_telegram_message
from dotenv import load_dotenv

# Load .env variables
load_dotenv()

# Use GPT-4o for telegram manager agent
gpt4 = ChatOpenAI(model_name="openai/gpt-4o-2024-08-06", temperature=0.1)

telegram_assistant = ManagerAgent(gpt4)

def monitor_bot(after_timestamp, config):
    while True:
        new_messages = receive_telegram_message(after_timestamp)
        if new_messages:
            for message in new_messages:
                sent_message = (f"Message: {message['text']}\n"
                                f"Current Date/time: {message['date']}")
                answer = telegram_assistant.invoke(sent_message, config)
                send_telegram_message(answer)
        after_timestamp = int(time.time())
        time.sleep(5)  # Sleep for 5 seconds before checking again
        

if __name__ == "__main__":
    print("Telegram Assistant Manager is running")
    config = {"configurable": {"thread_id": 42}}
    initial_timestamp = int(time.time())
    monitor_bot(initial_timestamp, config)