contextsql-garvit 0.1.2__tar.gz → 0.1.4__tar.gz

{contextsql_garvit-0.1.2 → contextsql_garvit-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: contextsql_garvit
-Version: 0.1.2
+Version: 0.1.4
 Summary: Local, free, business-context-aware text-to-SQL agent
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown

{contextsql_garvit-0.1.2 → contextsql_garvit-0.1.4}/contextsql/cli.py

@@ -7,7 +7,7 @@ from .config import CONFIG, KNOW, LOG, load_json, save_json
 from .db import make_url, get_engine
 
 HF_REPO = "Garvitarunav/contextsql-v4"
-HF_FILE = "contextsql-v4-q4_k_m.gguf"
+HF_FILE = "contextsql-v5-q4_k_m.gguf"
 
 def default_model():
     p = Path("models/contextsql-q4_k_m.gguf")
@@ -81,7 +81,8 @@ def cmd_ask(a):
     from .agent import Agent, LlamaGenerator
     cfg = need(CONFIG, "contextsql connect")
     know = need(KNOW, "contextsql learn")
-    if not cfg.get("model") or not Path(cfg["model"]).exists():
+    m = cfg.get("model") or ""
+    if not m or not Path(m).exists() or ("contextsql-v" in m and not m.endswith(HF_FILE)):
         cfg["model"] = default_model()
         save_json(CONFIG, cfg)
     print("Loading model ...")
@@ -89,16 +90,17 @@ def cmd_ask(a):
     q = " ".join(a.question).strip()
     if q:
         show(agent.ask(q)); return
-    print("Ask anything about your data (empty line to quit).")
+    print("Ask anything about your data (empty line to quit). Commands: /good  /fix SQL  /status  /glossary list  cls")
     while True:
         q = input("\n> ").strip()
         if not q: break
         if q.lower() in ("cls", "clear", "/cls", "/clear"):
             os.system("cls" if os.name == "nt" else "clear"); continue
         if q.startswith("/") or q.startswith("contextsql "):
-            try: 
-                main(shlex.split(q[1:] if q.startswith("/") else q[11:]))
-            except SystemExit as e: 
+            try:
+                cmd = q[1:] if q.startswith("/") else q[11:]
+                main(["fix", cmd[4:].strip()] if cmd.startswith("fix ") else shlex.split(cmd))
+            except SystemExit as e:
                 print(e.code if isinstance(e.code, str) else "")
             continue
         show(agent.ask(q))

{contextsql_garvit-0.1.2 → contextsql_garvit-0.1.4}/contextsql_garvit.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: contextsql_garvit
-Version: 0.1.2
+Version: 0.1.4
 Summary: Local, free, business-context-aware text-to-SQL agent
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown

{contextsql_garvit-0.1.2 → contextsql_garvit-0.1.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "contextsql_garvit"
-version = "0.1.2"
+version = "0.1.4"
 description = "Local, free, business-context-aware text-to-SQL agent"
 readme = "README.md"
 requires-python = ">=3.9"