Update trace instrument command (#272)

* Fix ui bug

* Fix arms package link error

* Fix bug

* Fix exclude key

* Remove row_number

* Fix template error

* Fix dockerfile

Dockerfile

@@ -26,4 +26,4 @@ RUN apt-get update && apt-get install -y libgl1 libglib2.0-0
 WORKDIR /app
 COPY --from=builder ${VIRTUAL_ENV} ${VIRTUAL_ENV}
 COPY . .
-CMD ["pai_rag", "serve"]
+CMD ["ENABLE_FASTAPI=false", "ENABLE_REQUESTS=false", "ENABLE_AIOHTTPCLIENT=false", "aliyun-instrument", "python" , "src/pai_rag/main.py"]

Dockerfile_gpu

@@ -29,4 +29,4 @@ RUN apt-get update && apt-get install -y libgl1 libglib2.0-0
 WORKDIR /app
 COPY --from=builder ${VIRTUAL_ENV} ${VIRTUAL_ENV}
 COPY . .
-CMD ["pai_rag", "serve"]
+CMD ["ENABLE_FASTAPI=false", "ENABLE_REQUESTS=false", "ENABLE_AIOHTTPCLIENT=false", "aliyun-instrument", "python" , "src/pai_rag/main.py"]

Dockerfile_ui

@@ -11,9 +11,6 @@ WORKDIR /app
 COPY . .
 
 RUN poetry install && rm -rf $POETRY_CACHE_DIR
-ENV PYTHON_AGENT_PATH="https://python-agent.oss-rg-china-mainland.aliyuncs.com/1.1.0.rc/aliyun-python-agent.tar.gz"
-RUN poetry run aliyun-bootstrap -a install
-
 FROM python:3.11-slim AS prod
 
 RUN rm -rf /etc/localtime && ln -s /usr/share/zoneinfo/Asia/Harbin  /etc/localtime

src/pai_rag/main.py

@@ -1,9 +1,4 @@
 import os
-os.environ["ENABLE_FASTAPI"] = "false"
-os.environ["ENABLE_REQUESTS"] = "false"
-os.environ["ENABLE_AIOHTTPCLIENT"] = "false"
-from aliyun.opentelemetry.instrumentation.auto_instrumentation import sitecustomize # ByPass unused import: use for arms trace initialization
-
 import click
 import uvicorn
 from fastapi import FastAPI