🔥(llm) remove mlflow intrumentation & deps
We don't use this instrumentation for now and plan to use something else.
This commit is contained in:
-11
@@ -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
|
||||
|
||||
@@ -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", [])
|
||||
|
||||
@@ -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):
|
||||
|
||||
@@ -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",
|
||||
|
||||
Reference in New Issue
Block a user