From bf1605f89d64bb6ff8777bec12ee1ea500204a86 Mon Sep 17 00:00:00 2001 From: prodesk98 Date: Fri, 18 Jul 2025 15:07:58 -0300 Subject: [PATCH 1/2] Update docker-compose configuration: add dependencies for mongo, neo4j, and qdrant services --- docker-compose.yaml | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/docker-compose.yaml b/docker-compose.yaml index 6643e20..fcf3853 100644 --- a/docker-compose.yaml +++ b/docker-compose.yaml @@ -41,6 +41,10 @@ services: command: celery -A workers.tasks worker --loglevel=INFO volumes: - ./:/app + depends_on: + - mongo + - neo4j + - qdrant app: build: @@ -57,6 +61,8 @@ services: command: fastapi dev main.py --host 0.0.0.0 --port 8000 volumes: - ./:/app + depends_on: + - worker volumes: neo4j-data: From 1aa701762ae60f5f78fd2e7f7ecc4274e7c45289 Mon Sep 17 00:00:00 2001 From: prodesk98 Date: Sun, 20 Jul 2025 15:12:31 -0300 Subject: [PATCH 2/2] Improve code readability: adjust formatting in connection.py and add comment in knowledge.py --- workers/connection.py | 4 ++-- workers/knowledge.py | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/workers/connection.py b/workers/connection.py index ef6b3ff..fd1e699 100644 --- a/workers/connection.py +++ b/workers/connection.py @@ -3,8 +3,8 @@ broker_transport_options = { 'region': env.AWS_REGION, - 'visibility_timeout': 3600, # 1 hour - 'polling_interval': 5, # seconds + 'visibility_timeout': 3600, # 1 hour + 'polling_interval': 5, # seconds 'predefined_queues': { 'default': { 'url': env.SQS_DEFAULT_QUEUE_URL, diff --git a/workers/knowledge.py b/workers/knowledge.py index 5b55b64..f051eb3 100644 --- a/workers/knowledge.py +++ b/workers/knowledge.py @@ -338,6 +338,7 @@ def process(self, key: str): aws_access_key_id=env.AWS_ACCESS_KEY_ID, aws_secret_access_key=env.AWS_SECRET_ACCESS_KEY, ) + # Create the LLMGraphTransformer with the allowed nodes and relationships llm_graph = LLMGraphTransformer( llm, allowed_nodes=nodes,