Fixed tests and bib.

doc_generator/query/create_chat_chain.py

@@ -227,8 +227,7 @@ def make_qa_chain(
     )
 
     return create_retrieval_chain(
-        retriever=question_generator,
-        combine_docs_chain=doc_chain
+        retriever=question_generator, combine_docs_chain=doc_chain
     )
 
 

doc_generator/query/query.py

@@ -28,8 +28,8 @@ def display_welcome_message(project_name):
     """Display Welcome Message"""
     print(f"Welcome to the {project_name} chatbot.")
     print(
-        f"Ask any questions related to the {project_name} codebase, "+
-        "and I'll try to help. Type 'exit' to quit.\n"
+        f"Ask any questions related to the {project_name} codebase, "
+        + "and I'll try to help. Type 'exit' to quit.\n"
     )
 
 

paper/paper.bib

@@ -8,7 +8,8 @@ @article{chomsky1956three
   number={3},
   pages={113--124},
   year={1956},
-  publisher={IEEE}
+  publisher={IEEE},
+  doi={10.1109/TIT.1956.1056813}
 }
 
 @article{miller2003cognitive,
@@ -19,21 +20,24 @@ @article{miller2003cognitive
   number={3},
   pages={141--144},
   year={2003},
-  publisher={Elsevier}
+  publisher={Elsevier},
+  doi={10.1016/S1364-6613(03)00029-9}
 }
 
 @article{graves2014neural,
   title={Neural turing machines},
   author={Graves, Alex and Wayne, Greg and Danihelka, Ivo},
   journal={arXiv preprint arXiv:1410.5401},
-  year={2014}
+  year={2014},
+  doi={10.48550/arXiv.1410.5401}
 }
 
 @article{bahdanau2014neural,
   title={Neural machine translation by jointly learning to align and translate},
   author={Bahdanau, Dzmitry and Cho, Kyunghyun and Bengio, Yoshua},
   journal={arXiv preprint arXiv:1409.0473},
-  year={2014}
+  year={2014},
+  doi={10.48550/arXiv.1409.0473}
 }
 
 @inproceedings{iyer2016summarizing,
@@ -42,21 +46,24 @@ @inproceedings{iyer2016summarizing
   booktitle={54th Annual Meeting of the Association for Computational Linguistics 2016},
   pages={2073--2083},
   year={2016},
-  organization={Association for Computational Linguistics}
+  organization={Association for Computational Linguistics},
+  doi={10.18653/v1/P16-1208}
 }
 
 @article{ling2016latent,
   title={Latent predictor networks for code generation},
   author={Ling, Wang and Grefenstette, Edward and Hermann, Karl Moritz and Ko{\v{c}}isk{\`y}, Tom{\'a}{\v{s}} and Senior, Andrew and Wang, Fumin and Blunsom, Phil},
   journal={arXiv preprint arXiv:1603.06744},
-  year={2016}
+  year={2016},
+  doi={10.48550/arXiv.1603.06744}
 }
 
 @article{yin2017syntactic,
   title={A syntactic neural model for general-purpose code generation},
   author={Yin, Pengcheng and Neubig, Graham},
   journal={arXiv preprint arXiv:1704.01696},
-  year={2017}
+  year={2017},
+  doi={10.48550/arXiv.1704.01696}
 }
 
 @inproceedings{allamanis2013mining,
@@ -65,14 +72,16 @@ @inproceedings{allamanis2013mining
   booktitle={2013 10th working conference on mining software repositories (MSR)},
   pages={207--216},
   year={2013},
-  organization={IEEE}
+  organization={IEEE},
+  doi={10.1109/MSR.2013.6624030}
 }
 
 @article{bhoopchand2016learning,
   title={Learning python code suggestion with a sparse pointer network},
   author={Bhoopchand, Avishkar and Rockt{\"a}schel, Tim and Barr, Earl and Riedel, Sebastian},
   journal={arXiv preprint arXiv:1611.08307},
-  year={2016}
+  year={2016},
+  doi={10.48550/arXiv.1611.08307}
 }
 
 @inproceedings{oda2015learning,
@@ -81,15 +90,17 @@ @inproceedings{oda2015learning
   booktitle={2015 30th IEEE/ACM International Conference on Automated Software Engineering (ASE)},
   pages={574--584},
   year={2015},
-  organization={IEEE}
+  organization={IEEE},
+  doi={10.1109/ASE.2015.72}
 }
 
 @inproceedings{quirk2015language,
   title={Language to code: Learning semantic parsers for if-this-then-that recipes},
   author={Quirk, Chris and Mooney, Raymond and Galley, Michel},
   booktitle={Proceedings of the 53rd Annual Meeting of the Association for Computational Linguistics and the 7th International Joint Conference on Natural Language Processing (Volume 1: Long Papers)},
   pages={878--888},
-  year={2015}
+  year={2015},
+  doi={10.3115/v1/P15-1085}
 }
 
 @article{radford2019language,
@@ -108,7 +119,8 @@ @article{brown2020language
   journal={Advances in neural information processing systems},
   volume={33},
   pages={1877--1901},
-  year={2020}
+  year={2020},
+  doi={10.48550/arXiv.2005.14165}
 }
 
 @article{ouyang2022training,
@@ -117,22 +129,25 @@ @article{ouyang2022training
   journal={Advances in neural information processing systems},
   volume={35},
   pages={27730--27744},
-  year={2022}
+  year={2022},
+  doi={10.48550/arXiv.2203.02155}
 }
 
 @article{vaswani2017attention,
   title={Attention is all you need},
   author={Vaswani, Ashish and Shazeer, Noam and Parmar, Niki and Uszkoreit, Jakob and Jones, Llion and Gomez, Aidan N and Kaiser, {\L}ukasz and Polosukhin, Illia},
   journal={Advances in neural information processing systems},
   volume={30},
-  year={2017}
+  year={2017},
+  doi={10.48550/arXiv.1706.03762}
 }
 
 @article{lester2021power,
   title={The power of scale for parameter-efficient prompt tuning},
   author={Lester, Brian and Al-Rfou, Rami and Constant, Noah},
   journal={arXiv preprint arXiv:2104.08691},
-  year={2021}
+  year={2021},
+  doi={10.48550/arXiv.2104.08691}
 }
 
 @misc{gpt-3.5-turbo,
@@ -243,7 +258,8 @@ @article{barone2017parallel
   title={A parallel corpus of python functions and documentation strings for automated code documentation and code generation},
   author={Barone, Antonio Valerio Miceli and Sennrich, Rico},
   journal={arXiv preprint arXiv:1707.02275},
-  year={2017}
+  year={2017},
+  doi={10.48550/arXiv.1707.02275}
 }
 
 @article{malkov2018efficient,
@@ -254,14 +270,16 @@ @article{malkov2018efficient
   number={4},
   pages={824--836},
   year={2018},
-  publisher={IEEE}
+  publisher={IEEE},
+  doi={10.1109/TPAMI.2018.2889473}
 }
 
 @article{dettmers2023qlora,
   title={QLoRA: Efficient Finetuning of Quantized LLMs},
   author={Dettmers, Tim and Pagnoni, Artidoro and Holtzman, Ari and Zettlemoyer, Luke},
   journal={arXiv preprint arXiv:2305.14314},
-  year={2023}
+  year={2023},
+  doi={10.48550/arXiv.2305.14314}
 }
 
 @inproceedings{
@@ -270,29 +288,33 @@ @inproceedings{
 author={Edward J Hu and Yelong Shen and Phillip Wallis and Zeyuan Allen-Zhu and Yuanzhi Li and Shean Wang and Lu Wang and Weizhu Chen},
 booktitle={International Conference on Learning Representations},
 year={2022},
-url={https://openreview.net/forum?id=nZeVKeeFYf9}
+url={https://openreview.net/forum?id=nZeVKeeFYf9},
+doi={10.48550/arXiv.2106.09685}
 }
 
 @article{zhang2023dynamically,
   title={: A Dynamically Adaptive Defense to a Novel Attack on Review Fraud Detection Engines},
   author={Zhang, Youzhi and Chakrabarty, Sayak and Liu, Rui and Pugliese, Andrea and Subrahmanian, VS},
   journal={IEEE Transactions on Computational Social Systems},
   year={2023},
-  publisher={IEEE}
+  publisher={IEEE},
+  doi={10.1109/TCSS.2023.3248149}
 }
 
 @article{makarychev2024single,
   title={Single-Pass Pivot Algorithm for Correlation Clustering. Keep it simple!},
   author={Makarychev, Konstantin and Chakrabarty, Sayak},
   journal={Advances in Neural Information Processing Systems},
   volume={36},
-  year={2024}
+  year={2024},
+  doi={10.48550/arXiv.2305.13560}
 }
 
 @article{datta2024consistency,
   title={On the consistency of maximum likelihood estimation of probabilistic principal component analysis},
   author={Datta, Arghya and Chakrabarty, Sayak},
   journal={Advances in Neural Information Processing Systems},
   volume={36},
-  year={2024}
+  year={2024},
+  doi={10.48550/arXiv.2311.05046}
 }