updates.

2026-02-02 07:12:05 -05:00
parent c26e4e5ef0
commit bd8c8616d0
3 changed files with 8 additions and 8 deletions
--- a/document_ingestion_mlflow_pipeline.py
+++ b/document_ingestion_mlflow_pipeline.py
@@ -216,7 +216,7 @@ def chunk_text_with_tracking(
 def generate_embeddings_with_tracking(
    chunks: list,
    run_id: str,
-    embeddings_url: str = "http://embeddings-predictor.ai-ml.svc.cluster.local",
+    embeddings_url: str = "http://ai-inference-serve-svc.ai-ml.svc.cluster.local:8000/embeddings",
    embeddings_model: str = "bge-small-en-v1.5",
    batch_size: int = 32,
    mlflow_tracking_uri: str = "http://mlflow.mlflow.svc.cluster.local:80",