dc-python-sdk 1.5.27__tar.gz → 1.5.28__tar.gz

{dc_python_sdk-1.5.27/src/dc_python_sdk.egg-info → dc_python_sdk-1.5.28}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dc-python-sdk
-Version: 1.5.27
+Version: 1.5.28
 Summary: Data Connector Python SDK
 Home-page: https://github.com/data-connector/dc-python-sdk
 Author: DataConnector

{dc_python_sdk-1.5.27 → dc_python_sdk-1.5.28}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "dc-python-sdk"
-version = "1.5.27"
+version = "1.5.28"
 description = "Data Connector Python SDK"
 readme = "README.md"
 requires-python = ">=3.6"

{dc_python_sdk-1.5.27 → dc_python_sdk-1.5.28}/setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = dc-python-sdk
-version = 1.5.27
+version = 1.5.28
 author = DataConnector
 author_email = josh@dataconnector.com
 description = A small example package

{dc_python_sdk-1.5.27 → dc_python_sdk-1.5.28/src/dc_python_sdk.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dc-python-sdk
-Version: 1.5.27
+Version: 1.5.28
 Summary: Data Connector Python SDK
 Home-page: https://github.com/data-connector/dc-python-sdk
 Author: DataConnector

{dc_python_sdk-1.5.27 → dc_python_sdk-1.5.28}/src/dc_sdk/src/ai_http.py

@@ -17,6 +17,44 @@ app = FastAPI()
 
 _DEFAULT_PROCESS_TTL_SECONDS = 600.0
 
+METHOD_CONTEXT = {
+    "authenticate": "Authenticate using self.credentials. Return True when authentication is successful, otherwise raise a meaningful error.",
+    "get_objects": "Return a list of object dictionaries with keys: object_id, object_name, object_label, and optional object_group/visible.",
+    "get_fields": "Return a list of field dictionaries for an object_id with keys: field_id, field_name, field_label, and optional metadata like data_type/size.",
+    "get_data": "Return paginated data in shape {'next_page': <token_or_none>, 'data': [row_dict, ...]}. Respect object_id, field_ids, n_rows, filters, next_page.",
+    "load_data": "Load mapped destination rows into object_id using mapping m and update_method. Return True on success or raise a specific load/write error.",
+    "close": "Close open sessions/connections safely. Should not fail when resources were never initialized.",
+    "FOLLOWUP": "Refine output quality based on user follow-up feedback after a method succeeds.",
+}
+
+REQUIRED_METHODS = [
+    "authenticate",
+    "get_objects",
+    "get_fields",
+    "get_data",
+    "load_data",
+    "close",
+]
+
+
+METHOD_SHAPE_REFERENCE = """
+Reference behavior patterns (from production connectors like PostgreSQL):
+- authenticate(self) -> bool
+  - validates credentials and initializes client/connection state.
+- get_objects(self) -> list[dict]
+  - each object has: object_id, object_name, object_label, optional object_group.
+  - if we need to hardcode the objects, we should not do it here. Instead return an empty list with a comment that says objects pulled from API.
+- get_fields(self, object_id, options=dict()) -> list[dict]
+  - each field has: field_id, field_name, field_label, optional data_type/size.
+- get_data(self, object_id, field_ids, n_rows=None, filters=None, next_page=None, options=dict()) -> dict
+  - returns {"next_page": token_or_none, "data": [row_dict, ...]}.
+- load_data(self, data, object_id, m, update_method, batch_number: int, total_batches: int) -> bool
+  - maps/transforms rows and writes data to destination object.
+- close(self) -> None
+  - gracefully closes resources (session/connection/tunnel/etc).
+"""
+
+
 # Default ports: connector HTTP (dc-sdk http) 5000, AI FastAPI 5001, code-server 5002.
 CONNECTOR_PORT = os.getenv("CONNECTOR_PORT", 5000)
 AI_PORT = os.getenv("AI_PORT", 5001)
@@ -325,12 +363,6 @@ def _build_full_fix_prompt(
     stage: Optional[str],
     connector_context: ConnectorContext,
 ) -> str:
-    from dc_sdk.src.ai import (
-        METHOD_CONTEXT,
-        METHOD_SHAPE_REFERENCE,
-        REQUIRED_METHODS,
-    )
-
     extra_block = ""
     if connector_context.notes:
         extra_block = f"\nAdditional connector details:\n{connector_context.notes}\n"