From 1bea2491c5c001409b905be8ea0adf37c67c8628 Mon Sep 17 00:00:00 2001
From: root <953994191@qq.com>
Date: Tue, 14 Apr 2026 03:12:23 +0800
Subject: [PATCH] =?UTF-8?q?=E7=AB=AF=E5=8F=A3=E8=AE=BF=E9=97=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 docker/docker-compose.yml | 2 +-
 frontend/frontend.py      | 8 +++-----
 2 files changed, 4 insertions(+), 6 deletions(-)

diff --git a/docker/docker-compose.yml b/docker/docker-compose.yml
index 4177d01..4c84944 100644
--- a/docker/docker-compose.yml
+++ b/docker/docker-compose.yml
@@ -46,7 +46,7 @@ services:
       dockerfile: docker/Dockerfile.frontend
     container_name: ai-frontend
     environment:
-      - API_URL=/ai/api/chat # 通过 Nginx 反向代理访问后端（路径前缀 /ai）
+      - API_URL=http://backend:8001/chat # Docker 内部使用服务名解析
     ports:
       - "8501:8501"
     networks:
diff --git a/frontend/frontend.py b/frontend/frontend.py
index 17cf977..f10f458 100644
--- a/frontend/frontend.py
+++ b/frontend/frontend.py
@@ -3,6 +3,7 @@ Streamlit 前端 - 支持模型选择
 """
 
 # 标准库
+import os
 import uuid
 
 # 第三方库
@@ -10,11 +11,8 @@ import requests
 import streamlit as st
 
 
-# 原来的硬编码,本地测试
-# API_URL = "http://115.190.121.151:8001/chat"
-
-# 改为相对路径（由 Nginx 代理转发，路径前缀为 /ai）
-API_URL = "/ai/api/chat"
+# 后端 API 地址：优先从环境变量读取，Docker 环境使用服务名，本地开发可覆盖
+API_URL = os.getenv("API_URL", "http://backend:8001/chat")
 
 st.set_page_config(page_title="AI 个人助手", page_icon="🤖")
 st.title("🤖 个人生活与数据分析助手")