jfig/botbot
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects 1 Releases Wiki Activity

WIP: tunning

Browse Source
This commit is contained in:
Joao Figueiredo
2025-05-03 18:52:37 +01:00
parent fb578fbf40
commit 302dd7e965
4 changed files with 28 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
matrix_service/main.py
View File

@@ -11,14 +11,16 @@ from nio.responses import LoginResponse
# --- Load environment variables ---
load_dotenv()
MATRIX_HOMESERVER_URL = os.getenv("MATRIX_HOMESERVER_URL")
MATRIX_USER_ID = os.getenv("MATRIX_USER_ID")
MATRIX_PASSWORD = os.getenv("MATRIX_PASSWORD")
MATRIX_LOGIN_TRIES = int(os.getenv("MATRIX_LOGIN_TRIES", 5))
MATRIX_HOMESERVER_URL = os.getenv("MATRIX_HOMESERVER_URL")
MATRIX_USER_ID = os.getenv("MATRIX_USER_ID")
MATRIX_PASSWORD = os.getenv("MATRIX_PASSWORD")
MATRIX_LOGIN_TRIES = int(os.getenv("MATRIX_LOGIN_TRIES", 5))
MATRIX_LOGIN_DELAY_INCREMENT = int(os.getenv("MATRIX_LOGIN_DELAY_INCREMENT", 5))
LOG_LEVEL = os.getenv("LOG_LEVEL", "INFO").upper()
AI_URL = os.getenv("AI_URL")
AI_TOKEN = os.getenv("AI_TOKEN")
AI_HANDLER_URL = os.getenv("AI_HANDLER_URL")
AI_HANDLER_TOKEN = os.getenv("AI_HANDLER_TOKEN")
@@ -76,15 +78,15 @@ async def main() -> None:
if isinstance(event, RoomMessageText):
logger.info(f"Received message in {room.room_id}: {event.body}")
payload = {
"roomId": event["room_id"],
"userId": event["sender"],
"content": event["content"]["body"],
"eventId": event["event_id"],
"timestamp": event["origin_server_ts"]
"roomId": event.room_id,
"userId": event.sender,
"eventId": event.event_id,
"serverTimestamp": event.server_timestamp,
"content": event.body
}
headers = {"Authorization": f"Bearer {AI_TOKEN}"}
headers = {"Authorization": f"Bearer {AI_HANDLER_TOKEN}"}
async with httpx.AsyncClient() as http:
resp = await http.post(f"{AI_URL}/api/v1/message", json=payload, headers=headers)
resp = await http.post(f"{AI_HANDLER_URL}/api/v1/message", json=payload, headers=headers)
resp.raise_for_status()
data = resp.json()
if data.get("reply"):