[feature][ai] Add basic func def

Author: chengebj

dingo_command/api/aiinstane.py

@@ -0,0 +1,35 @@
+# ai相关的容器实例的创建接口
+from fastapi import APIRouter, HTTPException
+
+from dingo_command.api.model.aiinstance import AiInstanceApiModel
+from dingo_command.services.ai_instance import AiInstanceService
+from dingo_command.services.custom_exception import Fail
+
+router = APIRouter()
+ai_instance_service = AiInstanceService()
+
+@router.post("/ai-instance/create", summary="创建容器实例", description="创建容器实例")
+async def create_ai_instance(ai_instance:AiInstanceApiModel):
+    # 创建容器实例
+    try:
+        # 创建成功
+        return ai_instance_service.create_ai_instance(ai_instance)
+    except Fail as e:
+        raise HTTPException(status_code=400, detail=e.error_message)
+    except Exception as e:
+        import traceback
+        traceback.print_exc()
+        raise HTTPException(status_code=400, detail=f"创建容器实例失败:{ai_instance.ai_instance_name}")
+
+@router.delete("/ai-instance/{id}", summary="删除容器实例", description="根据实例id删除容器实例数据")
+async def delete_instance_by_id(id:str):
+    # 删除容器实例
+    try:
+        # 删除成功
+        return ai_instance_service.delete_ai_pod_by_id(id)
+    except Fail as e:
+        raise HTTPException(status_code=400, detail=e.error_message)
+    except Exception as e:
+        import traceback
+        traceback.print_exc()
+        raise HTTPException(status_code=400, detail=f"删除容器实例失败:{id}")

dingo_command/api/model/aiinstance.py

@@ -0,0 +1,11 @@
+# 定义ai容器相关的model对象
+from typing import Optional, List, Any, Dict
+
+from pydantic import BaseModel, Field
+
+
+# 容器实例信息
+class AiInstanceApiModel(BaseModel):
+    ai_instance_id: Optional[str] = Field(None, description="容器实例的id")
+    ai_instance_name: Optional[str] = Field(None, description="容器实例的名称")
+    description: Optional[str] = Field(None, description="描述信息")

dingo_command/services/ai_instance.py

@@ -0,0 +1,12 @@
+# ai实例的service层
+from oslo_log import log
+
+LOG = log.getLogger(__name__)
+
+class AiInstanceService:
+
+    def delete_ai_pod_by_id(self, id):
+        return None
+
+    def create_ai_instance(self, ai_instance):
+        pass

dingo_command/utils/k8s_client.py

@@ -0,0 +1,16 @@
+# k8s的client
+
+from kubernetes import client, config
+
+
+def get_k8s_client(config_file: str, context: str):
+    """检测"""
+    if not config_file or not context:
+        raise ValueError(f"Invalid param config_file: {config_file}, context: {context}")
+    """加载配置"""
+    config.load_kube_config(
+        config_file=config_file,
+        context=context,
+    )
+    """返回api"""
+    return client.CoreV1Api()

requirements.txt

@@ -86,6 +86,7 @@ jupyter_core==5.7.2
 jupyterlab_pygments==0.3.0
 keystoneauth1==5.8.0
 kombu==5.4.2
+kubernetes==33.1.0
 libcomps==0.1.20.post1
 lockfile==0.12.2
 loguru==0.5.3