initial attempt at memories

Makefile

@@ -6,10 +6,11 @@ clean:
 	docker compose rm -f
 	docker volume prune --all -f
 # make dev
-dev: dev-stop docker-build
+dev: docker-build
 	cp .env-dev .env
 	docker compose -f docker-compose.yaml -f docker-compose.dev.yaml up -d --build --remove-orphans
 	make logs
+dev-restart: dev-stop dev
 dev-stop:
 	docker compose -f docker-compose.yaml -f docker-compose.dev.yaml down
 	docker compose rm -f 

Pipfile

@@ -35,6 +35,9 @@ brotli = "*"
 python-magic = "*"
 pillow = "*"
 valkey = {extras = ["libvalkey"], version = "*"}
+numpy = "*"
+pgvector = "*"
+psycopg = {extras = ["pool", "binary"], version = "*"}
 
 [dev-packages]
 autopep8 = "*"

bot.py

@@ -21,6 +21,7 @@
 from plugins.tts import TTS
 from plugins.users import Users
 from plugins.version import Version
+from plugins.vectordb import VectorDb
 
 env = Env()
 log_channel = env.str("MM_BOT_LOG_CHANNEL")
@@ -60,6 +61,7 @@
         Giphy(),
         Ntp(),
         Jira(),
+        VectorDb(),
         Version(),
     ],
     enable_logging=True,

docker-compose.dev.yaml

@@ -4,4 +4,20 @@ services:
     entrypoint: /bin/true
     volumes: []
     restart: "no"
-    command: []
+    command: []
+  pgadmin:
+    image: dpage/pgadmin4
+    restart: unless-stopped
+    environment:
+      PGADMIN_DEFAULT_EMAIL: [email protected]
+      PGADMIN_DEFAULT_PASSWORD: admin
+    ports:
+      - "5050:80"
+    volumes:
+      - pgadmin-data:/var/lib/pgadmin
+    links:
+      - pg
+    depends_on:
+      - pg
+volumes:
+  pgadmin-data:

docker-compose.yaml

@@ -8,6 +8,7 @@ services:
       VALKEY_DB: 0
     links:
       - valkey
+      - pg
     labels:
       com.centurylinklabs.watchtower.scope: mmchatgpt
     command: [ 'python', 'bot.py' ]
@@ -23,6 +24,14 @@ services:
       - redis-data:/data
     expose:
       - "6379"
+  pg:
+    image: pgvector/pgvector:pg17
+    env_file:
+      - .env
+    volumes:
+      - pg-data:/var/lib/postgresql/data
+    expose:
+      - "5432"
   #watchtower:
   #  image: containrrr/watchtower:1.7.1
   #  volumes:
@@ -32,3 +41,4 @@ services:
 volumes:
   redis-data:
   app-data:
+  pg-data: