robosystems-client 0.2.22__tar.gz → 0.2.23__tar.gz

{robosystems_client-0.2.22 → robosystems_client-0.2.23}/.github/workflows/tag-release.yml

@@ -71,31 +71,31 @@ jobs:
 
           if [ -z "$LAST_TAG" ]; then
             echo "No previous release found, analyzing all commits"
-            COMMIT_RANGE="$(git rev-list --max-parents=0 HEAD)..HEAD"
             LAST_TAG="initial"
+            # For initial release, use different commands that don't require a range
+            FILES_CHANGED=$(git ls-files | wc -l)
+            COMMITS_COUNT=$(git rev-list --count HEAD)
+            # Get stats by diffing empty tree against HEAD
+            EMPTY_TREE=$(git hash-object -t tree /dev/null)
+            ADDITIONS=$(git diff --shortstat "$EMPTY_TREE" HEAD | grep -oE '[0-9]+ insertions?' | grep -oE '[0-9]+' || echo "0")
+            DELETIONS=$(git diff --shortstat "$EMPTY_TREE" HEAD | grep -oE '[0-9]+ deletions?' | grep -oE '[0-9]+' || echo "0")
+            COMMIT_MESSAGES=$(git log --oneline --no-color HEAD | head -30)
+            DETAILED_CHANGES=$(git diff --name-status --no-color "$EMPTY_TREE" HEAD | head -50)
+            PR_REFS=$(git log --oneline --no-color HEAD | grep -oE '#[0-9]+' | sort -u | head -10 || echo "")
           else
-            COMMIT_RANGE="$LAST_TAG..HEAD"
             echo "Analyzing changes since last release: $LAST_TAG"
+            # Use explicit two-commit diff syntax instead of range syntax
+            FILES_CHANGED=$(git diff --name-only "$LAST_TAG" HEAD | wc -l)
+            COMMITS_COUNT=$(git rev-list --count "$LAST_TAG"..HEAD)
+            ADDITIONS=$(git diff --shortstat "$LAST_TAG" HEAD | grep -oE '[0-9]+ insertions?' | grep -oE '[0-9]+' || echo "0")
+            DELETIONS=$(git diff --shortstat "$LAST_TAG" HEAD | grep -oE '[0-9]+ deletions?' | grep -oE '[0-9]+' || echo "0")
+            COMMIT_MESSAGES=$(git log --oneline --no-color "$LAST_TAG"..HEAD | head -30)
+            DETAILED_CHANGES=$(git diff --name-status --no-color "$LAST_TAG" HEAD | head -50)
+            PR_REFS=$(git log --oneline --no-color "$LAST_TAG"..HEAD | grep -oE '#[0-9]+' | sort -u | head -10 || echo "")
           fi
 
-          # Get change statistics
-          FILES_CHANGED=$(git diff --name-only $COMMIT_RANGE | wc -l)
-          COMMITS_COUNT=$(git rev-list --count $COMMIT_RANGE)
-          ADDITIONS=$(git diff --shortstat $COMMIT_RANGE | grep -oE '[0-9]+ insertions?' | grep -oE '[0-9]+' || echo "0")
-          DELETIONS=$(git diff --shortstat $COMMIT_RANGE | grep -oE '[0-9]+ deletions?' | grep -oE '[0-9]+' || echo "0")
-
-          # Get commit messages (limit to prevent memory issues)
-          COMMIT_MESSAGES=$(git log --oneline --no-color $COMMIT_RANGE | head -30)
-
-          # Get file changes (limit to prevent memory issues)
-          DETAILED_CHANGES=$(git diff --name-status --no-color $COMMIT_RANGE | head -50)
-
-          # Get PR references from commit messages
-          PR_REFS=$(git log --oneline --no-color $COMMIT_RANGE | grep -oE '#[0-9]+' | sort -u | head -10 || echo "")
-
           # Save outputs
           echo "last_tag=$LAST_TAG" >> $GITHUB_OUTPUT
-          echo "commit_range=$COMMIT_RANGE" >> $GITHUB_OUTPUT
           echo "files_changed=$FILES_CHANGED" >> $GITHUB_OUTPUT
           echo "commits_count=$COMMITS_COUNT" >> $GITHUB_OUTPUT
           echo "additions=$ADDITIONS" >> $GITHUB_OUTPUT

{robosystems_client-0.2.22 → robosystems_client-0.2.23}/.gitignore

@@ -93,11 +93,6 @@ ipython_config.py
 #   install all needed dependencies.
 Dockerfile
 
-# celery beat schedule file
-celerybeat-schedule.db
-celerybeat-schedule
-celerybeat.pid
-
 # SageMath parsed files
 *.sage.py
 

{robosystems_client-0.2.22 → robosystems_client-0.2.23}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: robosystems-client
-Version: 0.2.22
+Version: 0.2.23
 Summary: Python Client for RoboSystems financial graph database API
 Author: RFS LLC
 License: MIT

{robosystems_client-0.2.22 → robosystems_client-0.2.23}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "robosystems-client"
-version = "0.2.22"
+version = "0.2.23"
 description = "Python Client for RoboSystems financial graph database API"
 readme = "README.md"
 requires-python = ">=3.10"

{robosystems_client-0.2.22 → robosystems_client-0.2.23}/pytest.ini

@@ -5,7 +5,6 @@ python_classes = Test*
 python_functions = test_*
 
 markers =
-    celery: marks tests that require celery
     integration: marks integration tests
     unit: marks unit tests
     asyncio: marks tests that use asyncio

{robosystems_client-0.2.22 → robosystems_client-0.2.23}/robosystems_client/api/agent/auto_select_agent.py

@@ -134,7 +134,7 @@ def sync_detailed(
   **Execution Strategies (automatic):**
   - Fast operations (<5s): Immediate synchronous response
   - Medium operations (5-30s): SSE streaming with progress updates
-  - Long operations (>30s): Async Celery worker with operation tracking
+  - Long operations (>30s): Background queue with operation tracking
 
   **Response Mode Override:**
   Use query parameter `?mode=sync|async` to override automatic strategy selection.
@@ -228,7 +228,7 @@ def sync(
   **Execution Strategies (automatic):**
   - Fast operations (<5s): Immediate synchronous response
   - Medium operations (5-30s): SSE streaming with progress updates
-  - Long operations (>30s): Async Celery worker with operation tracking
+  - Long operations (>30s): Background queue with operation tracking
 
   **Response Mode Override:**
   Use query parameter `?mode=sync|async` to override automatic strategy selection.
@@ -317,7 +317,7 @@ async def asyncio_detailed(
   **Execution Strategies (automatic):**
   - Fast operations (<5s): Immediate synchronous response
   - Medium operations (5-30s): SSE streaming with progress updates
-  - Long operations (>30s): Async Celery worker with operation tracking
+  - Long operations (>30s): Background queue with operation tracking
 
   **Response Mode Override:**
   Use query parameter `?mode=sync|async` to override automatic strategy selection.
@@ -409,7 +409,7 @@ async def asyncio(
   **Execution Strategies (automatic):**
   - Fast operations (<5s): Immediate synchronous response
   - Medium operations (5-30s): SSE streaming with progress updates
-  - Long operations (>30s): Async Celery worker with operation tracking
+  - Long operations (>30s): Background queue with operation tracking
 
   **Response Mode Override:**
   Use query parameter `?mode=sync|async` to override automatic strategy selection.

{robosystems_client-0.2.22 → robosystems_client-0.2.23}/robosystems_client/api/agent/execute_specific_agent.py

@@ -124,7 +124,7 @@ def sync_detailed(
   **Execution Strategies (automatic):**
   - Fast operations (<5s): Immediate synchronous response
   - Medium operations (5-30s): SSE streaming with progress updates
-  - Long operations (>30s): Async Celery worker with operation tracking
+  - Long operations (>30s): Background queue with operation tracking
 
   **Response Mode Override:**
   Use query parameter `?mode=sync|async` to override automatic strategy selection.
@@ -179,7 +179,7 @@ def sync(
   **Execution Strategies (automatic):**
   - Fast operations (<5s): Immediate synchronous response
   - Medium operations (5-30s): SSE streaming with progress updates
-  - Long operations (>30s): Async Celery worker with operation tracking
+  - Long operations (>30s): Background queue with operation tracking
 
   **Response Mode Override:**
   Use query parameter `?mode=sync|async` to override automatic strategy selection.
@@ -229,7 +229,7 @@ async def asyncio_detailed(
   **Execution Strategies (automatic):**
   - Fast operations (<5s): Immediate synchronous response
   - Medium operations (5-30s): SSE streaming with progress updates
-  - Long operations (>30s): Async Celery worker with operation tracking
+  - Long operations (>30s): Background queue with operation tracking
 
   **Response Mode Override:**
   Use query parameter `?mode=sync|async` to override automatic strategy selection.
@@ -282,7 +282,7 @@ async def asyncio(
   **Execution Strategies (automatic):**
   - Fast operations (<5s): Immediate synchronous response
   - Medium operations (5-30s): SSE streaming with progress updates
-  - Long operations (>30s): Async Celery worker with operation tracking
+  - Long operations (>30s): Background queue with operation tracking
 
   **Response Mode Override:**
   Use query parameter `?mode=sync|async` to override automatic strategy selection.

{robosystems_client-0.2.22 → robosystems_client-0.2.23}/robosystems_client/api/files/update_file.py

@@ -103,7 +103,7 @@ def sync_detailed(
   **What Happens (status='uploaded'):**
   1. File validated in S3
   2. Row count calculated
-  3. DuckDB staging triggered immediately (Celery task)
+  3. DuckDB staging triggered immediately (background task)
   4. If ingest_to_graph=true, graph ingestion queued
   5. File queryable in DuckDB within seconds
 
@@ -165,7 +165,7 @@ def sync(
   **What Happens (status='uploaded'):**
   1. File validated in S3
   2. Row count calculated
-  3. DuckDB staging triggered immediately (Celery task)
+  3. DuckDB staging triggered immediately (background task)
   4. If ingest_to_graph=true, graph ingestion queued
   5. File queryable in DuckDB within seconds
 
@@ -222,7 +222,7 @@ async def asyncio_detailed(
   **What Happens (status='uploaded'):**
   1. File validated in S3
   2. Row count calculated
-  3. DuckDB staging triggered immediately (Celery task)
+  3. DuckDB staging triggered immediately (background task)
   4. If ingest_to_graph=true, graph ingestion queued
   5. File queryable in DuckDB within seconds
 
@@ -282,7 +282,7 @@ async def asyncio(
   **What Happens (status='uploaded'):**
   1. File validated in S3
   2. Row count calculated
-  3. DuckDB staging triggered immediately (Celery task)
+  3. DuckDB staging triggered immediately (background task)
   4. If ingest_to_graph=true, graph ingestion queued
   5. File queryable in DuckDB within seconds
 

{robosystems_client-0.2.22 → robosystems_client-0.2.23}/robosystems_client/extensions/README.md

@@ -1,6 +1,6 @@
 # RoboSystems Python Client Extensions
 
-🚀 **Production-Ready Extensions** for the RoboSystems Financial Knowledge Graph API
+**Production-Ready Extensions** for the RoboSystems Financial Knowledge Graph API
 
 [![Python 3.7+](https://img.shields.io/badge/python-3.7+-blue.svg)](https://www.python.org/downloads/)
 [![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
@@ -17,7 +17,7 @@ The RoboSystems Python Client Extensions provide enhanced functionality for the
 - **Caching** with TTL and LRU eviction
 - **Full Async/Await Support** throughout
 
-## 🚀 Quick Start
+## Quick Start
 
 ### Installation
 
@@ -84,7 +84,7 @@ async def main():
 asyncio.run(main())
 ```
 
-## 🔐 Authentication
+## Authentication
 
 ### API Key Authentication (Recommended)
 
@@ -139,7 +139,7 @@ dev_ext = create_extensions(
 )
 ```
 
-## 🛠 Advanced Features
+## Advanced Features
 
 ### Query Builder
 
@@ -182,7 +182,7 @@ print(f"Complexity: {cost['complexity_category']} (score: {cost['complexity_scor
 
 # Get optimization recommendations  
 for rec in cost['recommendations']:
-    print(f"💡 {rec}")
+    print(f"Tip: {rec}")
 ```
 
 ### Result Processing
@@ -271,7 +271,7 @@ client.connect("operation_id")
 client.close()
 ```
 
-## 📊 Examples
+## Examples
 
 ### Financial Data Analysis
 
@@ -332,7 +332,7 @@ if final_batch:
     process_transaction_batch(final_batch)
     total_processed += len(final_batch)
 
-print(f"✅ Processed {total_processed:,} transactions total")
+print(f"Processed {total_processed:,} transactions total")
 ```
 
 ### Error Handling
@@ -361,7 +361,7 @@ except Exception as e:
     print(f"Query failed: {e}")
 ```
 
-## ⚡ Performance Optimization
+## Performance Optimization
 
 ### Connection Pooling
 
@@ -391,21 +391,21 @@ query = "MATCH (c:Company) WHERE c.revenue > 1000000 RETURN c"
 # Check syntax
 validation = validate_cypher_query(query)
 if not validation['valid']:
-    print("❌ Query has syntax errors:", validation['issues'])
+    print("Query has syntax errors:", validation['issues'])
     
 # Estimate cost
 cost = estimate_query_cost(query)
-print(f"📊 Query complexity: {cost['complexity_category']}")
+print(f"Query complexity: {cost['complexity_category']}")
 
 # Follow recommendations
 for rec in cost['recommendations']:
-    print(f"💡 Optimization tip: {rec}")
+    print(f"Optimization tip: {rec}")
 
 # Execute only if reasonable complexity
 if cost['complexity_category'] in ['low', 'medium']:
     result = extensions.execute_query("graph_id", query)
 else:
-    print("⚠️ Query may be too expensive - consider optimization")
+    print("Query may be too expensive - consider optimization")
 ```
 
 ### Caching Strategy
@@ -425,7 +425,7 @@ else:
     cache = results_cache
 ```
 
-## 🧪 Testing
+## Testing
 
 Run the test suite:
 
@@ -457,7 +457,7 @@ def test_query_execution():
         assert result["data"] == [{"count": 100}]
 ```
 
-## 🔧 Configuration
+## Configuration
 
 ### Environment Variables
 
@@ -492,7 +492,7 @@ config = RoboSystemsExtensionConfig(
 extensions = RoboSystemsExtensions(config)
 ```
 
-## 📚 API Reference
+## API Reference
 
 ### Core Classes
 
@@ -528,7 +528,7 @@ extensions = RoboSystemsExtensions(config)
 - **`format_duration(milliseconds)`** - Human-readable time formatting
 - **`create_extensions(method, **kwargs)`** - Extensions factory
 
-## 🐛 Troubleshooting
+## Troubleshooting
 
 ### Common Issues
 
@@ -545,9 +545,9 @@ pip install pandas # For DataFrame conversion (optional)
 extensions = AuthenticatedExtensions("your-api-key")
 try:
     result = extensions.execute_query("graph_id", "MATCH (n) RETURN count(n) LIMIT 1")
-    print("✅ Authentication successful")
+    print("Authentication successful")
 except Exception as e:
-    print(f"❌ Auth failed: {e}")
+    print(f"Auth failed: {e}")
 ```
 
 **Connection Issues**
@@ -585,11 +585,7 @@ logging.basicConfig(level=logging.DEBUG)
 extensions = AuthenticatedExtensions("your-key")
 ```
 
-## 📄 License
-
-MIT License - see [LICENSE](LICENSE) file for details.
-
-## 🤝 Contributing
+## Contributing
 
 1. Fork the repository
 2. Create a feature branch
@@ -597,7 +593,7 @@ MIT License - see [LICENSE](LICENSE) file for details.
 4. Run the test suite: `python run_tests.py`
 5. Submit a pull request
 
-## 📞 Support
+## Support
 
 - **API Reference**: [api.robosystems.ai](https://api.robosystems.ai)
 - **Issues**: [GitHub Issues](https://github.com/RoboFinSystems/robosystems-python-client/issues)

{robosystems_client-0.2.22 → robosystems_client-0.2.23}/robosystems_client/extensions/agent_client.py

@@ -177,7 +177,7 @@ class AgentClient:
               else datetime.now().isoformat(),
             )
 
-        # Check if this is a queued response (async Celery execution)
+        # Check if this is a queued response (async background task execution)
         is_queued = False
         queued_response = None
 

{robosystems_client-0.2.22 → robosystems_client-0.2.23}/robosystems_client/extensions/materialization_client.py

@@ -15,6 +15,7 @@ from ..api.materialize.get_materialization_status import (
   sync_detailed as get_materialization_status,
 )
 from ..models.materialize_request import MaterializeRequest
+from .operation_client import OperationClient, OperationProgress, MonitorOptions
 
 logger = logging.getLogger(__name__)
 
@@ -27,6 +28,7 @@ class MaterializationOptions:
   rebuild: bool = False
   force: bool = False
   on_progress: Optional[Callable[[str], None]] = None
+  timeout: Optional[int] = 600  # 10 minute default timeout
 
 
 @dataclass
@@ -66,6 +68,14 @@ class MaterializationClient:
     self.base_url = config["base_url"]
     self.headers = config.get("headers", {})
     self.token = config.get("token")
+    self._operation_client = None
+
+  @property
+  def operation_client(self) -> OperationClient:
+    """Get or create the operation client for SSE monitoring."""
+    if self._operation_client is None:
+      self._operation_client = OperationClient(self.config)
+    return self._operation_client
 
   def materialize(
     self,
@@ -75,13 +85,13 @@ class MaterializationClient:
     """
     Materialize graph from DuckDB staging tables.
 
-    Rebuilds the complete graph database from the current state of DuckDB
-    staging tables. Automatically discovers all tables, materializes them in
-    the correct order (nodes before relationships), and clears the staleness flag.
+    Submits a materialization job to Dagster and monitors progress via SSE.
+    The operation runs asynchronously on the server but this method waits
+    for completion and returns the final result.
 
     Args:
         graph_id: Graph database identifier
-        options: Materialization options (ignore_errors, rebuild, force)
+        options: Materialization options (ignore_errors, rebuild, force, timeout)
 
     Returns:
         MaterializationResult with detailed execution information
@@ -96,7 +106,7 @@ class MaterializationClient:
 
     try:
       if options.on_progress:
-        options.on_progress("Starting graph materialization...")
+        options.on_progress("Submitting materialization job...")
 
       request = MaterializeRequest(
         ignore_errors=options.ignore_errors,
@@ -125,6 +135,7 @@ class MaterializationClient:
 
       response = materialize_graph(**kwargs)
 
+      # Handle non-200 status codes
       if response.status_code != 200 or not response.parsed:
         error_msg = f"Materialization failed: {response.status_code}"
         if hasattr(response, "content"):
@@ -148,25 +159,68 @@ class MaterializationClient:
           error=error_msg,
         )
 
+      # Get the operation_id from the queued response
       result_data = response.parsed
+      operation_id = result_data.operation_id
 
       if options.on_progress:
-        options.on_progress(
-          f"✅ Materialization complete: {len(result_data.tables_materialized)} tables, "
-          f"{result_data.total_rows:,} rows in {result_data.execution_time_ms:.2f}ms"
-        )
-
-      return MaterializationResult(
-        status=result_data.status,
-        was_stale=result_data.was_stale,
-        stale_reason=result_data.stale_reason,
-        tables_materialized=result_data.tables_materialized,
-        total_rows=result_data.total_rows,
-        execution_time_ms=result_data.execution_time_ms,
-        message=result_data.message,
-        success=True,
+        options.on_progress(f"Materialization queued (operation: {operation_id})")
+
+      # Monitor the operation via SSE until completion
+      def on_sse_progress(progress: OperationProgress):
+        if options.on_progress:
+          msg = progress.message
+          if progress.percentage is not None:
+            msg += f" ({progress.percentage:.0f}%)"
+          options.on_progress(msg)
+
+      monitor_options = MonitorOptions(
+        on_progress=on_sse_progress,
+        timeout=options.timeout,
       )
 
+      op_result = self.operation_client.monitor_operation(operation_id, monitor_options)
+
+      # Convert operation result to materialization result
+      if op_result.status.value == "completed":
+        # Extract details from SSE completion event result
+        sse_result = op_result.result or {}
+
+        if options.on_progress:
+          tables = sse_result.get("tables_materialized", [])
+          rows = sse_result.get("total_rows", 0)
+          time_ms = sse_result.get("execution_time_ms", 0)
+          options.on_progress(
+            f"✅ Materialization complete: {len(tables)} tables, "
+            f"{rows:,} rows in {time_ms:.2f}ms"
+          )
+
+        return MaterializationResult(
+          status="success",
+          was_stale=sse_result.get("was_stale", False),
+          stale_reason=sse_result.get("stale_reason"),
+          tables_materialized=sse_result.get("tables_materialized", []),
+          total_rows=sse_result.get("total_rows", 0),
+          execution_time_ms=sse_result.get(
+            "execution_time_ms", op_result.execution_time_ms or 0
+          ),
+          message=sse_result.get("message", "Graph materialized successfully"),
+          success=True,
+        )
+      else:
+        # Operation failed or was cancelled
+        return MaterializationResult(
+          status=op_result.status.value,
+          was_stale=False,
+          stale_reason=None,
+          tables_materialized=[],
+          total_rows=0,
+          execution_time_ms=op_result.execution_time_ms or 0,
+          message=op_result.error or f"Operation {op_result.status.value}",
+          success=False,
+          error=op_result.error,
+        )
+
     except Exception as e:
       logger.error(f"Materialization failed: {e}")
       return MaterializationResult(

{robosystems_client-0.2.22 → robosystems_client-0.2.23}/robosystems_client/extensions/sse_client.py

@@ -104,8 +104,10 @@ class SSEClient:
 
     try:
       self.client = httpx.Client(timeout=self.config.timeout)
-      self._response = self.client.stream("GET", url, params=params, headers=headers)
-      self._response.__enter__()
+      self._context_manager = self.client.stream(
+        "GET", url, params=params, headers=headers
+      )
+      self._response = self._context_manager.__enter__()
 
       self.reconnect_attempts = 0
       self.emit("connected", None)
@@ -124,11 +126,9 @@ class SSEClient:
 
     try:
       event_buffer = {"event": None, "data": [], "id": None, "retry": None}
-      print("[SSE DEBUG] Starting to process events...")
 
       for line in self._response.iter_lines():
         if self.closed:
-          print("[SSE DEBUG] Stream closed, breaking out of loop")
           break
 
         line = line.strip()
@@ -136,7 +136,6 @@ class SSEClient:
         # Empty line indicates end of event
         if not line:
           if event_buffer["data"] or event_buffer["event"]:
-            print(f"[SSE DEBUG] Dispatching event: {event_buffer.get('event')}")
             self._dispatch_event(event_buffer)
           event_buffer = {"event": None, "data": [], "id": None, "retry": None}
           continue
@@ -172,13 +171,9 @@ class SSEClient:
 
       # Handle final event if stream ends without empty line
       if event_buffer["data"] or event_buffer["event"]:
-        print("[SSE DEBUG] Dispatching final event after stream end")
         self._dispatch_event(event_buffer)
 
-      print("[SSE DEBUG] Event processing loop ended")
-
     except Exception as error:
-      print(f"[SSE DEBUG] Exception in event processing: {error}")
       if not self.closed:
         self.emit("error", error)
 
@@ -214,13 +209,14 @@ class SSEClient:
     # Emit typed event
     self.emit(event_type, parsed_data)
 
-    # Check for completion events
+    # Check for completion events - just set flag, don't close from within loop
+    # The loop will break on next iteration and close() will be called in finally
     if event_type in [
       EventType.OPERATION_COMPLETED.value,
       EventType.OPERATION_ERROR.value,
       EventType.OPERATION_CANCELLED.value,
     ]:
-      self.close()
+      self.closed = True
 
   def _handle_error(
     self, error: Exception, operation_id: str, from_sequence: int
@@ -285,12 +281,13 @@ class SSEClient:
     """Close the SSE connection"""
     self.closed = True
 
-    if self._response:
+    if hasattr(self, "_context_manager") and self._context_manager:
       try:
-        self._response.__exit__(None, None, None)
+        self._context_manager.__exit__(None, None, None)
       except Exception:
         pass
-      self._response = None
+      self._context_manager = None
+    self._response = None
 
     if self.client:
       self.client.close()
@@ -334,10 +331,10 @@ class AsyncSSEClient:
 
     try:
       self.client = httpx.AsyncClient(timeout=self.config.timeout)
-      self._response = await self.client.stream(
+      self._context_manager = self.client.stream(
         "GET", url, params=params, headers=headers
       )
-      await self._response.__aenter__()
+      self._response = await self._context_manager.__aenter__()
 
       self.reconnect_attempts = 0
       self.emit("connected", None)
@@ -401,13 +398,9 @@ class AsyncSSEClient:
 
       # Handle final event if stream ends without empty line
       if event_buffer["data"] or event_buffer["event"]:
-        print("[SSE DEBUG] Dispatching final event after stream end")
         self._dispatch_event(event_buffer)
 
-      print("[SSE DEBUG] Event processing loop ended")
-
     except Exception as error:
-      print(f"[SSE DEBUG] Exception in event processing: {error}")
       if not self.closed:
         self.emit("error", error)
 
@@ -443,13 +436,14 @@ class AsyncSSEClient:
     # Emit typed event
     self.emit(event_type, parsed_data)
 
-    # Check for completion events
+    # Check for completion events - just set flag, don't close from within loop
+    # The loop will break on next iteration and close() will be called in finally
     if event_type in [
       EventType.OPERATION_COMPLETED.value,
       EventType.OPERATION_ERROR.value,
       EventType.OPERATION_CANCELLED.value,
     ]:
-      asyncio.create_task(self.close())
+      self.closed = True
 
   async def _handle_error(
     self, error: Exception, operation_id: str, from_sequence: int
@@ -512,12 +506,13 @@ class AsyncSSEClient:
     """Close the SSE connection (async)"""
     self.closed = True
 
-    if self._response:
+    if hasattr(self, "_context_manager") and self._context_manager:
       try:
-        await self._response.__aexit__(None, None, None)
+        await self._context_manager.__aexit__(None, None, None)
       except Exception:
         pass
-      self._response = None
+      self._context_manager = None
+    self._response = None
 
     if self.client:
       await self.client.aclose()

{robosystems_client-0.2.22 → robosystems_client-0.2.23}/robosystems_client/extensions/subgraph_workspace_client.py

@@ -197,7 +197,9 @@ class SubgraphWorkspaceClient:
         sse_url = f"{self.api._base_url}/v1/operations/{operation_id}/stream"
         headers = {"X-API-Key": self.api.token}
 
-        async with httpx.AsyncClient() as client:
+        # Use longer timeout for SSE streaming (Dagster jobs can take time)
+        timeout = httpx.Timeout(connect=30.0, read=120.0, write=30.0, pool=30.0)
+        async with httpx.AsyncClient(timeout=timeout) as client:
           async with client.stream("GET", sse_url, headers=headers) as sse_response:
             async for line in sse_response.aiter_lines():
               if line.startswith("data: "):