cp dines

Author: dines-rl

src/runloop_api_client/sdk/async_axon.py

@@ -64,7 +64,11 @@ class AsyncAxon:
     def __init__(self, client: AsyncRunloop, axon_id: str) -> None:
         self._client = client
         self._id = axon_id
-        self.sql = AsyncAxonSqlOps(client, axon_id)
+        self._sql = AsyncAxonSqlOps(client, axon_id)
+
+    @property
+    def sql(self) -> AsyncAxonSqlOps:
+        return self._sql
 
     @override
     def __repr__(self) -> str:

src/runloop_api_client/sdk/axon.py

@@ -64,7 +64,11 @@ class Axon:
     def __init__(self, client: Runloop, axon_id: str) -> None:
         self._client = client
         self._id = axon_id
-        self.sql = AxonSqlOps(client, axon_id)
+        self._sql = AxonSqlOps(client, axon_id)
+
+    @property
+    def sql(self) -> AxonSqlOps:
+        return self._sql
 
     @override
     def __repr__(self) -> str:

tests/smoketests/sdk/test_async_axon.py

@@ -3,11 +3,17 @@
 from __future__ import annotations
 
 import json
+import uuid
 
 import pytest
 
 from runloop_api_client.sdk import AsyncRunloopSDK
 
+
+def _unique_table() -> str:
+    return f"t_{uuid.uuid4().hex[:12]}"
+
+
 pytestmark = [pytest.mark.smoketest, pytest.mark.asyncio]
 
 THIRTY_SECOND_TIMEOUT = 30
@@ -76,12 +82,13 @@ class TestAsyncAxonSql:
     async def test_sql_query_create_and_select(self, async_sdk_client: AsyncRunloopSDK) -> None:
         """Test creating a table and querying it via sql.query."""
         axon = await async_sdk_client.axon.create()
+        table = _unique_table()
 
-        await axon.sql.query(sql="CREATE TABLE IF NOT EXISTS smoke_test (id INTEGER PRIMARY KEY, value TEXT)")
+        await axon.sql.query(sql=f"CREATE TABLE {table} (id INTEGER PRIMARY KEY, value TEXT)")
 
-        await axon.sql.query(sql="INSERT INTO smoke_test (id, value) VALUES (?, ?)", params=[1, "hello"])
+        await axon.sql.query(sql=f"INSERT INTO {table} (id, value) VALUES (?, ?)", params=[1, "hello"])
 
-        result = await axon.sql.query(sql="SELECT * FROM smoke_test WHERE id = ?", params=[1])
+        result = await axon.sql.query(sql=f"SELECT * FROM {table} WHERE id = ?", params=[1])
 
         assert result.columns is not None
         assert len(result.columns) > 0
@@ -92,13 +99,14 @@ async def test_sql_query_create_and_select(self, async_sdk_client: AsyncRunloopS
     async def test_sql_batch(self, async_sdk_client: AsyncRunloopSDK) -> None:
         """Test executing multiple statements atomically via sql.batch."""
         axon = await async_sdk_client.axon.create()
+        table = _unique_table()
 
         result = await axon.sql.batch(
             statements=[
-                {"sql": "CREATE TABLE IF NOT EXISTS batch_test (id INTEGER PRIMARY KEY, name TEXT)"},
-                {"sql": "INSERT INTO batch_test (id, name) VALUES (?, ?)", "params": [1, "alice"]},
-                {"sql": "INSERT INTO batch_test (id, name) VALUES (?, ?)", "params": [2, "bob"]},
-                {"sql": "SELECT * FROM batch_test ORDER BY id"},
+                {"sql": f"CREATE TABLE {table} (id INTEGER PRIMARY KEY, name TEXT)"},
+                {"sql": f"INSERT INTO {table} (id, name) VALUES (?, ?)", "params": [1, "alice"]},
+                {"sql": f"INSERT INTO {table} (id, name) VALUES (?, ?)", "params": [2, "bob"]},
+                {"sql": f"SELECT * FROM {table} ORDER BY id"},
             ],
         )
 

tests/smoketests/sdk/test_axon.py

@@ -13,6 +13,7 @@
 def _unique_table() -> str:
     return f"t_{uuid.uuid4().hex[:12]}"
 
+
 pytestmark = [pytest.mark.smoketest]
 
 THIRTY_SECOND_TIMEOUT = 30
@@ -98,13 +99,14 @@ def test_sql_query_create_and_select(self, sdk_client: RunloopSDK) -> None:
     def test_sql_batch(self, sdk_client: RunloopSDK) -> None:
         """Test executing multiple statements atomically via sql.batch."""
         axon = sdk_client.axon.create()
+        table = _unique_table()
 
         result = axon.sql.batch(
             statements=[
-                {"sql": "CREATE TABLE IF NOT EXISTS batch_test (id INTEGER PRIMARY KEY, name TEXT)"},
-                {"sql": "INSERT INTO batch_test (id, name) VALUES (?, ?)", "params": [1, "alice"]},
-                {"sql": "INSERT INTO batch_test (id, name) VALUES (?, ?)", "params": [2, "bob"]},
-                {"sql": "SELECT * FROM batch_test ORDER BY id"},
+                {"sql": f"CREATE TABLE {table} (id INTEGER PRIMARY KEY, name TEXT)"},
+                {"sql": f"INSERT INTO {table} (id, name) VALUES (?, ?)", "params": [1, "alice"]},
+                {"sql": f"INSERT INTO {table} (id, name) VALUES (?, ?)", "params": [2, "bob"]},
+                {"sql": f"SELECT * FROM {table} ORDER BY id"},
             ],
         )