add ip

main.py

@@ -1,8 +1,9 @@
-from fastapi import FastAPI
+from fastapi import FastAPI,Request
 from service.vector_store import vector_store
 from fastapi.responses import StreamingResponse
 from service.api import api
 from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
 
 app = FastAPI()
 app.add_middleware(
@@ -13,12 +14,23 @@ app.add_middleware(
     allow_headers=["*"],  # 允许所有头
 )
 
+class Query(BaseModel):
+    endpoint: str
+    query: str
+
+
+
 @app.get("/")
 def read_root():
     return {"Hello": "World!"}
 
 @app.get("/api/search")
-def search(inputs: str):
+def search(request: Request, inputs: str):
+    client_host = request.client.host
+    x_forwarded_for = request.headers.get("x-forwarded-for")
+    if x_forwarded_for:
+        client_host = x_forwarded_for.split(",")[0]  # 取X-Forwarded-For头的第一个值
+    print(f"client_host: {client_host}")
     return vector_store.search(inputs)
 
 @app.get("/api/suggestion")