contsql 0.2.0__tar.gz → 0.2.2__tar.gz

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

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

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

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

{contsql-0.2.0 → contsql-0.2.2}/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.2 | 2026-04-13 | asal sayı fixture + SUM checksum + ambiguous column retry
 """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, istisnası yok. COUNT/SUM gibi aggregation sorgularında bile GROUP BY entity_id, unvan kullan veya alt sorgu yaz.
 - 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,6 +312,27 @@ 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
@@ -296,7 +360,8 @@ def interactive_loop(conn, schema_text, domain_text):
             print(f"\n{read_schema(conn)}\n")
             continue
 
-        system_prompt = build_system_prompt(schema_text, domain_text, last_result_entities)
+        system_prompt = build_system_prompt(schema_text, domain_text, last_result_entities,
+                                           question=question)
         entities = run_query(conn, system_prompt, question)
         if entities is not None:
             last_result_entities = entities
@@ -331,7 +396,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.2}/pyproject.toml

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