From 111dc0a6400b54bdfd89eb847cf076ada98f55bf Mon Sep 17 00:00:00 2001 From: groupuser Date: Wed, 23 Apr 2025 02:14:43 +0000 Subject: [PATCH] Update server.py --- app/server.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/app/server.py b/app/server.py index 50646f8..ed6752d 100644 --- a/app/server.py +++ b/app/server.py @@ -3,6 +3,11 @@ from fastapi import FastAPI from fastapi.responses import RedirectResponse from langserve import add_routes from app.chain import build_chain +import mlflow + + +mlflow.set_tracking_uri("http://localhost:5000") +mlflow.langchain.autolog() # Input API class QueryInput(BaseModel):