Commit 98978c51 by 文靖昊

修改打印位置

parent 9012ab95
...@@ -46,7 +46,6 @@ class RagQuery(): ...@@ -46,7 +46,6 @@ class RagQuery():
city_ext = complete_administrative_division(m, divisions) city_ext = complete_administrative_division(m, divisions)
cities_ext.append(city_ext) cities_ext.append(city_ext)
location = [] location = []
print(location)
prompt = "" prompt = ""
for city in cities_ext: for city in cities_ext:
if city is not None and "县(区)" in city: if city is not None and "县(区)" in city:
...@@ -58,6 +57,7 @@ class RagQuery(): ...@@ -58,6 +57,7 @@ class RagQuery():
prompt += city["省"] + city["市"] + "管辖" prompt += city["省"] + city["市"] + "管辖"
for x in city["县(区)"]: for x in city["县(区)"]:
prompt += x + "," prompt += x + ","
print(location)
new_question = prompt + question new_question = prompt + question
print(new_question) print(new_question)
split_docs_list = [] split_docs_list = []
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment