Update prompt for local model. Add dependency to requirements.full

LazyAGI · wzh1994 · Feb 13, 2025 · Oct 24, 2024 · Oct 25, 2024 · Oct 31, 2024
commit cbdd520363ccfb2a9ae020ea64e82aef4851c285
diff --git a/lazyllm/tools/sql/sql_manager.py b/lazyllm/tools/sql/sql_manager.py
@@ -410,7 +410,7 @@ def __init__(self, db_file: str, tables_info_dict: dict = {}):
             raise ValueError(self.detail)
 
     def reset_engine(self, db_file: str, tables_info_dict: dict):
-        self.db_type = "sqlite"
+        self._db_type = "sqlite"
         self.status = DBStatus.SUCCESS
         self.detail = ""
         if not Path(db_file).is_file():

diff --git a/lazyllm/tools/sql_call/sql_call.py b/lazyllm/tools/sql_call/sql_call.py
@@ -12,7 +12,7 @@
 
 {desc}
 
-Alert: Just replay the sql query in a code block start with keyword "sql"
+Alert: Just replay the sql query in a code block start with triple-backticks and keyword "sql"
 """  # noqa E501
 
 
@@ -24,7 +24,7 @@
 
 {desc}
 
-Note: Please return the json pipeline in a code block and start with the keyword "json".
+Note: Please return the json pipeline in a code block start with triple-backticks and keyword "json".
 """  # noqa E501
 
 db_explain_instruct_template = """

diff --git a/requirements.full.txt b/requirements.full.txt
@@ -65,3 +65,4 @@ infinity-emb
 ctranslate2
 optimum>=1.16.2
 typer
+pymongo
diff --git a/tests/charge_tests/test_mongodb_manager.py b/tests/charge_tests/test_mongodb_manager.py
diff --git a/tests/charge_tests/test_sql_tool.py b/tests/charge_tests/test_sql_tool.py
@@ -49,7 +49,7 @@ def setUpClass(cls):
                 assert db_result.status == DBStatus.SUCCESS, db_result.detail
 
         # Recommend to use sensenova, gpt-4o, qwen online model
-        sql_llm = lazyllm.OnlineChatModule(source="sensenova")
+        sql_llm = lazyllm.OnlineChatModule(source="qwen")
         cls.sql_calls: list[SqlCall] = []
         for sql_manager in cls.sql_managers:
             cls.sql_calls.append(SqlCall(sql_llm, sql_manager, use_llm_for_sql_result=True))
-Original file line number
+Diff line change
@@ @@ -65,3 +65,4 @@ infinity-emb @@
     ctranslate2
     optimum>=1.16.2
     typer
+    pymongo