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: 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,