Transforms the Matrix and OpenAI interactor into indepandent services, fikes #2 #3
12
.env.example
12
.env.example
@@ -2,12 +2,12 @@
|
||||
LOG_LEVEL=INFO
|
||||
|
||||
# Matrix Configuration
|
||||
MATRIX_HOMESERVER_URL = "https://matrix.org"
|
||||
MATRIX_USER_ID = "@botbot_user:matrix.org"
|
||||
MATRIX_PASSWORD = "botbot_password"
|
||||
MATRIX_LOGIN_TRIES = 5 # Number of login attempts before giving up, default is 5
|
||||
MATRIX_LOGIN_DELAY_INCREMENT = 5 # Delay increment,in seconds, between login attempts, default is 5
|
||||
AI_HANDLER_URL = "http://ai_service:8000"
|
||||
MATRIX_HOMESERVER_URL="https://matrix.org"
|
||||
MATRIX_USER_ID="@botbot_user:matrix.org"
|
||||
MATRIX_PASSWORD="botbot_password"
|
||||
MATRIX_LOGIN_TRIES=5 # Number of login attempts before giving up, default is 5
|
||||
MATRIX_LOGIN_DELAY_INCREMENT=5 # Delay increment,in seconds, between login attempts, default is 5
|
||||
AI_HANDLER_URL="http://ai_service:8000"
|
||||
|
||||
# OpenAI API Key
|
||||
OPENAI_API_KEY=your_openai_api_key_here
|
||||
|
||||
@@ -10,11 +10,11 @@ openai.api_key = os.environ["OPENAI_API_KEY"]
|
||||
r = redis.Redis.from_url(os.environ.get("REDIS_URL", "redis://redis:6379"))
|
||||
|
||||
class MessagePayload(BaseModel):
|
||||
roomId: str
|
||||
userId: str
|
||||
content: str
|
||||
eventId: str
|
||||
timestamp: int
|
||||
roomId: str
|
||||
userId: str
|
||||
eventId: str
|
||||
serverTimestamp: int
|
||||
content: str
|
||||
|
||||
app = FastAPI()
|
||||
|
||||
|
||||
@@ -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"):
|
||||
|
||||
@@ -1,3 +1,4 @@
|
||||
matrix-nio[e2e]>=0.25.0
|
||||
python-dotenv>=1.0.0
|
||||
openai
|
||||
httpx>=0.23.0
|
||||
pydantic>=1.10
|
||||
Reference in New Issue
Block a user