diff --git a/cryptoai/routes/agent.py b/cryptoai/routes/agent.py index 72655a8..1422479 100644 --- a/cryptoai/routes/agent.py +++ b/cryptoai/routes/agent.py @@ -17,6 +17,7 @@ from cryptoai.utils.config_loader import ConfigLoader from fastapi.responses import StreamingResponse from cryptoai.routes.user import get_current_user import requests +from datetime import datetime from cryptoai.utils.db_manager import get_db_manager # 创建路由 @@ -42,7 +43,7 @@ async def get_agents(current_user: Dict[str, Any] = Depends(get_current_user)): { "id": "2", "name": "美股交易助手", - "hello_prompt": "您好,我是美股交易助手,为您提供专业的股票交易分析和建议", + "hello_prompt": "您好,我是美股交易助手,您可以直接输入股票名称,比如AAPL,然后我会为您提供专业的股票交易分析和建议", "description": "帮你分析做美股股票技术分析", }, # { @@ -66,13 +67,19 @@ async def chat(request: ChatRequest,current_user: Dict[str, Any] = Depends(get_c else: raise HTTPException(status_code=400, detail="Invalid agent ID") + inputs = {} + if request.agent_id == "2": + inputs = { + "current_date": datetime.now().strftime("%Y-%m-%d") + } + url = "https://mate.aimateplus.com/v1/chat-messages" headers = { "Authorization": f"Bearer {token}", "Content-Type": "application/json" } data = { - "inputs" : {}, + "inputs" : inputs, "query" : request.user_prompt, "response_mode" : "streaming", "user" : current_user["mail"] diff --git a/docker-compose.yml b/docker-compose.yml index e4cc555..1af7139 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -29,7 +29,7 @@ services: cryptoai-api: build: . container_name: cryptoai-api - image: cryptoai-api:0.0.9 + image: cryptoai-api:0.0.10 restart: always ports: - "8000:8000"