Spaces:
Runtime error
Runtime error
Commit ·
4a9a1db
1
Parent(s): 42994c6
Return empty array for empty requests
Browse files
server.py
CHANGED
|
@@ -47,7 +47,7 @@ app.wsgi_app = ProxyFix(
|
|
| 47 |
)
|
| 48 |
|
| 49 |
def get_real_ip():
|
| 50 |
-
return request.
|
| 51 |
|
| 52 |
@app.route("/", methods=["GET"])
|
| 53 |
def index():
|
|
@@ -110,27 +110,30 @@ def chromadb_query():
|
|
| 110 |
)
|
| 111 |
|
| 112 |
n_results = min(collection.count(), n_results)
|
| 113 |
-
query_result = collection.query(
|
| 114 |
-
query_texts=[data["query"]],
|
| 115 |
-
n_results=n_results,
|
| 116 |
-
)
|
| 117 |
|
| 118 |
-
|
| 119 |
-
|
| 120 |
-
|
| 121 |
-
|
| 122 |
-
|
| 123 |
-
|
| 124 |
-
|
| 125 |
-
|
| 126 |
-
|
| 127 |
-
|
| 128 |
-
|
| 129 |
-
|
| 130 |
-
|
| 131 |
-
|
| 132 |
-
|
| 133 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 134 |
|
| 135 |
return jsonify(messages)
|
| 136 |
|
|
|
|
| 47 |
)
|
| 48 |
|
| 49 |
def get_real_ip():
|
| 50 |
+
return request.remote_addr
|
| 51 |
|
| 52 |
@app.route("/", methods=["GET"])
|
| 53 |
def index():
|
|
|
|
| 110 |
)
|
| 111 |
|
| 112 |
n_results = min(collection.count(), n_results)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 113 |
|
| 114 |
+
messages = []
|
| 115 |
+
if n_results > 0:
|
| 116 |
+
query_result = collection.query(
|
| 117 |
+
query_texts=[data["query"]],
|
| 118 |
+
n_results=n_results,
|
| 119 |
+
)
|
| 120 |
+
|
| 121 |
+
documents = query_result["documents"][0]
|
| 122 |
+
ids = query_result["ids"][0]
|
| 123 |
+
metadatas = query_result["metadatas"][0]
|
| 124 |
+
distances = query_result["distances"][0]
|
| 125 |
+
|
| 126 |
+
messages = [
|
| 127 |
+
{
|
| 128 |
+
"id": ids[i],
|
| 129 |
+
"date": metadatas[i]["date"],
|
| 130 |
+
"role": metadatas[i]["role"],
|
| 131 |
+
"meta": metadatas[i]["meta"],
|
| 132 |
+
"content": documents[i],
|
| 133 |
+
"distance": distances[i],
|
| 134 |
+
}
|
| 135 |
+
for i in range(len(ids))
|
| 136 |
+
]
|
| 137 |
|
| 138 |
return jsonify(messages)
|
| 139 |
|