🔥(llm) remove mlflow intrumentation & deps

We don't use this instrumentation for now and plan to use something
else.
This commit is contained in:
Quentin BEY
2025-09-05 12:06:43 +02:00
parent 90d40eba6a
commit 9bbc3f160f
4 changed files with 0 additions and 35 deletions
-11
View File
@@ -143,14 +143,3 @@ services:
kc_postgresql:
condition: service_healthy
restart: true
ml-flow:
image: ghcr.io/mlflow/mlflow:v3.0.0
environment:
MLFLOW_TRACKING_URI: http://localhost:5050
MLFLOW_ARTIFACT_ROOT: /mlflow/artifacts
ports:
- "5050:5050"
volumes:
- ./data/mlflow:/mlflow/artifacts
command: mlflow server --host 0.0.0.0:5050
-11
View File
@@ -88,17 +88,6 @@ class ChatViewSet( # pylint: disable=too-many-ancestors
logger.info("Received messages: %s", request.data.get("messages", []))
if settings.ML_FLOW_TRACKING_URI:
# Set up MLflow experiment, don't import it globally to avoid issues
# when running management commands when MLflow is not started
import mlflow # pylint: disable=import-outside-toplevel
mlflow.set_tracking_uri(settings.ML_FLOW_TRACKING_URI)
mlflow.set_experiment(settings.ML_FLOW_EXPERIMENT_NAME)
# Enable automatic tracing for all OpenAI API calls
mlflow.openai.autolog()
# Warning: the messages should be stored more securely in production
conversation = self.get_object()
conversation.ui_messages = request.data.get("messages", [])
-12
View File
@@ -722,18 +722,6 @@ USER QUESTION:
environ_prefix=None,
)
# ML Flow
ML_FLOW_TRACKING_URI = values.Value(
None,
environ_name="ML_FLOW_TRACKING_URI",
environ_prefix=None,
)
ML_FLOW_EXPERIMENT_NAME = values.Value(
"conversations-openai-tracing",
environ_name="ML_FLOW_EXPERIMENT_NAME",
environ_prefix=None,
)
# pylint: disable=invalid-name
@property
def ENVIRONMENT(self):
-1
View File
@@ -25,7 +25,6 @@ license = { file = "LICENSE" }
readme = "README.md"
requires-python = ">=3.12"
dependencies = [
"mlflow",
"deprecated",
"beautifulsoup4==4.13.4",
"boto3==1.39.0",