diff --git a/backend.py b/backend.py index a75e82c..d047b7a 100644 --- a/backend.py +++ b/backend.py @@ -14,9 +14,9 @@ from util.log_util import logger @asynccontextmanager async def lifespan(app : FastAPI): # logger.info(f'{AppConfig.app_name}开始启动,当前运行环境{os.environ.get('APP_ENV', '未读取到环境变量APP_ENV')}') - print(f'开始启动') + logger.info(f'开始启动') yield - print(f'关闭成功') + logger.info(f'关闭成功') app = FastAPI(title="RAG SQL Generator API", description="调用RAG生成SQL语句的后端服务", lifespan=lifespan) diff --git a/middlewares/cors_middleware.py b/middlewares/cors_middleware.py index 55508e7..9dbd4ed 100644 --- a/middlewares/cors_middleware.py +++ b/middlewares/cors_middleware.py @@ -10,15 +10,15 @@ def add_cors_middleware(app: FastAPI): :return: """ # 前端页面url - origins = [ - 'http://localhost:80', - 'http://127.0.0.1:80', - ] + # origins = [ + # 'http://localhost:80', + # 'http://127.0.0.1:80', + # ] # 后台api允许跨域 app.add_middleware( CORSMiddleware, - allow_origins=origins, + allow_origins=["*"], allow_credentials=True, allow_methods=['*'], allow_headers=['*'], diff --git a/util/use_pgvector.py b/util/use_pgvector.py index 836a418..9a95c2f 100644 --- a/util/use_pgvector.py +++ b/util/use_pgvector.py @@ -3,6 +3,7 @@ from psycopg2.extras import execute_values import numpy as np from env import PgvectorDataBaseConfig +from util.log_util import logger # 数据库连接配置 DB_CONFIG = { @@ -18,10 +19,10 @@ def connect_to_db(): """建立数据库连接""" try: conn = psycopg2.connect(**DB_CONFIG) - print("成功连接到数据库") + logger.info("成功连接到数据库") return conn except Exception as e: - print(f"连接数据库失败: {e}") + logger.error(f"连接数据库失败: {e}") return None @@ -51,9 +52,9 @@ def setup_vector_extension(conn): ); """) conn.commit() - print("成功设置向量扩展和表结构") + # print("成功设置向量扩展和表结构") except Exception as e: - print(f"设置数据库结构失败: {e}") + logger.error(f"设置数据库结构失败: {e}") conn.rollback() @@ -68,9 +69,9 @@ def insert_vectors(conn, data): data ) conn.commit() - print(f"成功插入 {len(data)} 条向量数据") + logger.info(f"成功插入 {len(data)} 条向量数据") except Exception as e: - print(f"插入数据失败: {e}") + logger.error(f"插入数据失败: {e}") conn.rollback() @@ -88,7 +89,7 @@ def search_similar_table(conn, query_vector, limit=5): results = cur.fetchall() return results except Exception as e: - print(f"搜索相似向量失败: {e}") + logger.error(f"搜索相似向量失败: {e}") return [] def search_similar_case(conn, query_vector, limit=5): @@ -105,7 +106,7 @@ def search_similar_case(conn, query_vector, limit=5): results = cur.fetchall() return results except Exception as e: - print(f"搜索相似向量失败: {e}") + logger.error(f"搜索相似向量失败: {e}") return [] def main(): @@ -133,9 +134,9 @@ def main(): query_vector = [1.0, 1.0, 1.0] similar_docs = search_similar_table(conn, query_vector) - print("\n相似文档搜索结果:") + logger.info("\n相似文档搜索结果:") for content, distance in similar_docs: - print(f"内容: {content}, 距离: {distance}") + logger.info(f"内容: {content}, 距离: {distance}") # 关闭连接 conn.close()