update

openai_function_utils/openai_function_impl.py

@@ -59,12 +59,14 @@ def get_publication_by_year(year: str):
     with open(database_addr, 'r') as fin:
         all_pub_info = json.load(fin)
     data = {}
-    for field in all_pub_info:
-        print(f'field: {field}')
-        to_search = all_pub_info[field]
-        for i in to_search:
-            if int(year) == i['year']:
-                data.update(i)
+    for pub in all_pub_info:
+        if int(year) == pub['year']:
+            data.update(pub)
+    # for field in all_pub_info:
+    #     to_search = all_pub_info[field]
+    #     for i in to_search:
+    #         if int(year) == i['year']:
+    #             data.update(i)
     return json.dumps(data)
 
 

static/index.html

@@ -33,7 +33,7 @@
         <button class="question_btn">Give me a pub of J coleman</button>
         <button class="question_btn">What is the position of J coleman</button>
         <button class="question_btn">Give me a pub on neural network</button>
-        <button class="question_btn">Give me a pub published in 2023 (deprecated)</button>
+        <button class="question_btn">Give me a pub published in 2023</button>
         <button class="question_btn">Give me a pub on math</button>
     </div>
 </body>