good version for 算法注册

services/openai-proxy/main.py

@@ -0,0 +1,109 @@
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
+import uvicorn
+import json
+import logging
+from .ai_algorithm import OpenAIProxy
+from .config import settings
+
+# 配置日志
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+)
+logger = logging.getLogger(__name__)
+
+# 初始化FastAPI应用
+app = FastAPI(
+    title="OpenAI代理服务",
+    description="提供OpenAI API代理功能的服务",
+    version="1.0.0"
+)
+
+# 初始化代理
+openai_proxy = OpenAIProxy()
+
+# 定义请求模型
+class CompletionRequest(BaseModel):
+    model: str = "gpt-3.5-turbo"
+    messages: list
+    temperature: float = 0.7
+    max_tokens: int = 1000
+
+# 定义响应模型
+class CompletionResponse(BaseModel):
+    id: str
+    object: str
+    created: int
+    model: str
+    choices: list
+    usage: dict
+
+# 定义生成仿真输入请求模型
+class GenerateSimulationInputRequest(BaseModel):
+    prompt: str
+    input_type: str = "text"
+
+@app.post("/v1/chat/completions")
+async def chat_completions(request: CompletionRequest):
+    """OpenAI聊天完成接口"""
+    try:
+        logger.info(f"Received chat completion request for model: {request.model}")
+        response = openai_proxy.complete(
+            model=request.model,
+            messages=request.messages,
+            temperature=request.temperature,
+            max_tokens=request.max_tokens
+        )
+        logger.info(f"Chat completion completed")
+        return response
+    except Exception as e:
+        logger.error(f"Chat completion error: {str(e)}")
+        raise HTTPException(status_code=500, detail=str(e))
+
+@app.post("/generate-simulation-input")
+async def generate_simulation_input(request: GenerateSimulationInputRequest):
+    """生成仿真输入数据"""
+    try:
+        logger.info(f"Received simulation input generation request")
+        response = openai_proxy.generate_simulation_input(
+            prompt=request.prompt,
+            input_type=request.input_type
+        )
+        logger.info(f"Simulation input generation completed")
+        return response
+    except Exception as e:
+        logger.error(f"Simulation input generation error: {str(e)}")
+        raise HTTPException(status_code=500, detail=str(e))
+
+@app.get("/health")
+async def health_check():
+    """健康检查接口"""
+    return {
+        "status": "healthy",
+        "service": "openai-proxy",
+        "version": "1.0.0"
+    }
+
+@app.get("/info")
+async def service_info():
+    """服务信息接口"""
+    return {
+        "name": "OpenAI代理服务",
+        "description": "提供OpenAI API代理功能的服务",
+        "version": "1.0.0",
+        "endpoints": {
+            "/v1/chat/completions": "POST - OpenAI聊天完成",
+            "/generate-simulation-input": "POST - 生成仿真输入数据",
+            "/health": "GET - 健康检查",
+            "/info": "GET - 服务信息"
+        }
+    }
+
+if __name__ == "__main__":
+    uvicorn.run(
+        "main:app",
+        host=settings.HOST,
+        port=settings.PORT,
+        reload=settings.DEBUG
+    )