Add support for docker deploy

.dockerignore

@@ -0,0 +1,4 @@
+__pycache__/
+*.pyc
+*.pyo
+.git/

.env-example

@@ -24,19 +24,3 @@ SPECBOT_AI_MODEL=gpt-4-0613
 REPAIR_PRO_AI_MODEL=claude-3-5-sonnet-20240620
 OPENAI_API_KEY=*** Provide your API Key ***
 OPENAI_BASE_URL=*** Provide your Base url ***
-
-# 数据库配置
-DB_NAME=
-DB_USER=
-DB_PASSWORD=
-DB_HOST=
-DB_PORT=
-
-# 模型名称配置
-MODEL_NAME=
-
-# 配置项
-VECTOR_EXTENSION=
-TABLE_NAME=
-VECTOR_DIMENSION=
-LANGUAGE=

Dockerfile

@@ -0,0 +1,36 @@
+# 使用 openEuler 22.03 基础镜像
+FROM openeuler/openeuler:22.03
+
+# 设置工作目录为 /app
+WORKDIR /app
+
+# 更新系统并安装必要的软件包
+RUN yum update -y && \
+    yum install -y \
+    python3 \
+    python3-pip \
+    python3-devel \
+    gcc \
+    make \
+    && yum clean all
+
+# 创建符号链接，仅为 python 创建
+RUN ln -s /usr/bin/python3 /usr/bin/python
+
+# 复制 requirements.txt 到容器中
+COPY requirements.txt .
+
+# 安装Python依赖
+RUN pip install --no-cache-dir -r requirements.txt
+
+# 复制当前目录下的所有文件到容器中
+COPY . .
+
+# 按照模块安装
+RUN pip install .
+
+# 暴露容器的8000端口
+EXPOSE 8000
+
+# 启动FastAPI应用
+CMD ["python", "infra_ai_service/server.py"]

README.CN.md

@@ -1,18 +1,8 @@
 # 注意事项
 
 ## 前置条件
-- 需要安装qdrant本地服务
-```shell
-# docker环境准备
-sudo apt update
-sudo apt install docker.io
-sudo systemctl start docker
-sudo systemctl enable docker
-# 安装
-sudo docker run -p 6333:6333 -d --name qdrant qdrant/qdrant
-# 测试
-curl http://localhost:6333
-```
+- 需要安装数据库postgresql及其pgvector插件
+- 需要新增.env文件，将数据库配置，外部服务配置等添加，并将MODEL_NAME设置为multi-qa-MiniLM-L6-cos-v1
 
 ## 本地启动
 ```shell
@@ -25,6 +15,14 @@ pytest .
 python infra_ai_service/server.py
 ```
 
+# 容器化部署
+```shell
+# 构建容器
+docker build -t infra_ai_service .
+# 运行ai_service服务
+docker run -p 8001:8000 -d --name infra_ai_service_imp ai_service
+
+```
 
 ## 访问spec-repair API
 

infra_ai_service/api/ai_enhance/embedding.py

@@ -1,7 +1,6 @@
 from fastapi import APIRouter
 
-from infra_ai_service.api.ai_enhance.text_process import TextInput
-from infra_ai_service.model.model import EmbeddingOutput
+from infra_ai_service.model.model import EmbeddingOutput, TextInput
 from infra_ai_service.service.embedding_service import create_embedding
 
 router = APIRouter()

infra_ai_service/api/router.py

@@ -6,21 +6,19 @@
 from infra_ai_service.api.ai_enhance.spec_repair_process import (
     router as spec_repair_process,
 )
-from infra_ai_service.api.ai_enhance.text_process import (
-    router as text_process_router,
-)
 from infra_ai_service.api.ai_enhance.vector_search import (
     router as vector_search_router,
 )
+from infra_ai_service.api.system.status import router as status_router
 
 api_router = APIRouter()
 api_router.include_router(
     spec_repair_process, prefix="/spec-repair", tags=["repair"]
 )
-api_router.include_router(text_process_router, prefix="/text", tags=["text"])
 api_router.include_router(
     embedding_router, prefix="/embedding", tags=["embedding"]
 )
 api_router.include_router(
     vector_search_router, prefix="/search", tags=["search"]
 )
+api_router.include_router(status_router, prefix="/status", tags=["status"])

infra_ai_service/api/system/status.py

@@ -0,0 +1,8 @@
+from fastapi import APIRouter, Response, status
+
+router = APIRouter()
+
+
+@router.get("/status/", status_code=status.HTTP_200_OK)
+async def status():
+    return Response(content="", media_type="application/json")

tox.ini

@@ -39,7 +39,7 @@ deps =
     coverage
     {[testenv]deps}
 commands =
-    pytest tests/ --cov=infra_ai_service --cov-report=term-missing
+    pytest tests/ --cov=infra_ai_service --cov-report=term-missing --disable-warnings
     coverage report --fail-under=50
     coverage html