contsql 0.2.0__tar.gz → 0.2.3__tar.gz

{contsql-0.2.0 → contsql-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: contsql
-Version: 0.2.0
+Version: 0.2.3
 Requires-Python: >=3.10
 Requires-Dist: duckdb
 Requires-Dist: requests

{contsql-0.2.0 → contsql-0.2.3}/contsql.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: contsql
-Version: 0.2.0
+Version: 0.2.3
 Requires-Python: >=3.10
 Requires-Dist: duckdb
 Requires-Dist: requests

{contsql-0.2.0 → contsql-0.2.3}/contsql.py

@@ -1,5 +1,5 @@
 #!/usr/bin/env python3
-# v0.2 | 2026-04-13 | case insensitive ILIKE guardrail + önceki sorgu entity context
+# v0.2.3 | 2026-04-13 | entity_id+unvan zorunlu kural + slash commands (/s /schema /trace /help)
 """contsql — Minimal DuckDB SQL agent. Soru sor, SQL üret, çalıştır, göster."""
 
 import argparse
@@ -58,9 +58,25 @@ def read_domain_notes(db_path):
     return ""
 
 
+# ── Referans algılama ──
+
+REFERANS_TRIGGERS = [
+    "bu firma", "bunlar", "bunların", "yukarıdaki", "yukarıdakiler",
+    "aynı firma", "o firma", "önceki", "listedeki", "sonuçtaki",
+    "bu müşteri", "bu muta", "onların",
+]
+
+
+def has_reference_trigger(question):
+    """Kullanıcı sorusu önceki sorgu sonucuna referans veriyor mu?"""
+    q_lower = question.lower()
+    return any(trigger in q_lower for trigger in REFERANS_TRIGGERS)
+
+
 # ── System prompt ──
 
-def build_system_prompt(schema_text, domain_text="", last_result_entities=None):
+def build_system_prompt(schema_text, domain_text="", last_result_entities=None,
+                        question=None):
     prompt = f"""Sen bir SQL asistanısın. Kullanıcının sorusuna uygun SQL yaz.
 
 Kurallar:
@@ -69,6 +85,7 @@ Kurallar:
 - SQL öncesi veya sonrası açıklama ekleme.
 - Emin değilsen "Bu soruyu mevcut tablolarla cevaplayamıyorum" de.
 - Veri uydurma. Sorgu sonucu olmadan liste verme.
+- HER sorguda entity_id ve unvan kolonlarını dahil et. Firmalar bu iki alanla tanımlanır. Sadece COUNT/SUM gibi tek değer döndüren aggregation sorgularında entity_id gerekmez.
 - String karşılaştırmalarında LIKE yerine her zaman ILIKE kullan. Türkçe karakter eşleştirmesi (İ↔i, I↔ı, Ş↔ş, Ü↔ü, Ö↔ö, Ç↔ç, Ğ↔ğ) için ILIKE şart.
 
 Veritabanı şeması:
@@ -76,11 +93,10 @@ Veritabanı şeması:
 """
     if domain_text:
         prompt += f"\nDomain bilgisi:\n{domain_text}\n"
-    if last_result_entities:
+    if last_result_entities and question and has_reference_trigger(question):
         prompt += (
             f"\nÖNCEKİ SORGU SONUCUNDAKI FİRMALAR (entity_id): {last_result_entities}\n"
-            "Kullanıcı 'bu firmalar', 'bunların', 'aynıları', 'yukarıdakiler' gibi "
-            "referans verirse bu entity_id listesini WHERE koşulunda kullan.\n"
+            "Bu entity_id listesini WHERE koşulunda kullan.\n"
         )
     return prompt
 
@@ -154,6 +170,33 @@ def ask_model(system_prompt, question):
         return f"LLM HATA: {e}", time.time() - t0, 0
 
 
+def generate_sql(conn, question, last_result_entities=None, domain_text=""):
+    """Soru → SQL string. Test runner için callable. Başarısızsa None."""
+    schema_text = read_schema(conn)
+    system_prompt = build_system_prompt(schema_text, domain_text, last_result_entities,
+                                       question=question)
+    response, _, _ = ask_model(system_prompt, question)
+    sql = extract_sql(response)
+    if not sql or check_sql_safety(sql):
+        return None
+    sql = _like_to_ilike(sql)
+
+    # Ambiguous column retry: EXPLAIN ile ön kontrol
+    try:
+        conn.execute(f"EXPLAIN {sql}")
+    except Exception as e:
+        if "ambiguous" not in str(e).lower():
+            return sql
+        retry_q = f"{question}\n\nÖNCEKİ SQL HATA: {e}\nJOIN'de tablo alias kullan."
+        resp2, _, _ = ask_model(system_prompt, retry_q)
+        sql2 = extract_sql(resp2)
+        if sql2 and not check_sql_safety(sql2):
+            return _like_to_ilike(sql2)
+        return None
+
+    return sql
+
+
 # ── Result formatting ──
 
 def _fmt_value(v):
@@ -269,17 +312,72 @@ def run_query(conn, system_prompt, question):
             print("  ⚠ Önceki sorgu çok geniş — firma referansı için soruyu daraltın.")
         return entities
     except duckdb.Error as e:
+        # Ambiguous column retry
+        if "ambiguous" in str(e).lower():
+            print(f"🔄 Ambiguous column, retry...")
+            retry_q = f"{question}\n\nSQL HATA: {e}\nJOIN'de tablo alias kullan."
+            resp2, _, _ = ask_model(system_prompt, retry_q)
+            sql2 = extract_sql(resp2)
+            if sql2 and not check_sql_safety(sql2):
+                sql2 = _like_to_ilike(sql2)
+                print(f"🔍 Retry SQL: {sql2}")
+                try:
+                    result = conn.execute(sql2)
+                    columns = [desc[0] for desc in result.description]
+                    rows = result.fetchall()
+                    query_ms = (time.time() - t0) * 1000
+                    print(f"\n📊 SONUÇ ({len(rows)} satır, {query_ms:.0f}ms)")
+                    print(format_table(columns, rows))
+                    entities = _extract_entity_ids(columns, rows)
+                    return entities
+                except duckdb.Error as e2:
+                    print(f"\n❌ Retry hatası: {e2}")
+                    return None
         print(f"\n❌ SQL hatası: {e}")
         print(f"🔍 SQL: {sql}")
         return None
 
 
+def handle_slash_command(cmd, state):
+    """Slash command işle. True → normal sorgu akışına girme."""
+    cmd = cmd.strip().lower()
+
+    if cmd == "/s":
+        state["last_result_entities"] = None
+        print("🧹 Bellek temizlendi.")
+        return True
+
+    if cmd == "/schema":
+        print(f"\n{state['schema_text']}\n")
+        return True
+
+    if cmd == "/trace":
+        state["trace"] = not state.get("trace", False)
+        print(f"🔍 Trace: {'açık' if state['trace'] else 'kapalı'}")
+        return True
+
+    if cmd == "/help":
+        print("Komutlar:")
+        print("  /s       — önceki sorgu hafızasını temizle")
+        print("  /schema  — veritabanı şemasını göster")
+        print("  /trace   — SQL trace modunu aç/kapa")
+        print("  /help    — bu mesaj")
+        print("  quit     — çıkış")
+        return True
+
+    return False
+
+
 def interactive_loop(conn, schema_text, domain_text):
     """REPL döngüsü."""
-    print(f"\ncontsql hazır. Model: {MODEL}")
+    print(f"\ncontsql hazır. Model: {MODEL} | /help komutlar")
     print("Çıkmak için: quit/exit/q\n")
 
-    last_result_entities = None
+    state = {
+        "last_result_entities": None,
+        "trace": False,
+        "schema_text": schema_text,
+    }
 
     while True:
         try:
@@ -292,14 +390,18 @@ def interactive_loop(conn, schema_text, domain_text):
             continue
         if question.lower() in ("quit", "exit", "q", "çık"):
             break
-        if question.lower() in ("schema", "şema"):
-            print(f"\n{read_schema(conn)}\n")
+
+        if question.startswith("/"):
+            if not handle_slash_command(question, state):
+                print(f"Bilinmeyen komut: {question}. /help yazın.")
             continue
 
-        system_prompt = build_system_prompt(schema_text, domain_text, last_result_entities)
+        system_prompt = build_system_prompt(schema_text, domain_text,
+                                           state["last_result_entities"],
+                                           question=question)
         entities = run_query(conn, system_prompt, question)
         if entities is not None:
-            last_result_entities = entities
+            state["last_result_entities"] = entities
         print()
 
 
@@ -331,7 +433,8 @@ def main():
 
     # Tek soru veya interaktif
     if args.question:
-        system_prompt = build_system_prompt(schema_text, domain_text)
+        system_prompt = build_system_prompt(schema_text, domain_text,
+                                           question=args.question)
         run_query(conn, system_prompt, args.question)
     else:
         interactive_loop(conn, schema_text, domain_text)

{contsql-0.2.0 → contsql-0.2.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "contsql"
-version = "0.2.0"
+version = "0.2.3"
 requires-python = ">=3.10"
 dependencies = ["duckdb", "requests"]