quillsql 2.2.7__py3-none-any.whl → 2.2.8__py3-none-any.whl

quillsql/core.py

@@ -1,4 +1,5 @@
 import os
+import codecs
 from dotenv import load_dotenv
 
 import requests
@@ -29,7 +30,7 @@ load_dotenv()
 
 ENV = os.getenv("PYTHON_ENV")
 DEV_HOST = "http://localhost:8080"
-PROD_HOST = "https://quill-344421.uc.r.appspot.com"
+PROD_HOST = "https://api.quill.co"
 HOST = DEV_HOST if ENV == "development" else PROD_HOST
 
 SINGLE_TENANT = "QUILL_SINGLE_TENANT"
@@ -386,6 +387,80 @@ class Quill:
                 "status": "error",
                 "data": responseMetadata,
             }
+        
+    async def stream(
+        self,
+        tenants,
+        metadata,
+        flags=None,
+    ):
+        if not tenants:
+            raise ValueError("You may not pass an empty tenants array.")
+
+        if not metadata:
+            yield {"type": "error", "errorText": "Missing metadata."}
+            return
+
+        task = metadata.get("task")
+        if not task:
+            yield {"type": "error", "errorText": "Missing task."}
+            return
+
+        try:
+            # Set tenant IDs in the connection
+            self.target_connection.tenant_ids = extract_tenant_ids(tenants)
+
+            # Handle tenant flags synthesis
+            tenant_flags = None
+            if tenants[0] == SINGLE_TENANT and flags:
+                if flags and isinstance(flags[0], dict):
+                    tenant_flags = [{'tenantField': SINGLE_TENANT, 'flags': flags}]
+                else:
+                    tenant_flags = flags
+
+            payload = { 
+                **metadata,
+                "tenants": tenants,
+                "flags": tenant_flags,
+            }
+            # Custom JSON Encoder to handle Enums
+            class EnumEncoder(json.JSONEncoder):
+                def default(self, obj):
+                    if isinstance(obj, Enum):
+                        return obj.value  # Convert enum to its value (string in this case)
+                    return super().default(obj)
+            url = f"{self.baseUrl}/sdk/{task}"
+            headers = {"Authorization": f"Bearer {self.private_key}", "Content-Type": "application/json","Accept": "text/event-stream"}
+            encoded = json.dumps(payload, cls=EnumEncoder)
+
+            resp = requests.post(url, data=encoded, headers=headers, stream=True)
+            decoder = codecs.getincrementaldecoder('utf-8')()
+            buf = ""
+            for chunk in resp.iter_content(chunk_size=4096):
+                buf += decoder.decode(chunk)
+                while "\n\n" in buf:
+                    raw_event, buf = buf.split("\n\n", 1)
+                    data_lines = []
+                    for line in raw_event.splitlines():
+                        if line.startswith("data:"):
+                            data_lines.append(line[len("data:"):].strip())
+                    if not data_lines:
+                        continue
+                    payload = "\n".join(data_lines)
+                    if payload == "[DONE]":
+                        break
+                    yield json.loads(payload)
+
+            # flush any partial code points at the end
+            buf += decoder.decode(b"", final=True)
+            yield buf
+            return
+        except Exception as err:
+            yield {
+                "type": "error",
+                "errorText": str(err).splitlines()[0],
+            }
+            return
 
     def apply_limit(self, query, limit):
         # Simple logic: if query already has a limit, don't add another

{quillsql-2.2.7.dist-info → quillsql-2.2.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: quillsql
-Version: 2.2.7
+Version: 2.2.8
 Summary: Quill SDK for Python.
 Home-page: https://github.com/quill-sql/quill-python
 Author: Quill
@@ -67,3 +67,44 @@ async def quill_post(data: Request, user: dict = Depends(authenticate_jwt)):
 
 Then you can run your app like normally. Pass in this route to our react library
 on the frontend and you all set!
+
+## Streaming
+
+```python
+from quillsql import Quill
+from fastapi.responses import StreamingResponse
+import asyncio
+
+quill = Quill(
+    private_key=os.getenv("QULL_PRIVATE_KEY"),
+    database_connection_string=os.getenv("POSTGRES_READ"),
+    database_type="postgresql"
+)
+
+@app.post("/quill-stream")
+async def quill_post(data: Request, user: dict = Depends(authenticate_jwt)):
+    # assuming user fetched via auth middleware has an userId
+    user_id = user["user_id"]
+    body = await data.json()
+    metadata = body.get("metadata")
+
+    quill_stream = quill.stream(
+        tenants=[{"tenantField": "user_id", "tenantIds": [user_id]}],
+        metadata=metadata,
+    )
+
+    async def event_generator():
+        # Full event types list: https://ai-sdk.dev/docs/ai-sdk-ui/stream-protocol#data-stream-protocol
+        async for event in quill_stream:
+            if event["type"] == "start":
+                pass
+            elif event["type"] == "text-delta":
+                yield event['delta']
+            elif event["type"] == "finish":
+                return
+            elif event["type"] == "error":
+                yield event['errorText']
+            await asyncio.sleep(0)
+
+    return StreamingResponse(event_generator(), media_type="text/event-stream")
+```

{quillsql-2.2.7.dist-info → quillsql-2.2.8.dist-info}/RECORD

@@ -1,5 +1,5 @@
 quillsql/__init__.py,sha256=wjJfszle5vheUbgUfJMHQqtqhx2W3UaDN4ndcRIfmkQ,236
-quillsql/core.py,sha256=IQZlNdEAfiEuWF4PnA4hI6iK3Yk5OI8QKwKVklwW8GA,23943
+quillsql/core.py,sha256=yLDGW_BZTcY5FBxJVefCxzPBgzChy_vLHfJNylz6wuo,26737
 quillsql/error.py,sha256=n9VKHw4FAgg7ZEAz2YQ8L_8FdRG_1shwGngf2iWhUSM,175
 quillsql/assets/__init__.py,sha256=oXQ2ZS5XDXkXTYjADxNfGt55cIn_rqfgWL2EDqjTyoI,45
 quillsql/assets/pgtypes.py,sha256=-B_2wUaoAsdX7_HnJhUlx4ptZQ6x-cXwuST9ACgGFdE,33820
@@ -15,7 +15,7 @@ quillsql/utils/post_quill_executor.py,sha256=DB1RHNfqHPYarMM10vSv--UjpCZqe4qYTjq
 quillsql/utils/run_query_processes.py,sha256=QwnMr5UwXdtO_W88lv5nBaf6pJ_h5oWQnYd8K9oHQ5s,1030
 quillsql/utils/schema_conversion.py,sha256=TFfMibN9nOsxNRhHw5YIFl3jGTvipG81bxX4LFDulUY,314
 quillsql/utils/tenants.py,sha256=ZD2FuKz0gjBVSsThHDv1P8PU6EL8E009NWihE5hAH-Q,2022
-quillsql-2.2.7.dist-info/METADATA,sha256=EE-TzuvH71mqurYprv3i-rzqd4_REL4bB14WEHqPCaA,1786
-quillsql-2.2.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-quillsql-2.2.7.dist-info/top_level.txt,sha256=eU2vHnVqwpYQJ3ADl1Q-DIBzbYejZRUhcMdN_4zMCz8,9
-quillsql-2.2.7.dist-info/RECORD,,
+quillsql-2.2.8.dist-info/METADATA,sha256=qfMIakgsSk045GZzk7sNtLu-2gVIQ82toNiAp3lwif0,3052
+quillsql-2.2.8.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
+quillsql-2.2.8.dist-info/top_level.txt,sha256=eU2vHnVqwpYQJ3ADl1Q-DIBzbYejZRUhcMdN_4zMCz8,9
+quillsql-2.2.8.dist-info/RECORD,,

{quillsql-2.2.7.dist-info → quillsql-2.2.8.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (80.10.1)
 Root-Is-Purelib: true
 Tag: py3-none-any