maceter636 commited on
Commit
4a9a1db
·
1 Parent(s): 42994c6

Return empty array for empty requests

Browse files
Files changed (1) hide show
  1. server.py +24 -21
server.py CHANGED
@@ -47,7 +47,7 @@ app.wsgi_app = ProxyFix(
47
  )
48
 
49
  def get_real_ip():
50
- return request.environ.get('HTTP_X_FORWARDED_FOR', request.remote_addr)
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
- documents = query_result["documents"][0]
119
- ids = query_result["ids"][0]
120
- metadatas = query_result["metadatas"][0]
121
- distances = query_result["distances"][0]
122
-
123
- messages = [
124
- {
125
- "id": ids[i],
126
- "date": metadatas[i]["date"],
127
- "role": metadatas[i]["role"],
128
- "meta": metadatas[i]["meta"],
129
- "content": documents[i],
130
- "distance": distances[i],
131
- }
132
- for i in range(len(ids))
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