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

{dc_python_sdk-1.5.26/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.26
+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.26 → 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.26"
+version = "1.5.28"
 description = "Data Connector Python SDK"
 readme = "README.md"
 requires-python = ">=3.6"

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

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

{dc_python_sdk-1.5.26 → 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.26
+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.26 → 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)
@@ -60,7 +98,7 @@ class ChatMessage(BaseModel):
 class FeedbackRequest(BaseModel):
     credentials: Dict
     feedback: str
-    connector_context: Optional[ConnectorContext] = None
+    connector_context: ConnectorContext
     failed_stage: Optional[str] = None
     # Prior turns from last response’s ai_messages; new user content is only req.feedback this round.
     ai_messages: Optional[List[ChatMessage]] = None
@@ -71,7 +109,7 @@ class FeedbackRequest(BaseModel):
 class RunRequest(BaseModel):
     credentials: List[Dict]
     max_attempts: Optional[int] = 5
-    connector_context: Optional[ConnectorContext] = None
+    connector_context: ConnectorContext
     ai_messages: Optional[List[ChatMessage]] = None
     # When objects_dynamic is false: required — every id must appear in get_objects results.
     # When objects_dynamic is true and get_objects returns []: required — used to probe get_fields/get_data.
@@ -98,21 +136,8 @@ def read_connector():
     with open(f"{workspace}/src/connector.py", "r") as f:
         return f.read()
 
-def _resolved_connector_context(opt: Optional[ConnectorContext]) -> ConnectorContext:
-    from dc_sdk.src.ai import CONNECTOR_CONTEXT as defaults
-
-    merged: Dict[str, Any] = {
-        "name": defaults["name"],
-        "auth_type": defaults["auth_type"],
-        "base_url": defaults["base_url"],
-        "objects_dynamic": False,
-        "notes": None,
-    }
-    if opt is not None:
-        for key, value in opt.model_dump(exclude_unset=True).items():
-            if value is not None:
-                merged[key] = value
-    return ConnectorContext(**merged)
+def _resolved_connector_context(ctx: ConnectorContext) -> ConnectorContext:
+    return ctx
 
 def _chat_messages_to_openai(messages: Optional[List[ChatMessage]]) -> List[Dict[str, str]]:
     if not messages:
@@ -338,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"