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):