test: fix

Author: vemonet

compose.text2sparql.yml

@@ -3,39 +3,34 @@ services:
     extends:
       file: compose.yml
       service: vectordb
-    # ports:
-    #   - 6334:6334
 
-  # https://docs.tentris.io/running_with_docker.html
-  tentris-dbpedia:
-    image: ghcr.io/tentris/tentris:latest
+  text2sparql-api:
+    build: .
     restart: unless-stopped
     ports:
-      - "9080:9080"
+      - 8081:8765
     volumes:
-      - /local/data/tentris/dbpedia:/data
-      - ./data/dumps/dbpedia:/dumps
-      - ./data/tentris/tentris-license.toml:/config/tentris-license.toml:ro
-    # environment:
-    #   TENTRIS_DEFAULT_GRAPH_MODE: union
-    #   # TENTRIS_RDF_FILE: /dumps/*
-
-  # tentris-corporate:
-  #   image: ghcr.io/tentris/tentris:latest
-  #   restart: unless-stopped
-  #   ports:
-  #     - "9080:9080"
-  #   volumes:
-  #     - ./data/tentris/corporate:/data
-  #     - ./data/tentris/corporate-dumps:/dumps
-  #     - ./data/tentris/tentris-license.toml:/config/tentris-license.toml:ro
+      - ./src:/app/src
+      - ./tests:/app/tests
+      - ./data:/app/data
+    env_file:
+      - .env
+    environment:
+      - BENCH_MODEL=openrouter/openai/gpt-oss-120b:free
+      - PYTHONUNBUFFERED=1
+      - VECTORDB_URL=http://vectordb:6334/
+      # - DBPEDIA_URL=http://141.57.8.18:9081/sparql
+      # - CORPORATE_URL=http://141.57.8.18:9080/sparql
+      - DBPEDIA_URL=http://virtuoso-dbpedia:8890/sparql
+      - CORPORATE_URL=http://virtuoso-corporate:8890/sparql
+    entrypoint: uv run
+    command: uvicorn tests.text2sparql.api:app --host 0.0.0.0 --port 8765
 
   virtuoso-dbpedia:
     image: openlink/virtuoso-opensource-7:latest
     restart: unless-stopped
     ports:
-      - "8890:8890"   # HTTP SPARQL endpoint
-      - "1111:1111"   # ISQL interface
+      - "8890:8890"
     environment:
       DBA_PASSWORD: dba
       VIRT_Parameters_NumberOfBuffers: 2720000
@@ -49,8 +44,7 @@ services:
   #   image: openlink/virtuoso-opensource-7:latest
   #   restart: unless-stopped
   #   # ports:
-  #   #   - "8890:8890"   # HTTP SPARQL endpoint
-  #   #   - "1111:1111"   # ISQL interface
+  #   #   - "8890:8890"
   #   environment:
   #     DBA_PASSWORD: dba
   #     VIRT_Parameters_NumberOfBuffers: 2720000
@@ -60,24 +54,23 @@ services:
   #     - ./data/dumps/corporate:/dumps
   #     - ./data/virtuoso/corporate:/database
 
-  text2sparql-api:
-    build: .
-    restart: unless-stopped
-    ports:
-      - 8081:8765
-    volumes:
-      - ./src:/app/src
-      - ./tests:/app/tests
-      - ./data:/app/data
-    env_file:
-      - .env
-    environment:
-      # - DBPEDIA_URL=https://text2sparql.aksw.org/2026/dbpedia/sparql
-      # - CORPORATE_URL=https://text2sparql.aksw.org/2026/corporate/sparql
-      - DBPEDIA_URL=http://virtuoso-dbpedia:8890/sparql
-      - CORPORATE_URL=http://virtuoso-corporate:8890/sparql
-      - VECTORDB_URL=http://vectordb:6334/
-      - BENCH_MODEL=openrouter/openai/gpt-oss-120b
-      - PYTHONUNBUFFERED=1
-    entrypoint: uv run
-    command: uvicorn tests.text2sparql.api:app --host 0.0.0.0 --port 8765
+  # # https://docs.tentris.io/running_with_docker.html
+  # tentris-dbpedia:
+  #   image: ghcr.io/tentris/tentris:latest
+  #   restart: unless-stopped
+  #   ports:
+  #     - "9080:9080"
+  #   volumes:
+  #     - /local/data/tentris/dbpedia:/data
+  #     - ./data/dumps/dbpedia:/dumps
+  #     - ./data/tentris/tentris-license.toml:/config/tentris-license.toml:ro
+
+  # tentris-corporate:
+  #   image: ghcr.io/tentris/tentris:latest
+  #   restart: unless-stopped
+  #   ports:
+  #     - "9080:9080"
+  #   volumes:
+  #     - ./data/tentris/corporate:/data
+  #     - ./data/tentris/corporate-dumps:/dumps
+  #     - ./data/tentris/tentris-license.toml:/config/tentris-license.toml:ro

tests/text2sparql/index.py

@@ -22,6 +22,7 @@ def init_vectordb(
     docs: list[Document] = []
 
     # Index example queries
+    # examples = ["Generated-CK", "Text2SPARQL-ck"] if "corporate" in dataset_iri else ["QALD-9+", "LC-QuAD", "Text2SPARQL-db"]
     examples = ["Generated-CK"] if "corporate" in dataset_iri else ["QALD-9+", "LC-QuAD"]
 
     queries = pd.read_csv(QUERIES_FILE)
@@ -110,12 +111,10 @@ def init_vectordb(
     )
     args = parser.parse_args()
 
-    dataset_iri = args.dataset_iri
-
     # Init vectordb for the specified dataset
     init_vectordb(
-        endpoint_url=DATASETS_ENDPOINTS[dataset_iri],
-        dataset_iri=dataset_iri,
+        endpoint_url=DATASETS_ENDPOINTS[args.dataset_iri],
+        dataset_iri=args.dataset_iri,
         limit_schema={
             "top_classes_percentile": 0,
             "top_n_predicates": 20,