renamed new_model to model

Author: Ne0bliviscaris

detective_test.py

@@ -1,4 +1,4 @@
-from modules.new_model import Detective
+from modules.model import Detective
 
 # question = "find last house on Franklin Ave"
 question = "find house with the largest address number on Franklin Ave"

modules/new_model.py renamed to modules/model.py

@@ -1,5 +1,4 @@
 import json
-import re
 
 from langchain.chains.sql_database.query import create_sql_query_chain
 from langchain_core.output_parsers import StrOutputParser
@@ -111,7 +110,6 @@ def build_langchain(self):
             temperature=0,
             seed=1,
             model=MODEL,
-            # num_predict=128,  # Output tokens limit
             num_predict=512,  # Output tokens limit
             top_p=0.95,
             format="json",

translator_test.py

@@ -1,4 +1,4 @@
-from modules.new_model import Translator
+from modules.model import Translator
 
 response = Translator("find all murder cases from SQL City")
 # response = Translator("Find who lives in the last house in Franklin Ave. Order house numbers")
@@ -12,3 +12,4 @@
 print((response.full_response))
 print("\nQuery:\n", response.sql_query)
 print("\nThinking:\n", response.thinking)
+# print("Prompt:\n", response.prompt())