groupuser/rag-chain-agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update server.py

Browse Source
This commit is contained in:
groupuser 2025-04-23 02:14:43 +00:00
parent cbb73ba3bd
commit 111dc0a640
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
app/server.py
View File

@ -3,6 +3,11 @@ from fastapi import FastAPI
from fastapi.responses import RedirectResponse from fastapi.responses import RedirectResponse
from langserve import add_routes from langserve import add_routes
from app.chain import build_chain from app.chain import build_chain
import mlflow
mlflow.set_tracking_uri("http://localhost:5000")
mlflow.langchain.autolog()
# Input API # Input API
class QueryInput(BaseModel): class QueryInput(BaseModel):